2 * Copyright (c) 2015 Cisco Systems, Inc. and others. All rights reserved.
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
9 package org.opendaylight.netconf.topology.example;
11 import akka.actor.ActorRef;
12 import akka.actor.ActorSystem;
13 import akka.actor.TypedActor;
14 import com.google.common.util.concurrent.FutureCallback;
15 import com.google.common.util.concurrent.Futures;
16 import com.google.common.util.concurrent.ListenableFuture;
17 import java.util.HashMap;
19 import javax.annotation.Nonnull;
20 import javax.annotation.Nullable;
21 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
22 import org.opendaylight.netconf.topology.NodeManager;
23 import org.opendaylight.netconf.topology.NodeManagerCallback.NodeManagerCallbackFactory;
24 import org.opendaylight.netconf.topology.TopologyManagerCallback;
25 import org.opendaylight.netconf.topology.util.BaseNodeManager.BaseNodeManagerBuilder;
26 import org.opendaylight.netconf.topology.util.NodeWriter;
27 import org.opendaylight.netconf.topology.util.NoopRoleChangeStrategy;
28 import org.opendaylight.netconf.topology.util.SalNodeWriter;
29 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
30 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
31 import org.slf4j.Logger;
32 import org.slf4j.LoggerFactory;
34 public class ExampleTopologyManagerCallback implements TopologyManagerCallback {
36 private static final Logger LOG = LoggerFactory.getLogger(ExampleTopologyManagerCallback.class);
38 private final DataBroker dataBroker;
39 private final ActorSystem actorSystem;
40 private boolean isMaster;
42 private final String topologyId;
43 private final NodeWriter naSalNodeWriter;
44 private final Map<NodeId, NodeManager> nodes = new HashMap<>();
45 private final NodeManagerCallbackFactory nodeHandlerFactory;
47 public ExampleTopologyManagerCallback(final ActorSystem actorSystem,
48 final DataBroker dataBroker,
49 final String topologyId,
50 final NodeManagerCallbackFactory nodeHandlerFactory) {
51 this(actorSystem, dataBroker, topologyId, nodeHandlerFactory, new SalNodeWriter(dataBroker, topologyId));
54 public ExampleTopologyManagerCallback(final ActorSystem actorSystem,
55 final DataBroker dataBroker,
56 final String topologyId,
57 final NodeManagerCallbackFactory nodeHandlerFactory,
58 final NodeWriter naSalNodeWriter) {
59 this(actorSystem, dataBroker, topologyId, nodeHandlerFactory, naSalNodeWriter, false);
63 public ExampleTopologyManagerCallback(final ActorSystem actorSystem,
64 final DataBroker dataBroker,
65 final String topologyId,
66 final NodeManagerCallbackFactory nodeHandlerFactory,
67 final NodeWriter naSalNodeWriter,
69 this.dataBroker = dataBroker;
70 this.actorSystem = actorSystem;
71 this.topologyId = topologyId;
72 this.nodeHandlerFactory = nodeHandlerFactory;
73 this.naSalNodeWriter = naSalNodeWriter;
75 this.isMaster = isMaster;
79 public ListenableFuture<Node> onNodeCreated(NodeId nodeId, Node node) {
80 // Init node admin and a writer for it
82 // TODO let end user code notify the baseNodeManager about state changes and handle them here on topology level
83 final NodeManager naBaseNodeManager =
84 createNodeManager(nodeId);
86 nodes.put(nodeId, naBaseNodeManager);
88 // Set initial state ? in every peer or just master ? TODO
90 naSalNodeWriter.init(nodeId, naBaseNodeManager.getInitialState(nodeId, node));
93 // trigger connect on this node
94 return naBaseNodeManager.onNodeCreated(nodeId, node);
98 public ListenableFuture<Node> onNodeUpdated(final NodeId nodeId, final Node node) {
100 naSalNodeWriter.init(nodeId, nodes.get(nodeId).getInitialState(nodeId, node));
102 // Trigger onNodeUpdated only on this node
103 return nodes.get(nodeId).onNodeUpdated(nodeId, node);
107 public ListenableFuture<Void> onNodeDeleted(final NodeId nodeId) {
108 // Trigger delete only on this node
109 final ListenableFuture<Void> future = nodes.get(nodeId).onNodeDeleted(nodeId);
110 Futures.addCallback(future, new FutureCallback<Void>() {
112 public void onSuccess(Void result) {
113 // remove proxy from node list and stop the actor
114 final NodeManager remove = nodes.remove(nodeId);
115 TypedActor.get(actorSystem).stop(remove);
119 public void onFailure(Throwable t) {
120 // NOOP will be handled on higher level
128 public ListenableFuture<Node> getCurrentStatusForNode(@Nonnull NodeId nodeId) {
129 return nodes.get(nodeId).getCurrentStatusForNode(nodeId);
133 public void onRoleChanged(RoleChangeDTO roleChangeDTO) {
134 isMaster = roleChangeDTO.isOwner();
135 // our post-election logic
138 private NodeManager createNodeManager(NodeId nodeId) {
139 return new BaseNodeManagerBuilder().setNodeId(nodeId.getValue())
140 .setActorContext(TypedActor.context())
141 .setDelegateFactory(nodeHandlerFactory)
142 .setRoleChangeStrategy(new NoopRoleChangeStrategy())
143 .setTopologyId(topologyId)
148 public void onReceive(Object o, ActorRef actorRef) {
154 public Node getInitialState(@Nonnull NodeId nodeId, @Nonnull Node configNode) {
155 return nodes.get(nodeId).getInitialState(nodeId, configNode);
160 public Node getFailedState(@Nonnull NodeId nodeId, @Nullable Node configNode) {
161 return nodes.get(nodeId).getFailedState(nodeId, configNode);