Add slave/master end-to-end test
[netconf.git] / netconf / netconf-topology-singleton / src / main / java / org / opendaylight / netconf / topology / singleton / impl / NetconfTopologyContext.java
1 /*
2  * Copyright (c) 2016 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 package org.opendaylight.netconf.topology.singleton.impl;
10
11 import static org.opendaylight.netconf.topology.singleton.impl.utils.NetconfTopologyUtils.DEFAULT_SCHEMA_REPOSITORY;
12
13 import akka.actor.ActorRef;
14 import akka.cluster.Cluster;
15 import akka.dispatch.OnComplete;
16 import akka.pattern.Patterns;
17 import akka.util.Timeout;
18 import com.google.common.base.Preconditions;
19 import com.google.common.util.concurrent.Futures;
20 import com.google.common.util.concurrent.ListenableFuture;
21 import java.util.concurrent.atomic.AtomicBoolean;
22 import javax.annotation.Nonnull;
23 import org.opendaylight.controller.md.sal.dom.api.DOMMountPointService;
24 import org.opendaylight.mdsal.singleton.common.api.ClusterSingletonService;
25 import org.opendaylight.mdsal.singleton.common.api.ServiceGroupIdentifier;
26 import org.opendaylight.netconf.sal.connect.util.RemoteDeviceId;
27 import org.opendaylight.netconf.topology.singleton.api.RemoteDeviceConnector;
28 import org.opendaylight.netconf.topology.singleton.impl.actors.NetconfNodeActor;
29 import org.opendaylight.netconf.topology.singleton.impl.utils.NetconfTopologySetup;
30 import org.opendaylight.netconf.topology.singleton.impl.utils.NetconfTopologyUtils;
31 import org.opendaylight.netconf.topology.singleton.messages.RefreshSetupMasterActorData;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.NetconfNode;
33 import org.slf4j.Logger;
34 import org.slf4j.LoggerFactory;
35 import scala.concurrent.Future;
36
37 class NetconfTopologyContext implements ClusterSingletonService, AutoCloseable {
38
39     private static final Logger LOG = LoggerFactory.getLogger(NetconfTopologyContext.class);
40
41     private final ServiceGroupIdentifier serviceGroupIdent;
42     private final Timeout actorResponseWaitTime;
43     private final DOMMountPointService mountService;
44
45     private NetconfTopologySetup netconfTopologyDeviceSetup;
46     private RemoteDeviceId remoteDeviceId;
47     private RemoteDeviceConnector remoteDeviceConnector;
48     private NetconfNodeManager netconfNodeManager;
49     private ActorRef masterActorRef;
50     private final AtomicBoolean closed = new AtomicBoolean(false);
51     private final AtomicBoolean stopped = new AtomicBoolean(false);
52     private volatile boolean isMaster;
53
54     NetconfTopologyContext(final NetconfTopologySetup netconfTopologyDeviceSetup,
55                            final ServiceGroupIdentifier serviceGroupIdent,
56                            final Timeout actorResponseWaitTime, final DOMMountPointService mountService) {
57         this.netconfTopologyDeviceSetup = Preconditions.checkNotNull(netconfTopologyDeviceSetup);
58         this.serviceGroupIdent = serviceGroupIdent;
59         this.actorResponseWaitTime = actorResponseWaitTime;
60         this.mountService = mountService;
61
62         remoteDeviceId = NetconfTopologyUtils.createRemoteDeviceId(netconfTopologyDeviceSetup.getNode().getNodeId(),
63                 netconfTopologyDeviceSetup.getNode().getAugmentation(NetconfNode.class));
64
65         remoteDeviceConnector = new RemoteDeviceConnectorImpl(netconfTopologyDeviceSetup, remoteDeviceId);
66
67         netconfNodeManager = createNodeDeviceManager();
68     }
69
70     @Override
71     public void instantiateServiceInstance() {
72         LOG.info("Master was selected: {}", remoteDeviceId.getHost().getIpAddress());
73
74         isMaster = true;
75
76         // master should not listen on netconf-node operational datastore
77         if (netconfNodeManager != null) {
78             netconfNodeManager.close();
79             netconfNodeManager = null;
80         }
81
82         if (!closed.get()) {
83             final String masterAddress =
84                     Cluster.get(netconfTopologyDeviceSetup.getActorSystem()).selfAddress().toString();
85             masterActorRef = netconfTopologyDeviceSetup.getActorSystem().actorOf(NetconfNodeActor.props(
86                     netconfTopologyDeviceSetup, remoteDeviceId, DEFAULT_SCHEMA_REPOSITORY, DEFAULT_SCHEMA_REPOSITORY,
87                     actorResponseWaitTime, mountService),
88                     NetconfTopologyUtils.createMasterActorName(remoteDeviceId.getName(), masterAddress));
89
90             remoteDeviceConnector.startRemoteDeviceConnection(newMasterSalFacade());
91         }
92
93     }
94
95     // called when master is down/changed to slave
96     @Override
97     public ListenableFuture<Void> closeServiceInstance() {
98
99         if (!closed.get()) {
100             // in case that master changes role to slave, new NodeDeviceManager must be created and listener registered
101             netconfNodeManager = createNodeDeviceManager();
102         }
103         stopDeviceConnectorAndActor();
104
105         return Futures.immediateFuture(null);
106     }
107
108     @Override
109     public ServiceGroupIdentifier getIdentifier() {
110         return serviceGroupIdent;
111     }
112
113     private NetconfNodeManager createNodeDeviceManager() {
114         final NetconfNodeManager ndm =
115                 new NetconfNodeManager(netconfTopologyDeviceSetup, remoteDeviceId, actorResponseWaitTime, mountService);
116         ndm.registerDataTreeChangeListener(netconfTopologyDeviceSetup.getTopologyId(),
117                 netconfTopologyDeviceSetup.getNode().getKey());
118
119         return ndm;
120     }
121
122     @Override
123     public void close() throws Exception {
124         if (!closed.compareAndSet(false, true)) {
125             return;
126         }
127
128         if (netconfNodeManager != null) {
129             netconfNodeManager.close();
130         }
131         stopDeviceConnectorAndActor();
132
133     }
134
135     /**
136      * Refresh, if configuration data was changed.
137      * @param setup new setup
138      */
139     void refresh(@Nonnull final NetconfTopologySetup setup) {
140         netconfTopologyDeviceSetup = Preconditions.checkNotNull(setup);
141         remoteDeviceId = NetconfTopologyUtils.createRemoteDeviceId(netconfTopologyDeviceSetup.getNode().getNodeId(),
142                 netconfTopologyDeviceSetup.getNode().getAugmentation(NetconfNode.class));
143
144         if (isMaster) {
145             remoteDeviceConnector.stopRemoteDeviceConnection();
146         }
147         if (!isMaster) {
148             netconfNodeManager.refreshDevice(netconfTopologyDeviceSetup, remoteDeviceId);
149         }
150         remoteDeviceConnector = new RemoteDeviceConnectorImpl(netconfTopologyDeviceSetup, remoteDeviceId);
151
152         if (isMaster) {
153             final Future<Object> future = Patterns.ask(masterActorRef, new RefreshSetupMasterActorData(
154                     netconfTopologyDeviceSetup, remoteDeviceId), actorResponseWaitTime);
155
156             future.onComplete(new OnComplete<Object>() {
157                 @Override
158                 public void onComplete(final Throwable failure, final Object success) throws Throwable {
159                     if (failure != null) {
160                         LOG.error("Failed to refresh master actor data: {}", failure);
161                         return;
162                     }
163                     remoteDeviceConnector.startRemoteDeviceConnection(newMasterSalFacade());
164                 }
165             }, netconfTopologyDeviceSetup.getActorSystem().dispatcher());
166         }
167     }
168
169     private void stopDeviceConnectorAndActor() {
170         if (!stopped.compareAndSet(false, true)) {
171             return;
172         }
173         if (remoteDeviceConnector != null) {
174             remoteDeviceConnector.stopRemoteDeviceConnection();
175         }
176
177         if (masterActorRef != null) {
178             netconfTopologyDeviceSetup.getActorSystem().stop(masterActorRef);
179             masterActorRef = null;
180         }
181     }
182
183     protected MasterSalFacade newMasterSalFacade() {
184         return new MasterSalFacade(remoteDeviceId, netconfTopologyDeviceSetup.getActorSystem(), masterActorRef,
185                 actorResponseWaitTime, mountService, netconfTopologyDeviceSetup.getDataBroker());
186     }
187 }