Integrate the new notification framework with the new NetworkGraph
implementation.
Change-Id: I93033a5747c216dd336f68ac25f9c5664dd0f688
diff --git a/src/main/java/net/onrc/onos/ofcontroller/networkgraph/NetworkGraphModule.java b/src/main/java/net/onrc/onos/ofcontroller/networkgraph/NetworkGraphModule.java
index 7315f1c..3edf5c2 100644
--- a/src/main/java/net/onrc/onos/ofcontroller/networkgraph/NetworkGraphModule.java
+++ b/src/main/java/net/onrc/onos/ofcontroller/networkgraph/NetworkGraphModule.java
@@ -11,6 +11,7 @@
import net.floodlightcontroller.core.module.IFloodlightModule;
import net.floodlightcontroller.core.module.IFloodlightService;
import net.floodlightcontroller.restserver.IRestApiService;
+import net.onrc.onos.datagrid.IDatagridService;
import net.onrc.onos.ofcontroller.networkgraph.web.NetworkGraphWebRoutable;
public class NetworkGraphModule implements IFloodlightModule, INetworkGraphService {
@@ -20,6 +21,7 @@
private NetworkGraphImpl networkGraph;
//private NetworkGraphDatastore southboundNetworkGraph;
+ private IDatagridService datagridService;
private IRestApiService restApi;
@@ -44,6 +46,7 @@
public Collection<Class<? extends IFloodlightService>> getModuleDependencies() {
List<Class<? extends IFloodlightService>> dependencies =
new ArrayList<Class<? extends IFloodlightService>>();
+ dependencies.add(IDatagridService.class);
dependencies.add(IRestApiService.class);
return dependencies;
}
@@ -52,6 +55,7 @@
public void init(FloodlightModuleContext context)
throws FloodlightModuleException {
restApi = context.getServiceImpl(IRestApiService.class);
+ datagridService = context.getServiceImpl(IDatagridService.class);
networkGraph = new NetworkGraphImpl();
//southboundNetworkGraph = new NetworkGraphDatastore(networkGraph);
@@ -60,6 +64,7 @@
@Override
public void startUp(FloodlightModuleContext context) {
restApi.addRestletRoutable(new NetworkGraphWebRoutable());
+ networkGraph.startup(datagridService);
}
@Override