src/main/java/net/onrc/onos/core/topology/web/TopologySwitchesResource.java

package net.onrc.onos.core.topology.web;

import java.io.IOException;

import net.onrc.onos.core.topology.ITopologyService;
import net.onrc.onos.core.topology.Topology;
import net.onrc.onos.core.topology.serializers.PortSerializer;
import net.onrc.onos.core.topology.serializers.SwitchSerializer;

import org.codehaus.jackson.Version;
import org.codehaus.jackson.map.ObjectMapper;
import org.codehaus.jackson.map.module.SimpleModule;
import org.restlet.resource.Get;
import org.restlet.resource.ServerResource;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class TopologySwitchesResource extends ServerResource {

    private static final Logger log = LoggerFactory.getLogger(TopologySwitchesResource.class);

    @Get("json")
    public String retrieve() {
        ITopologyService topologyService = (ITopologyService) getContext().getAttributes().
                get(ITopologyService.class.getCanonicalName());

        Topology topology = topologyService.getTopology();

        ObjectMapper mapper = new ObjectMapper();
        SimpleModule module = new SimpleModule("module", new Version(1, 0, 0, null));
        module.addSerializer(new SwitchSerializer());
        module.addSerializer(new PortSerializer());
        mapper.registerModule(module);

        try {
            topology.acquireReadLock();
            return mapper.writeValueAsString(topology.getSwitches());
        } catch (IOException e) {
            log.error("Error writing switch list to JSON", e);
            return "";
        } finally {
            topology.releaseReadLock();
        }
    }
}