Toshio Koide | ebdbb62 | 2014-02-12 20:28:38 -0800 | [diff] [blame] | 1 | package net.onrc.onos.intent; |
| 2 | |
Toshio Koide | 0c9106d | 2014-02-19 15:26:38 -0800 | [diff] [blame] | 3 | import net.onrc.onos.ofcontroller.networkgraph.NetworkGraph; |
Pavlin Radoslavov | 6397a7f | 2014-02-18 14:56:52 -0800 | [diff] [blame] | 4 | import net.onrc.onos.ofcontroller.networkgraph.NetworkGraphImpl; |
Toshio Koide | ebdbb62 | 2014-02-12 20:28:38 -0800 | [diff] [blame] | 5 | import net.onrc.onos.ofcontroller.networkgraph.Link; |
| 6 | import net.onrc.onos.ofcontroller.networkgraph.LinkImpl; |
Toshio Koide | 0c9106d | 2014-02-19 15:26:38 -0800 | [diff] [blame] | 7 | import net.onrc.onos.ofcontroller.networkgraph.Port; |
Toshio Koide | ebdbb62 | 2014-02-12 20:28:38 -0800 | [diff] [blame] | 8 | import net.onrc.onos.ofcontroller.networkgraph.Switch; |
| 9 | import net.onrc.onos.ofcontroller.networkgraph.SwitchImpl; |
| 10 | |
Toshio Koide | 066506e | 2014-02-20 19:52:09 -0800 | [diff] [blame^] | 11 | /** |
| 12 | * @author Toshio Koide (t-koide@onlab.us) |
| 13 | */ |
Pavlin Radoslavov | 6397a7f | 2014-02-18 14:56:52 -0800 | [diff] [blame] | 14 | public class MockNetworkGraph extends NetworkGraphImpl { |
Toshio Koide | 0c9106d | 2014-02-19 15:26:38 -0800 | [diff] [blame] | 15 | class DetachableLinkImpl extends LinkImpl { |
| 16 | public DetachableLinkImpl(NetworkGraph graph, Port srcPort, Port dstPort) { |
| 17 | super(graph, srcPort, dstPort); |
| 18 | } |
| 19 | |
| 20 | public void detachFromGraph() { |
| 21 | unsetFromPorts(); |
Toshio Koide | 066506e | 2014-02-20 19:52:09 -0800 | [diff] [blame^] | 22 | } |
Toshio Koide | 0c9106d | 2014-02-19 15:26:38 -0800 | [diff] [blame] | 23 | } |
Toshio Koide | ebdbb62 | 2014-02-12 20:28:38 -0800 | [diff] [blame] | 24 | public Switch addSwitch(Long switchId) { |
| 25 | SwitchImpl sw = new SwitchImpl(this, switchId); |
Pavlin Radoslavov | 6d224ee | 2014-02-18 16:43:15 -0800 | [diff] [blame] | 26 | this.putSwitch(sw); |
Toshio Koide | ebdbb62 | 2014-02-12 20:28:38 -0800 | [diff] [blame] | 27 | return sw; |
| 28 | |
| 29 | } |
| 30 | |
| 31 | public Link addLink(Long srcDpid, Long srcPortNo, Long dstDpid, Long dstPortNo) { |
Toshio Koide | 0c9106d | 2014-02-19 15:26:38 -0800 | [diff] [blame] | 32 | return new DetachableLinkImpl( |
Toshio Koide | ebdbb62 | 2014-02-12 20:28:38 -0800 | [diff] [blame] | 33 | this, |
| 34 | getSwitch(srcDpid).getPort(srcPortNo), |
| 35 | getSwitch(dstDpid).getPort(dstPortNo)); |
| 36 | } |
| 37 | |
| 38 | public Link[] addBidirectionalLinks(Long srcDpid, Long srcPortNo, Long dstDpid, Long dstPortNo) { |
| 39 | Link[] links = new Link[2]; |
| 40 | links[0] = addLink(srcDpid, srcPortNo, dstDpid, dstPortNo); |
| 41 | links[1] = addLink(dstDpid, dstPortNo, srcDpid, srcPortNo); |
| 42 | |
| 43 | return links; |
| 44 | } |
Toshio Koide | 066506e | 2014-02-20 19:52:09 -0800 | [diff] [blame^] | 45 | |
Toshio Koide | ebdbb62 | 2014-02-12 20:28:38 -0800 | [diff] [blame] | 46 | public void createSampleTopology() { |
| 47 | // add 10 switches (24 ports switch) |
| 48 | for (Long dpid=1L; dpid<10L; dpid++) { |
| 49 | SwitchImpl sw = (SwitchImpl) addSwitch(dpid); |
| 50 | for (Long j=1L; j<=24L; j++) { |
| 51 | sw.addPort(j); |
| 52 | } |
| 53 | } |
| 54 | |
| 55 | // add loop path |
| 56 | addBidirectionalLinks(1L, 1L, 2L, 2L); |
| 57 | addBidirectionalLinks(2L, 1L, 3L, 2L); |
| 58 | addBidirectionalLinks(3L, 1L, 4L, 2L); |
| 59 | addBidirectionalLinks(4L, 1L, 5L, 2L); |
| 60 | addBidirectionalLinks(5L, 1L, 6L, 2L); |
| 61 | addBidirectionalLinks(6L, 1L, 7L, 2L); |
| 62 | addBidirectionalLinks(7L, 1L, 8L, 2L); |
| 63 | addBidirectionalLinks(8L, 1L, 9L, 2L); |
| 64 | addBidirectionalLinks(9L, 1L, 1L, 2L); |
| 65 | |
| 66 | // add other links |
| 67 | addBidirectionalLinks(1L, 3L, 5L, 3L); |
| 68 | addBidirectionalLinks(2L, 4L, 5L, 4L); |
| 69 | addBidirectionalLinks(2L, 5L, 7L, 5L); |
| 70 | addBidirectionalLinks(3L, 6L, 7L, 6L); |
| 71 | addBidirectionalLinks(3L, 7L, 8L, 7L); |
| 72 | addBidirectionalLinks(3L, 8L, 9L, 8L); |
| 73 | addBidirectionalLinks(4L, 9l, 9L, 9L); |
| 74 | |
| 75 | // set capacity of all links to 1000Mbps |
| 76 | for (Link link: getLinks()) { |
| 77 | ((LinkImpl)link).setCapacity(1000.0); |
| 78 | } |
| 79 | } |
Toshio Koide | 0c9106d | 2014-02-19 15:26:38 -0800 | [diff] [blame] | 80 | |
| 81 | public void removeLink(Long srcDpid, Long srcPortNo, Long dstDpid, Long dstPortNo) { |
| 82 | DetachableLinkImpl link = (DetachableLinkImpl)getSwitch(srcDpid).getPort(srcPortNo).getOutgoingLink(); |
Pavlin Radoslavov | 7c8f69a | 2014-02-19 19:01:45 -0800 | [diff] [blame] | 83 | if (link.getDstSwitch().getDpid().equals(dstDpid) && link.getDstPort().getNumber().equals(dstPortNo)) { |
Toshio Koide | 0c9106d | 2014-02-19 15:26:38 -0800 | [diff] [blame] | 84 | link.detachFromGraph(); |
| 85 | } |
| 86 | } |
Toshio Koide | ebdbb62 | 2014-02-12 20:28:38 -0800 | [diff] [blame] | 87 | } |