bgp/rib-impl/src/main/java/org/opendaylight/controller/config/yang/bgp/rib/impl/RIBImplModule.java

   1 /*
   2  * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 /**
   9  * Generated file
  10
  11  * Generated from: yang module name: bgp-rib-impl  yang module local name: rib-impl
  12  * Generated by: org.opendaylight.controller.config.yangjmxgenerator.plugin.JMXGenerator
  13  * Generated at: Wed Nov 06 13:02:32 CET 2013
  14  *
  15  * Do not modify this file unless it is present under src/main directory
  16  */
  17 package org.opendaylight.controller.config.yang.bgp.rib.impl;
  18
  19 import java.util.Collections;
  20 import java.util.Hashtable;
  21 import java.util.List;
  22 import java.util.Map;
  23 import java.util.stream.Collectors;
  24 import org.opendaylight.controller.config.api.JmxAttributeValidationException;
  25 import org.opendaylight.controller.md.sal.dom.api.DOMDataBroker;
  26 import org.opendaylight.controller.sal.core.api.model.SchemaService;
  27 import org.opendaylight.protocol.bgp.mode.api.PathSelectionMode;
  28 import org.opendaylight.protocol.bgp.openconfig.spi.BGPConfigModuleTracker;
  29 import org.opendaylight.protocol.bgp.openconfig.spi.BGPOpenconfigMapper;
  30 import org.opendaylight.protocol.bgp.openconfig.spi.InstanceConfigurationIdentifier;
  31 import org.opendaylight.protocol.bgp.openconfig.spi.pojo.BGPRibInstanceConfiguration;
  32 import org.opendaylight.protocol.bgp.rib.impl.RIBImpl;
  33 import org.opendaylight.protocol.bgp.rib.impl.spi.BGPBestPathSelection;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.BgpTableType;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.TablesKey;
  36 import org.opendaylight.yangtools.sal.binding.generator.impl.GeneratedClassLoadingStrategy;
  37 import org.opendaylight.yangtools.yang.model.api.SchemaContextListener;
  38 import org.osgi.framework.BundleContext;
  39
  40 /**
  41  *
  42  */
  43 public final class RIBImplModule extends org.opendaylight.controller.config.yang.bgp.rib.impl.AbstractRIBImplModule {
  44
  45     private static final String IS_NOT_SET = "is not set.";
  46     private BundleContext bundleContext;
  47
  48     public RIBImplModule(final org.opendaylight.controller.config.api.ModuleIdentifier name,
  49             final org.opendaylight.controller.config.api.DependencyResolver dependencyResolver) {
  50         super(name, dependencyResolver);
  51     }
  52
  53     public RIBImplModule(final org.opendaylight.controller.config.api.ModuleIdentifier name,
  54             final org.opendaylight.controller.config.api.DependencyResolver dependencyResolver, final RIBImplModule oldModule,
  55             final java.lang.AutoCloseable oldInstance) {
  56         super(name, dependencyResolver, oldModule, oldInstance);
  57     }
  58
  59     @Override
  60     public void customValidation() {
  61         JmxAttributeValidationException.checkNotNull(getExtensions(), IS_NOT_SET, extensionsJmxAttribute);
  62         JmxAttributeValidationException.checkNotNull(getRibId(), IS_NOT_SET, ribIdJmxAttribute);
  63         JmxAttributeValidationException.checkNotNull(getLocalAs(), IS_NOT_SET, localAsJmxAttribute);
  64         JmxAttributeValidationException.checkNotNull(getBgpRibId(), IS_NOT_SET, bgpRibIdJmxAttribute);
  65         JmxAttributeValidationException.checkNotNull(getLocalTable(), IS_NOT_SET, localTableJmxAttribute);
  66     }
  67
  68     @Override
  69     public java.lang.AutoCloseable createInstance() {
  70         final Map<TablesKey, PathSelectionMode> pathSelectionStrategies = mapBestPathSelectionStrategyByFamily(getRibPathSelectionModeDependency());
  71         final RIBImpl rib = new RIBImpl(getRibId(), getLocalAs(), getBgpRibId(), getClusterId(), getExtensionsDependency(),
  72             getBgpDispatcherDependency(), getCodecTreeFactoryDependency(), getDomDataProviderDependency(), getLocalTableDependency(), pathSelectionStrategies, classLoadingStrategy(),
  73             new RIBImplModuleTracker(getGlobalWriter()), getOpenconfigProviderDependency());
  74         registerSchemaContextListener(rib);
  75
  76         rib.registerRootRuntimeBean(getRootRuntimeBeanRegistratorWrapper());
  77         return rib;
  78     }
  79
  80     private GeneratedClassLoadingStrategy classLoadingStrategy() {
  81         return getExtensionsDependency().getClassLoadingStrategy();
  82     }
  83
  84     private void registerSchemaContextListener(final RIBImpl rib) {
  85         final DOMDataBroker domBroker = getDomDataProviderDependency();
  86         if(domBroker instanceof SchemaService) {
  87             ((SchemaService) domBroker).registerSchemaContextListener(rib);
  88         } else {
  89             // FIXME:Get bundle context and register global schema service from bundle
  90             // context.
  91             bundleContext.registerService(SchemaContextListener.class, rib, new Hashtable<String,String>());
  92         }
  93     }
  94
  95     private Map<TablesKey, PathSelectionMode> mapBestPathSelectionStrategyByFamily(final List<BGPBestPathSelection> bestPathSelectionDependency) {
  96         return Collections.unmodifiableMap(bestPathSelectionDependency.stream().collect(
  97             Collectors.toMap(st -> new TablesKey(st.getAfi(), st.getSafi()), st -> st.getStrategy())));
  98     }
  99
 100     public void setBundleContext(final BundleContext bundleContext) {
 101         this.bundleContext = bundleContext;
 102     }
 103
 104     private BGPOpenconfigMapper<BGPRibInstanceConfiguration> getGlobalWriter() {
 105         if (getOpenconfigProviderDependency() != null) {
 106             return getOpenconfigProviderDependency().getOpenConfigMapper(BGPRibInstanceConfiguration.class);
 107         }
 108         return null;
 109     }
 110
 111     private final class RIBImplModuleTracker implements BGPConfigModuleTracker {
 112
 113         private final BGPOpenconfigMapper<BGPRibInstanceConfiguration> globalWriter;
 114         private final BGPRibInstanceConfiguration bgpRibConfig;
 115
 116         public RIBImplModuleTracker(final BGPOpenconfigMapper<BGPRibInstanceConfiguration> globalWriter) {
 117             this.globalWriter = globalWriter;
 118             final InstanceConfigurationIdentifier identifier = new InstanceConfigurationIdentifier(getIdentifier().getInstanceName());
 119             final List<BgpTableType> tableDependency = getLocalTableDependency();
 120             this.bgpRibConfig = new BGPRibInstanceConfiguration(identifier, getLocalAs(), getBgpRibId(), getClusterId(), tableDependency,
 121                     mapBestPathSelectionStrategyByFamily(getRibPathSelectionModeDependency()));
 122         }
 123
 124         @Override
 125         public void onInstanceCreate() {
 126             if (globalWriter != null) {
 127                 globalWriter.writeConfiguration(this.bgpRibConfig);
 128             }
 129         }
 130
 131         @Override
 132         public void onInstanceClose() {
 133             if (globalWriter != null) {
 134                 globalWriter.removeConfiguration(this.bgpRibConfig);
 135             }
 136         }
 137
 138     }
 139
 140
 141 }