src/main/java/net/onrc/onos/core/topology/NetworkGraphModule.java

package net.onrc.onos.core.topology;

import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.CopyOnWriteArrayList;

import net.floodlightcontroller.core.module.FloodlightModuleContext;
import net.floodlightcontroller.core.module.FloodlightModuleException;
import net.floodlightcontroller.core.module.IFloodlightModule;
import net.floodlightcontroller.core.module.IFloodlightService;
import net.floodlightcontroller.restserver.IRestApiService;
import net.onrc.onos.core.datagrid.IDatagridService;
import net.onrc.onos.core.topology.web.NetworkGraphWebRoutable;
import net.onrc.onos.registry.controller.IControllerRegistryService;

public class NetworkGraphModule implements IFloodlightModule, INetworkGraphService {

	// This is initialized as a module for now

	private TopologyManager topologyManager;
	//private NetworkGraphDatastore southboundNetworkGraph;
	private IDatagridService datagridService;
	private IControllerRegistryService registryService;

	private CopyOnWriteArrayList<INetworkGraphListener> networkGraphListeners;

	private IRestApiService restApi;

	@Override
	public Collection<Class<? extends IFloodlightService>> getModuleServices() {
		List<Class<? extends IFloodlightService>> services =
				new ArrayList<Class<? extends IFloodlightService>>();
		services.add(INetworkGraphService.class);
		return services;
	}

	@Override
	public Map<Class<? extends IFloodlightService>, IFloodlightService>
			getServiceImpls() {
		Map<Class<? extends IFloodlightService>, IFloodlightService> impls =
				new HashMap<Class<? extends IFloodlightService>, IFloodlightService>();
		impls.put(INetworkGraphService.class, this);
		return impls;
	}

	@Override
	public Collection<Class<? extends IFloodlightService>> getModuleDependencies() {
		List<Class<? extends IFloodlightService>> dependencies =
				new ArrayList<Class<? extends IFloodlightService>>();
		dependencies.add(IDatagridService.class);
		dependencies.add(IRestApiService.class);
		dependencies.add(IControllerRegistryService.class);
		return dependencies;
	}

	@Override
	public void init(FloodlightModuleContext context)
			throws FloodlightModuleException {
		restApi = context.getServiceImpl(IRestApiService.class);
		datagridService = context.getServiceImpl(IDatagridService.class);
		registryService = context.getServiceImpl(IControllerRegistryService.class);

		networkGraphListeners = new CopyOnWriteArrayList<>();
		topologyManager = new TopologyManager(registryService, networkGraphListeners);
		//southboundNetworkGraph = new NetworkGraphDatastore(networkGraph);
	}

	@Override
	public void startUp(FloodlightModuleContext context) {
		restApi.addRestletRoutable(new NetworkGraphWebRoutable());
		topologyManager.startup(datagridService);
	}

	@Override
	public NetworkGraph getNetworkGraph() {
		return topologyManager.getNetworkGraph();
	}

	@Override
	public NetworkGraphDiscoveryInterface getNetworkGraphDiscoveryInterface() {
		return topologyManager;
	}

	@Override
	public void registerNetworkGraphListener(INetworkGraphListener listener) {
	    networkGraphListeners.addIfAbsent(listener);
	}

	@Override
	public void deregisterNetworkGraphListener(INetworkGraphListener listener) {
	    networkGraphListeners.remove(listener);
	}

}