opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/config/yang/config/actor_system_provider/impl/ActorSystemProviderImpl.java

   1 /*
   2  * Copyright (c) 2015 Brocade Communications 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 package org.opendaylight.controller.config.yang.config.actor_system_provider.impl;
   9
  10 import akka.actor.ActorSystem;
  11 import akka.actor.Props;
  12 import akka.japi.Effect;
  13 import akka.osgi.BundleDelegatingClassLoader;
  14 import com.typesafe.config.Config;
  15 import com.typesafe.config.ConfigFactory;
  16 import java.util.concurrent.TimeUnit;
  17 import org.opendaylight.controller.cluster.ActorSystemProvider;
  18 import org.opendaylight.controller.cluster.ActorSystemProviderListener;
  19 import org.opendaylight.controller.cluster.common.actor.AkkaConfigurationReader;
  20 import org.opendaylight.controller.cluster.common.actor.FileAkkaConfigurationReader;
  21 import org.opendaylight.controller.cluster.common.actor.QuarantinedMonitorActor;
  22 import org.opendaylight.controller.cluster.datastore.TerminationMonitor;
  23 import org.opendaylight.yangtools.concepts.ListenerRegistration;
  24 import org.opendaylight.yangtools.util.ListenerRegistry;
  25 import org.osgi.framework.Bundle;
  26 import org.osgi.framework.BundleContext;
  27 import org.slf4j.Logger;
  28 import org.slf4j.LoggerFactory;
  29 import scala.concurrent.duration.Duration;
  30
  31 public class ActorSystemProviderImpl implements ActorSystemProvider, AutoCloseable {
  32     private static final String ACTOR_SYSTEM_NAME = "opendaylight-cluster-data";
  33     private static final String CONFIGURATION_NAME = "odl-cluster-data";
  34     static final Logger LOG = LoggerFactory.getLogger(ActorSystemProviderImpl.class);
  35     private final ActorSystem actorSystem;
  36     private final ListenerRegistry<ActorSystemProviderListener> listeners = new ListenerRegistry<>();
  37
  38     public ActorSystemProviderImpl(final BundleContext bundleContext) {
  39         LOG.info("Creating new ActorSystem");
  40
  41         final Bundle bundle = bundleContext.getBundle();
  42
  43         final BundleDelegatingClassLoader classLoader = new BundleDelegatingClassLoader(bundle, Thread.currentThread()
  44                 .getContextClassLoader());
  45
  46         final AkkaConfigurationReader configurationReader = new FileAkkaConfigurationReader();
  47         final Config akkaConfig = ConfigFactory.load(configurationReader.read()).getConfig(CONFIGURATION_NAME);
  48
  49         actorSystem = ActorSystem.create(ACTOR_SYSTEM_NAME, akkaConfig, classLoader);
  50
  51         actorSystem.actorOf(Props.create(TerminationMonitor.class), TerminationMonitor.ADDRESS);
  52
  53         actorSystem.actorOf(QuarantinedMonitorActor.props(new Effect() {
  54
  55             @Override
  56             public void apply() throws Exception {
  57                 // restart the entire karaf container
  58                 LOG.warn("Restarting karaf container");
  59                 System.setProperty("karaf.restart", "true");
  60                 bundleContext.getBundle(0).stop();
  61             }
  62         }), QuarantinedMonitorActor.ADDRESS);
  63
  64     }
  65
  66     @Override
  67     public ActorSystem getActorSystem() {
  68         return actorSystem;
  69     }
  70
  71     @Override
  72     public ListenerRegistration<ActorSystemProviderListener> registerActorSystemProviderListener(
  73             ActorSystemProviderListener listener) {
  74         return listeners.register(listener);
  75     }
  76
  77     @Override
  78     public void close() {
  79         LOG.info("Shutting down ActorSystem");
  80
  81         actorSystem.shutdown();
  82         try {
  83             actorSystem.awaitTermination(Duration.create(10, TimeUnit.SECONDS));
  84         } catch (Exception e) {
  85             LOG.warn("Error awaiting actor termination", e);
  86         }
  87     }
  88 }