X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=opendaylight%2Fblueprint%2Fsrc%2Fmain%2Fjava%2Forg%2Fopendaylight%2Fcontroller%2Fblueprint%2FBlueprintContainerRestartServiceImpl.java;h=d0e39873d99b3d223e70d15f11694853214da428;hb=refs%2Fchanges%2F79%2F38279%2F7;hp=3e04f213495c2cc37c52b19fdf4348e524a9b2bb;hpb=36de1fd1c214d98ae97e6d7725f323dd01e34977;p=controller.git diff --git a/opendaylight/blueprint/src/main/java/org/opendaylight/controller/blueprint/BlueprintContainerRestartServiceImpl.java b/opendaylight/blueprint/src/main/java/org/opendaylight/controller/blueprint/BlueprintContainerRestartServiceImpl.java index 3e04f21349..d0e39873d9 100644 --- a/opendaylight/blueprint/src/main/java/org/opendaylight/controller/blueprint/BlueprintContainerRestartServiceImpl.java +++ b/opendaylight/blueprint/src/main/java/org/opendaylight/controller/blueprint/BlueprintContainerRestartServiceImpl.java @@ -7,25 +7,65 @@ */ package org.opendaylight.controller.blueprint; +import com.google.common.base.Optional; +import com.google.common.collect.Lists; import com.google.common.util.concurrent.ThreadFactoryBuilder; +import java.lang.management.ManagementFactory; +import java.util.AbstractMap.SimpleEntry; import java.util.ArrayList; +import java.util.Dictionary; +import java.util.Hashtable; import java.util.LinkedHashSet; import java.util.List; +import java.util.Map.Entry; import java.util.Set; +import java.util.concurrent.CountDownLatch; import java.util.concurrent.ExecutorService; import java.util.concurrent.Executors; +import java.util.concurrent.TimeUnit; +import javax.annotation.Nullable; +import javax.management.InstanceNotFoundException; +import javax.management.ObjectName; +import javax.xml.parsers.ParserConfigurationException; import org.apache.aries.blueprint.services.BlueprintExtenderService; +import org.apache.aries.util.AriesFrameworkUtil; +import org.opendaylight.controller.config.api.ConfigRegistry; +import org.opendaylight.controller.config.api.ConflictingVersionException; +import org.opendaylight.controller.config.api.ModuleIdentifier; +import org.opendaylight.controller.config.api.ValidationException; +import org.opendaylight.controller.config.facade.xml.ConfigExecution; +import org.opendaylight.controller.config.facade.xml.ConfigSubsystemFacade; +import org.opendaylight.controller.config.facade.xml.ConfigSubsystemFacadeFactory; +import org.opendaylight.controller.config.facade.xml.TestOption; +import org.opendaylight.controller.config.facade.xml.mapping.config.Config; +import org.opendaylight.controller.config.facade.xml.strategy.EditStrategyType; +import org.opendaylight.controller.config.util.ConfigRegistryJMXClient; +import org.opendaylight.controller.config.util.xml.DocumentedException; +import org.opendaylight.controller.config.util.xml.XmlElement; +import org.opendaylight.controller.config.util.xml.XmlMappingConstants; +import org.opendaylight.controller.config.util.xml.XmlUtil; import org.osgi.framework.Bundle; +import org.osgi.framework.BundleContext; import org.osgi.framework.ServiceReference; +import org.osgi.framework.ServiceRegistration; +import org.osgi.service.blueprint.container.EventConstants; +import org.osgi.service.event.Event; +import org.osgi.service.event.EventHandler; import org.slf4j.Logger; import org.slf4j.LoggerFactory; +import org.w3c.dom.Document; +import org.w3c.dom.Element; -/** Implementation of the BlueprintContainerRestartService. +/** + * Implementation of the BlueprintContainerRestartService. * * @author Thomas Pantelis */ class BlueprintContainerRestartServiceImpl implements AutoCloseable, BlueprintContainerRestartService { private static final Logger LOG = LoggerFactory.getLogger(BlueprintContainerRestartServiceImpl.class); + private static final String CONFIG_MODULE_NAMESPACE_PROP = "config-module-namespace"; + private static final String CONFIG_MODULE_NAME_PROP = "config-module-name"; + private static final String CONFIG_INSTANCE_NAME_PROP = "config-instance-name"; private final ExecutorService restartExecutor = Executors.newSingleThreadExecutor(new ThreadFactoryBuilder(). setDaemon(true).setNameFormat("BlueprintContainerRestartService").build()); @@ -49,8 +89,10 @@ class BlueprintContainerRestartServiceImpl implements AutoCloseable, BlueprintCo } private void restartContainerAndDependentsInternal(Bundle forBundle) { + // We use a LinkedHashSet to preserve insertion order as we walk the service usage hierarchy. Set containerBundlesSet = new LinkedHashSet<>(); - findDependentContainersRecursively(forBundle, containerBundlesSet); + List> configModules = new ArrayList<>(); + findDependentContainersRecursively(forBundle, containerBundlesSet, configModules); List containerBundles = new ArrayList<>(containerBundlesSet); @@ -58,11 +100,25 @@ class BlueprintContainerRestartServiceImpl implements AutoCloseable, BlueprintCo containerBundles.subList(1, containerBundles.size())); // Destroy the containers in reverse order with 'forBundle' last, ie bottom-up in the service tree. - for(int i = containerBundles.size() - 1; i >= 0; i--) { - Bundle bundle = containerBundles.get(i); + for(Bundle bundle: Lists.reverse(containerBundles)) { blueprintExtenderService.destroyContainer(bundle, blueprintExtenderService.getContainer(bundle)); } + // The blueprint containers are created asynchronously so we register a handler for blueprint events + // that are sent when a container is complete, successful or not. The CountDownLatch tells when all + // containers are complete. This is done to ensure all blueprint containers are finished before we + // restart config modules. + final CountDownLatch containerCreationComplete = new CountDownLatch(containerBundles.size()); + ServiceRegistration eventHandlerReg = registerEventHandler(forBundle.getBundleContext(), new EventHandler() { + @Override + public void handleEvent(Event event) { + LOG.debug("handleEvent {} for bundle {}", event.getTopic(), event.getProperty(EventConstants.BUNDLE)); + if(containerBundles.contains(event.getProperty(EventConstants.BUNDLE))) { + containerCreationComplete.countDown(); + } + } + }); + // Restart the containers top-down starting with 'forBundle'. for(Bundle bundle: containerBundles) { List paths = BlueprintBundleTracker.findBlueprintPaths(bundle); @@ -71,29 +127,117 @@ class BlueprintContainerRestartServiceImpl implements AutoCloseable, BlueprintCo blueprintExtenderService.createContainer(bundle, paths); } + + try { + containerCreationComplete.await(5, TimeUnit.MINUTES); + } catch(InterruptedException e) { + LOG.debug("CountDownLatch await was interrupted - returning"); + return; + } + + AriesFrameworkUtil.safeUnregisterService(eventHandlerReg); + + // Now restart any associated config system Modules. + restartConfigModules(forBundle.getBundleContext(), configModules); + } + + private void restartConfigModules(BundleContext bundleContext, List> configModules) { + if(configModules.isEmpty()) { + return; + } + + ServiceReference configFacadeFactoryRef = + bundleContext.getServiceReference(ConfigSubsystemFacadeFactory.class); + if(configFacadeFactoryRef == null) { + LOG.debug("ConfigSubsystemFacadeFactory service reference not found"); + return; + } + + ConfigSubsystemFacadeFactory configFacadeFactory = bundleContext.getService(configFacadeFactoryRef); + if(configFacadeFactory == null) { + LOG.debug("ConfigSubsystemFacadeFactory service not found"); + return; + } + + ConfigSubsystemFacade configFacade = configFacadeFactory.createFacade("BlueprintContainerRestartService"); + try { + restartConfigModules(configModules, configFacade); + } catch(Exception e) { + LOG.error("Error restarting config modules", e); + } finally { + configFacade.close(); + bundleContext.ungetService(configFacadeFactoryRef); + } + + } + + private void restartConfigModules(List> configModules, + ConfigSubsystemFacade configFacade) throws ParserConfigurationException, DocumentedException, + ValidationException, ConflictingVersionException { + + Document document = XmlUtil.newDocument(); + Element dataElement = XmlUtil.createElement(document, XmlMappingConstants.DATA_KEY, Optional.absent()); + Element modulesElement = XmlUtil.createElement(document, XmlMappingConstants.MODULES_KEY, + Optional.of(XmlMappingConstants.URN_OPENDAYLIGHT_PARAMS_XML_NS_YANG_CONTROLLER_CONFIG)); + dataElement.appendChild(modulesElement); + + Config configMapping = configFacade.getConfigMapping(); + + ConfigRegistry configRegistryClient = new ConfigRegistryJMXClient(ManagementFactory.getPlatformMBeanServer()); + for(Entry entry: configModules) { + String moduleNamespace = entry.getKey(); + ModuleIdentifier moduleId = entry.getValue(); + try { + ObjectName instanceON = configRegistryClient.lookupConfigBean(moduleId.getFactoryName(), + moduleId.getInstanceName()); + + LOG.debug("Found config module instance ObjectName: {}", instanceON); + + Element moduleElement = configMapping.moduleToXml(moduleNamespace, moduleId.getFactoryName(), + moduleId.getInstanceName(), instanceON, document); + modulesElement.appendChild(moduleElement); + } catch(InstanceNotFoundException e) { + LOG.warn("Error looking up config module: namespace {}, module name {}, instance {}", + moduleNamespace, moduleId.getFactoryName(), moduleId.getInstanceName(), e); + } + } + + if(LOG.isDebugEnabled()) { + LOG.debug("Pushing config xml: {}", XmlUtil.toString(dataElement)); + } + + ConfigExecution execution = new ConfigExecution(configMapping, XmlElement.fromDomElement(dataElement), + TestOption.testThenSet, EditStrategyType.recreate); + configFacade.executeConfigExecution(execution); + configFacade.commitSilentTransaction(); } /** * Recursively finds the services registered by the given bundle and the bundles using those services. - * User bundles that have an associated blueprint container are added to containerBundles. + * User bundles that have an associated blueprint container are added to containerBundles. In addition, + * if a registered service has an associated config system Module, as determined via the presence of + * certain service properties, the ModuleIdentifier is added to the configModules list. * * @param bundle the bundle to traverse * @param containerBundles the current set of bundles containing blueprint containers */ - private void findDependentContainersRecursively(Bundle bundle, Set containerBundles) { - if(containerBundles.contains(bundle)) { + private void findDependentContainersRecursively(Bundle bundle, Set containerBundles, + List> configModules) { + if(!containerBundles.add(bundle)) { + // Already seen this bundle... return; } - containerBundles.add(bundle); ServiceReference[] references = bundle.getRegisteredServices(); if (references != null) { for (ServiceReference reference : references) { + possiblyAddConfigModuleIdentifier(reference, configModules); + Bundle[] usingBundles = reference.getUsingBundles(); if(usingBundles != null) { for(Bundle usingBundle: usingBundles) { if(blueprintExtenderService.getContainer(usingBundle) != null) { - findDependentContainersRecursively(usingBundle, containerBundles); + findDependentContainersRecursively(usingBundle, containerBundles, configModules); } } } @@ -101,6 +245,48 @@ class BlueprintContainerRestartServiceImpl implements AutoCloseable, BlueprintCo } } + private void possiblyAddConfigModuleIdentifier(ServiceReference reference, + List> configModules) { + Object moduleNamespace = reference.getProperty(CONFIG_MODULE_NAMESPACE_PROP); + if(moduleNamespace == null) { + return; + } + + String moduleName = getRequiredConfigModuleProperty(CONFIG_MODULE_NAME_PROP, moduleNamespace, + reference); + String instanceName = getRequiredConfigModuleProperty(CONFIG_INSTANCE_NAME_PROP, moduleNamespace, + reference); + if(moduleName == null || instanceName == null) { + return; + } + + LOG.debug("Found service with config module: namespace {}, module name {}, instance {}", + moduleNamespace, moduleName, instanceName); + + configModules.add(new SimpleEntry<>(moduleNamespace.toString(), + new ModuleIdentifier(moduleName, instanceName))); + } + + @Nullable + private String getRequiredConfigModuleProperty(String propName, Object moduleNamespace, + ServiceReference reference) { + Object value = reference.getProperty(propName); + if(value == null) { + LOG.warn("OSGi service with {} property is missing property {} therefore the config module can't be restarted", + CONFIG_MODULE_NAMESPACE_PROP, propName); + return null; + } + + return value.toString(); + } + + private ServiceRegistration registerEventHandler(BundleContext bundleContext, EventHandler handler) { + Dictionary props = new Hashtable<>(); + props.put(org.osgi.service.event.EventConstants.EVENT_TOPIC, + new String[]{EventConstants.TOPIC_CREATED, EventConstants.TOPIC_FAILURE}); + return bundleContext.registerService(EventHandler.class.getName(), handler, props); + } + @Override public void close() { restartExecutor.shutdownNow();