| package net.floodlightcontroller.linkdiscovery.internal; |
| |
| import java.util.ArrayList; |
| import java.util.List; |
| |
| import net.floodlightcontroller.core.INetMapTopologyObjects.IPortObject; |
| import net.floodlightcontroller.core.INetMapTopologyObjects.ISwitchObject; |
| import net.floodlightcontroller.linkdiscovery.ILinkStorage; |
| import net.floodlightcontroller.linkdiscovery.LinkInfo; |
| import net.floodlightcontroller.routing.Link; |
| import net.onrc.onos.util.GraphDBConnection; |
| import net.onrc.onos.util.GraphDBConnection.Transaction; |
| |
| import org.openflow.util.HexString; |
| import org.slf4j.Logger; |
| import org.slf4j.LoggerFactory; |
| |
| import com.thinkaurelius.titan.core.TitanException; |
| import com.tinkerpop.blueprints.Direction; |
| import com.tinkerpop.blueprints.Edge; |
| import com.tinkerpop.blueprints.Vertex; |
| import com.tinkerpop.gremlin.java.GremlinPipeline; |
| import com.tinkerpop.pipes.PipeFunction; |
| import com.tinkerpop.pipes.transform.PathPipe; |
| |
| public class LinkStorageImpl implements ILinkStorage { |
| |
| protected static Logger log = LoggerFactory.getLogger(LinkStorageImpl.class); |
| protected String conf; |
| |
| @Override |
| public void update(Link link, DM_OPERATION op) { |
| update(link, (LinkInfo)null, op); |
| } |
| |
| @Override |
| public void update(List<Link> links, DM_OPERATION op) { |
| for (Link lt: links) { |
| update(lt, (LinkInfo)null, op); |
| } |
| } |
| |
| @Override |
| public void update(Link link, LinkInfo linkinfo, DM_OPERATION op) { |
| switch (op) { |
| case UPDATE: |
| case CREATE: |
| case INSERT: |
| updateLink(link, linkinfo, op); |
| break; |
| case DELETE: |
| deleteLink(link); |
| break; |
| } |
| } |
| |
| public void updateLink(Link lt, LinkInfo linkinfo, DM_OPERATION op) { |
| GraphDBConnection conn = GraphDBConnection.getInstance(this.conf); |
| IPortObject vportSrc = null, vportDst = null; |
| |
| log.trace("updateLink(): op {} {} {}", new Object[]{op, lt, linkinfo}); |
| |
| try { |
| // get source port vertex |
| String dpid = HexString.toHexString(lt.getSrc()); |
| short port = lt.getSrcPort(); |
| vportSrc = conn.utils().searchPort(conn, dpid, port); |
| |
| // get dest port vertex |
| dpid = HexString.toHexString(lt.getDst()); |
| port = lt.getDstPort(); |
| vportDst = conn.utils().searchPort(conn, dpid, port); |
| |
| if (vportSrc != null && vportDst != null) { |
| // check if the link exists |
| |
| Iterable<IPortObject> currPorts = vportSrc.getLinkedPorts(); |
| List<IPortObject> currLinks = new ArrayList<IPortObject>(); |
| for (IPortObject V : currPorts) { |
| currLinks.add(V); |
| } |
| |
| if (currLinks.contains(vportDst)) { |
| // TODO: update linkinfo |
| if (op.equals(DM_OPERATION.INSERT) || op.equals(DM_OPERATION.CREATE)) { |
| log.debug("addOrUpdateLink(): failed link exists {} {} src {} dst {}", |
| new Object[]{op, lt, vportSrc, vportDst}); |
| } |
| } else { |
| vportSrc.setLinkPort(vportDst); |
| |
| conn.endTx(Transaction.COMMIT); |
| log.debug("updateLink(): link added {} {} src {} dst {}", new Object[]{op, lt, vportSrc, vportDst}); |
| } |
| } else { |
| log.error("updateLink(): failed invalid vertices {} {} src {} dst {}", new Object[]{op, lt, vportSrc, vportDst}); |
| conn.endTx(Transaction.ROLLBACK); |
| } |
| } catch (TitanException e) { |
| /* |
| * retry till we succeed? |
| */ |
| e.printStackTrace(); |
| log.error("updateLink(): titan exception {} {} {}", new Object[]{op, lt, e.toString()}); |
| } |
| } |
| |
| @Override |
| public void deleteLinks(List<Link> links) { |
| |
| for (Link lt : links) { |
| deleteLink(lt); |
| } |
| } |
| |
| |
| @Override |
| public void deleteLink(Link lt) { |
| GraphDBConnection conn = GraphDBConnection.getInstance(this.conf); |
| IPortObject vportSrc = null, vportDst = null; |
| int count = 0; |
| |
| log.debug("deleteLink(): {}", lt); |
| |
| try { |
| // get source port vertex |
| String dpid = HexString.toHexString(lt.getSrc()); |
| short port = lt.getSrcPort(); |
| vportSrc = conn.utils().searchPort(conn, dpid, port); |
| |
| // get dst port vertex |
| dpid = HexString.toHexString(lt.getDst()); |
| port = lt.getDstPort(); |
| vportDst = conn.utils().searchPort(conn, dpid, port); |
| // FIXME: This needs to remove all edges |
| |
| if (vportSrc != null && vportDst != null) { |
| |
| /* for (Edge e : vportSrc.asVertex().getEdges(Direction.OUT)) { |
| log.debug("deleteLink(): {} in {} out {}", |
| new Object[]{e.getLabel(), e.getVertex(Direction.IN), e.getVertex(Direction.OUT)}); |
| if (e.getLabel().equals("link") && e.getVertex(Direction.IN).equals(vportDst)) { |
| graph.removeEdge(e); |
| count++; |
| } |
| }*/ |
| vportSrc.removeLink(vportDst); |
| |
| conn.endTx(Transaction.COMMIT); |
| log.debug("deleteLink(): deleted edges src {} dst {}", new Object[]{ |
| lt, vportSrc, vportDst}); |
| |
| } else { |
| log.error("deleteLink(): failed invalid vertices {} src {} dst {}", new Object[]{lt, vportSrc, vportDst}); |
| conn.endTx(Transaction.ROLLBACK); |
| } |
| |
| } catch (TitanException e) { |
| /* |
| * retry till we succeed? |
| */ |
| log.error("deleteLink(): titan exception {} {}", new Object[]{lt, e.toString()}); |
| conn.endTx(Transaction.ROLLBACK); |
| e.printStackTrace(); |
| } |
| } |
| |
| // TODO: Fix me |
| @Override |
| public List<Link> getLinks(Long dpid, short port) { |
| GraphDBConnection conn = GraphDBConnection.getInstance(this.conf); |
| IPortObject vportSrc; |
| |
| List<Link> links = new ArrayList<Link>(); |
| |
| // BEGIN: Trial code |
| // author: Naoki Shiota |
| vportSrc = conn.utils().searchPort(conn, HexString.toHexString(dpid), port); |
| if (vportSrc != null) { |
| |
| for (Edge e : vportSrc.asVertex().getEdges(Direction.IN)) { |
| if(e.getLabel().equals("link")) { |
| Vertex v = e.getVertex(Direction.OUT); |
| short dst_port = v.getProperty("number"); |
| for(Edge e2 : v.getEdges(Direction.IN)) { |
| if(e2.getLabel().equals("on")) { |
| Vertex v2 = e2.getVertex(Direction.OUT); |
| long dst_dpid = HexString.toLong((String) v2.getProperty("dpid")); |
| |
| Link lt = new Link(dpid, port, dst_dpid, dst_port); |
| links.add(lt); |
| } |
| } |
| } |
| } |
| } |
| // END: Trial code |
| |
| return links; |
| } |
| |
| @Override |
| public void init(String conf) { |
| //TODO extract the DB location from properties |
| |
| this.conf = conf; |
| |
| } |
| |
| // TODO: Fix me |
| @Override |
| public void deleteLinksOnPort(Long dpid, short port) { |
| // BEGIN: Trial code |
| // author: Naoki Shiota |
| List<Link> linksToDelete = getLinks(dpid,port); |
| |
| for(Link l : linksToDelete) { |
| deleteLink(l); |
| } |
| // END: Trial code |
| } |
| |
| // TODO: Fix me |
| @Override |
| public List<Link> getLinks(String dpid) { |
| GraphDBConnection conn = GraphDBConnection.getInstance(this.conf); |
| ISwitchObject vswitch; |
| List<Link> links = new ArrayList<Link>(); |
| |
| // BEGIN: Trial code |
| // author: Naoki Shiota |
| vswitch = conn.utils().searchSwitch(conn, dpid); |
| |
| for(IPortObject vportSrc : vswitch.getPorts()) { |
| // array concatenation may be heavy... |
| List<Link> sublinks = getLinks(HexString.toLong(dpid), vportSrc.getNumber()); |
| links.addAll(sublinks); |
| } |
| // END: Trial code |
| |
| return links; |
| } |
| |
| public List<Link> getActiveLinks() { |
| |
| GraphDBConnection conn = GraphDBConnection.getInstance(this.conf); |
| |
| Iterable<ISwitchObject> switches = conn.utils().getActiveSwitches(conn); |
| |
| List<Link> links = new ArrayList<Link>(); |
| for (ISwitchObject sw : switches) { |
| GremlinPipeline<Vertex, Link> pipe = new GremlinPipeline<Vertex, Link>(); |
| ExtractLink extractor = new ExtractLink(); |
| |
| pipe.start(sw.asVertex()); |
| pipe.enablePath(true); |
| pipe.out("on").out("link").in("on").path().step(extractor); |
| |
| while (pipe.hasNext() ) { |
| Link l = pipe.next(); |
| links.add(l); |
| } |
| |
| } |
| return links; |
| } |
| |
| static class ExtractLink implements PipeFunction<PathPipe<Vertex>, Link> { |
| |
| @Override |
| public Link compute(PathPipe<Vertex> pipe ) { |
| // TODO Auto-generated method stub |
| long s_dpid = 0; |
| long d_dpid = 0; |
| short s_port = 0; |
| short d_port = 0; |
| List<Vertex> V = new ArrayList<Vertex>(); |
| V = pipe.next(); |
| Vertex src_sw = V.get(0); |
| Vertex dest_sw = V.get(3); |
| Vertex src_port = V.get(1); |
| Vertex dest_port = V.get(2); |
| s_dpid = HexString.toLong((String) src_sw.getProperty("dpid")); |
| d_dpid = HexString.toLong((String) dest_sw.getProperty("dpid")); |
| s_port = (Short) src_port.getProperty("number"); |
| d_port = (Short) dest_port.getProperty("number"); |
| |
| Link l = new Link(s_dpid,s_port,d_dpid,d_port); |
| |
| return l; |
| } |
| } |
| |
| public void finalize() { |
| close(); |
| } |
| |
| @Override |
| public void close() { |
| // TODO Auto-generated method stub |
| // graph.shutdown(); |
| } |
| |
| |
| } |