tests/honeynode/minimal-distribution-core/src/main/java/io/fd/honeycomb/infra/distro/initializer/InitializerRegistryAdapter.java

   1 /*
   2  * Copyright (c) 2016 Cisco and/or its affiliates.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at:
   7  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 package io.fd.honeycomb.infra.distro.initializer;
  18
  19 import io.fd.honeycomb.data.init.DataTreeInitializer;
  20 import io.fd.honeycomb.data.init.InitializerRegistry;
  21 import io.fd.honeycomb.translate.MappingContext;
  22 import io.fd.honeycomb.translate.ModificationCache;
  23 import io.fd.honeycomb.translate.read.ReadContext;
  24 import io.fd.honeycomb.translate.read.registry.InitRegistry;
  25
  26 import javax.annotation.Nonnull;
  27
  28 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  29 import org.opendaylight.controller.md.sal.binding.api.DataTreeIdentifier;
  30 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  31 import org.opendaylight.yang.gen.v1.http.org.openroadm.device.rev170206.org.openroadm.device.container.OrgOpenroadmDevice;
  32 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  33 import org.slf4j.Logger;
  34 import org.slf4j.LoggerFactory;
  35
  36 final class InitializerRegistryAdapter implements InitializerRegistry {
  37
  38     private static final Logger LOG = LoggerFactory.getLogger(InitializerRegistryAdapter.class);
  39
  40     private final DataTreeInitializer configInitializer;
  41     private final DataTreeInitializer contextInitializer;
  42     private final InitRegistry initRegistry;
  43     private final DataBroker dataBroker;
  44     private final MappingContext realtimeMappingContext;
  45     /** Add by Martial. */
  46     private final DataBroker deviceDataBroker;
  47     private final DataBroker hcConfigDataBroker;
  48     private final InstanceIdentifier<OrgOpenroadmDevice> DEVICE_CONTAINER_ID = InstanceIdentifier
  49             .create(OrgOpenroadmDevice.class);
  50
  51     InitializerRegistryAdapter(final DataTreeInitializer configInitializer,
  52                                final DataTreeInitializer contextInitializer,
  53                                final InitRegistry initRegistry,
  54                                final DataBroker noopConfigDataBroker,
  55                                final MappingContext realtimeMappingContext,
  56                                final DataBroker deviceDataBroker,
  57                                final DataBroker hcConfigDataBroker) {
  58         this.configInitializer = configInitializer;
  59         this.contextInitializer = contextInitializer;
  60         this.initRegistry = initRegistry;
  61         this.dataBroker = noopConfigDataBroker;
  62         this.realtimeMappingContext = realtimeMappingContext;
  63         this.deviceDataBroker = deviceDataBroker;
  64         this.hcConfigDataBroker = hcConfigDataBroker;
  65     }
  66
  67     @Override
  68     public void initialize() throws DataTreeInitializer.InitializeException {
  69         LOG.info("Config initialization started");
  70         try {
  71             LOG.info("Register device databroker listener ...");
  72             deviceDataBroker
  73             .registerDataTreeChangeListener(new DataTreeIdentifier<>(LogicalDatastoreType.CONFIGURATION,
  74                     DEVICE_CONTAINER_ID), new DeviceChangeListener(deviceDataBroker, hcConfigDataBroker));
  75             LOG.info("Device databroker listener registered");
  76             // Initialize contexts first so that other initializers can find any relevant mapping before initializing
  77             // configuration to what is already in VPP
  78             contextInitializer.initialize();
  79             LOG.info("Persisted context restored successfully");
  80             // Initialize all registered initializers
  81             initRegistry.initAll(dataBroker, new InitReadContext(realtimeMappingContext));
  82             LOG.info("Configuration initialized successfully");
  83             // Initialize stored configuration on top
  84             configInitializer.initialize();
  85             LOG.info("Persisted configuration restored successfully");
  86         } catch (Exception e) {
  87             LOG.warn("Failed to initialize config", e);
  88         }
  89
  90         LOG.info("Honeycomb initialized");
  91     }
  92
  93     private static final class InitReadContext implements ReadContext {
  94
  95         private final ModificationCache modificationCache;
  96         private final MappingContext realtimeMappingContext;
  97
  98         InitReadContext(final MappingContext realtimeMappingContext) {
  99             modificationCache = new ModificationCache();
 100             this.realtimeMappingContext = realtimeMappingContext;
 101         }
 102
 103         @Nonnull
 104         @Override
 105         public ModificationCache getModificationCache() {
 106             return modificationCache;
 107         }
 108
 109         @Nonnull
 110         @Override
 111         public MappingContext getMappingContext() {
 112             return realtimeMappingContext;
 113         }
 114
 115         @Override
 116         public void close() {
 117             modificationCache.close();
 118         }
 119     }
 120 }