Brian O'Connor | 7f8e301 | 2014-02-15 23:59:29 -0800 | [diff] [blame] | 1 | package net.onrc.onos.intent.runtime; |
| 2 | |
| 3 | import java.util.ArrayList; |
| 4 | import java.util.Collection; |
| 5 | import java.util.Collections; |
| 6 | import java.util.HashMap; |
| 7 | import java.util.HashSet; |
| 8 | import java.util.List; |
| 9 | import java.util.Map; |
| 10 | import java.util.Set; |
| 11 | |
| 12 | import net.floodlightcontroller.util.MACAddress; |
| 13 | import net.onrc.onos.intent.FlowEntry; |
| 14 | import net.onrc.onos.intent.Intent; |
Brian O'Connor | 12861f7 | 2014-02-19 20:40:32 -0800 | [diff] [blame] | 15 | import net.onrc.onos.intent.IntentOperation; |
| 16 | import net.onrc.onos.intent.IntentOperation.Operator; |
Toshio Koide | 103ccb2 | 2014-02-18 21:51:25 -0800 | [diff] [blame] | 17 | import net.onrc.onos.intent.IntentOperationList; |
Brian O'Connor | 7f8e301 | 2014-02-15 23:59:29 -0800 | [diff] [blame] | 18 | import net.onrc.onos.intent.PathIntent; |
Brian O'Connor | 7f8e301 | 2014-02-15 23:59:29 -0800 | [diff] [blame] | 19 | import net.onrc.onos.intent.ShortestPathIntent; |
Toshio Koide | d9fa2a8 | 2014-02-19 17:35:18 -0800 | [diff] [blame] | 20 | import net.onrc.onos.ofcontroller.networkgraph.LinkEvent; |
Brian O'Connor | 488e5ed | 2014-02-20 19:50:01 -0800 | [diff] [blame] | 21 | //import net.onrc.onos.ofcontroller.networkgraph.NetworkGraph; |
Brian O'Connor | 7f8e301 | 2014-02-15 23:59:29 -0800 | [diff] [blame] | 22 | |
| 23 | /** |
Toshio Koide | d9fa2a8 | 2014-02-19 17:35:18 -0800 | [diff] [blame] | 24 | * |
Brian O'Connor | 7f8e301 | 2014-02-15 23:59:29 -0800 | [diff] [blame] | 25 | * @author Brian O'Connor <bocon@onlab.us> |
| 26 | * |
| 27 | */ |
| 28 | |
| 29 | public class PlanCalcRuntime { |
Toshio Koide | d9fa2a8 | 2014-02-19 17:35:18 -0800 | [diff] [blame] | 30 | |
Brian O'Connor | 488e5ed | 2014-02-20 19:50:01 -0800 | [diff] [blame] | 31 | // NetworkGraph graph; |
Brian O'Connor | 12861f7 | 2014-02-19 20:40:32 -0800 | [diff] [blame] | 32 | |
Brian O'Connor | 488e5ed | 2014-02-20 19:50:01 -0800 | [diff] [blame] | 33 | public PlanCalcRuntime(/*NetworkGraph graph*/) { |
| 34 | // this.graph = graph; |
Brian O'Connor | 12861f7 | 2014-02-19 20:40:32 -0800 | [diff] [blame] | 35 | } |
| 36 | |
| 37 | public List<Set<FlowEntry>> computePlan(IntentOperationList intentOps) { |
| 38 | Set<Collection<FlowEntry>> flowEntries = computeFlowEntries(intentOps); |
| 39 | return buildPhases(flowEntries); |
| 40 | } |
| 41 | |
| 42 | private Set<Collection<FlowEntry>> computeFlowEntries(IntentOperationList intentOps) { |
| 43 | Set<Collection<FlowEntry>> flowEntries = new HashSet<>(); |
| 44 | for(IntentOperation i : intentOps) { |
| 45 | PathIntent intent = (PathIntent) i.intent; |
| 46 | Intent parent = intent.getParentIntent(); |
Brian O'Connor | 488e5ed | 2014-02-20 19:50:01 -0800 | [diff] [blame] | 47 | long srcPort, dstPort; |
| 48 | long lastDstSw = -1, lastDstPort = -1; |
Brian O'Connor | 12861f7 | 2014-02-19 20:40:32 -0800 | [diff] [blame] | 49 | MACAddress srcMac, dstMac; |
| 50 | if(parent instanceof ShortestPathIntent) { |
| 51 | ShortestPathIntent pathIntent = (ShortestPathIntent) parent; |
Brian O'Connor | 488e5ed | 2014-02-20 19:50:01 -0800 | [diff] [blame] | 52 | // Switch srcSwitch = graph.getSwitch(pathIntent.getSrcSwitchDpid()); |
| 53 | // srcPort = srcSwitch.getPort(pathIntent.getSrcPortNumber()); |
| 54 | srcPort = pathIntent.getSrcPortNumber(); |
Brian O'Connor | 12861f7 | 2014-02-19 20:40:32 -0800 | [diff] [blame] | 55 | srcMac = MACAddress.valueOf(pathIntent.getSrcMac()); |
| 56 | dstMac = MACAddress.valueOf(pathIntent.getDstMac()); |
Brian O'Connor | 488e5ed | 2014-02-20 19:50:01 -0800 | [diff] [blame] | 57 | // Switch dstSwitch = graph.getSwitch(pathIntent.getDstSwitchDpid()); |
| 58 | lastDstSw = pathIntent.getDstSwitchDpid(); |
| 59 | // lastDstPort = dstSwitch.getPort(pathIntent.getDstPortNumber()); |
| 60 | lastDstPort = pathIntent.getDstPortNumber(); |
Brian O'Connor | 12861f7 | 2014-02-19 20:40:32 -0800 | [diff] [blame] | 61 | } |
| 62 | else { |
| 63 | // TODO: log this error |
| 64 | continue; |
| 65 | } |
| 66 | List<FlowEntry> entries = new ArrayList<>(); |
| 67 | for(LinkEvent linkEvent : intent.getPath()) { |
Brian O'Connor | 488e5ed | 2014-02-20 19:50:01 -0800 | [diff] [blame] | 68 | // Link link = graph.getLink(linkEvent.getSrc().getDpid(), |
| 69 | // linkEvent.getSrc().getNumber(), |
| 70 | // linkEvent.getDst().getDpid(), |
| 71 | // linkEvent.getDst().getNumber()); |
| 72 | // Switch sw = link.getSrcSwitch(); |
| 73 | long sw = linkEvent.getSrc().getDpid(); |
| 74 | // dstPort = link.getSrcPort(); |
| 75 | dstPort = linkEvent.getSrc().getNumber(); |
Brian O'Connor | 12861f7 | 2014-02-19 20:40:32 -0800 | [diff] [blame] | 76 | FlowEntry fe = new FlowEntry(sw, srcPort, dstPort, srcMac, dstMac, i.operator); |
| 77 | entries.add(fe); |
Brian O'Connor | 488e5ed | 2014-02-20 19:50:01 -0800 | [diff] [blame] | 78 | // srcPort = link.getDstPort(); |
| 79 | srcPort = linkEvent.getDst().getNumber(); |
Brian O'Connor | 12861f7 | 2014-02-19 20:40:32 -0800 | [diff] [blame] | 80 | } |
Brian O'Connor | 488e5ed | 2014-02-20 19:50:01 -0800 | [diff] [blame] | 81 | if(lastDstSw >= 0 && lastDstPort >= 0) { |
| 82 | //Switch sw = lastDstPort.getSwitch(); |
| 83 | long sw = lastDstSw; |
Brian O'Connor | 12861f7 | 2014-02-19 20:40:32 -0800 | [diff] [blame] | 84 | dstPort = lastDstPort; |
| 85 | FlowEntry fe = new FlowEntry(sw, srcPort, dstPort, srcMac, dstMac, i.operator); |
| 86 | entries.add(fe); |
| 87 | } |
| 88 | // install flow entries in reverse order |
| 89 | Collections.reverse(entries); |
| 90 | flowEntries.add(entries); |
Brian O'Connor | 7f8e301 | 2014-02-15 23:59:29 -0800 | [diff] [blame] | 91 | } |
Brian O'Connor | 12861f7 | 2014-02-19 20:40:32 -0800 | [diff] [blame] | 92 | return flowEntries; |
| 93 | } |
Toshio Koide | d9fa2a8 | 2014-02-19 17:35:18 -0800 | [diff] [blame] | 94 | |
Brian O'Connor | 12861f7 | 2014-02-19 20:40:32 -0800 | [diff] [blame] | 95 | private List<Set<FlowEntry>> buildPhases(Set<Collection<FlowEntry>> flowEntries) { |
| 96 | Map<FlowEntry, Integer> map = new HashMap<>(); |
| 97 | List<Set<FlowEntry>> plan = new ArrayList<>(); |
| 98 | for(Collection<FlowEntry> c : flowEntries) { |
| 99 | for(FlowEntry e : c) { |
| 100 | Integer i = map.get(e); |
| 101 | if(i == null) { |
| 102 | i = Integer.valueOf(0); |
Brian O'Connor | 7f8e301 | 2014-02-15 23:59:29 -0800 | [diff] [blame] | 103 | } |
Brian O'Connor | 12861f7 | 2014-02-19 20:40:32 -0800 | [diff] [blame] | 104 | switch(e.getOperator()) { |
| 105 | case ADD: |
| 106 | i += 1; |
| 107 | break; |
| 108 | case REMOVE: |
| 109 | i -= 1; |
| 110 | break; |
Brian O'Connor | 488e5ed | 2014-02-20 19:50:01 -0800 | [diff] [blame] | 111 | default: |
| 112 | break; |
Brian O'Connor | 7f8e301 | 2014-02-15 23:59:29 -0800 | [diff] [blame] | 113 | } |
Brian O'Connor | 12861f7 | 2014-02-19 20:40:32 -0800 | [diff] [blame] | 114 | map.put(e, i); |
| 115 | System.out.println(e + " " + e.getOperator()); |
| 116 | } |
Brian O'Connor | 7f8e301 | 2014-02-15 23:59:29 -0800 | [diff] [blame] | 117 | } |
Brian O'Connor | 12861f7 | 2014-02-19 20:40:32 -0800 | [diff] [blame] | 118 | |
| 119 | // really simple first iteration of plan |
| 120 | //TODO: optimize the map in phases |
| 121 | Set<FlowEntry> phase = new HashSet<>(); |
| 122 | for(FlowEntry e : map.keySet()) { |
| 123 | Integer i = map.get(e); |
| 124 | if(i == 0) { |
| 125 | continue; |
| 126 | } |
| 127 | else if(i > 0) { |
| 128 | e.setOperator(Operator.ADD); |
| 129 | } |
| 130 | else if(i < 0) { |
| 131 | e.setOperator(Operator.REMOVE); |
| 132 | } |
| 133 | phase.add(e); |
| 134 | } |
| 135 | plan.add(phase); |
| 136 | |
| 137 | return plan; |
| 138 | } |
Brian O'Connor | 7f8e301 | 2014-02-15 23:59:29 -0800 | [diff] [blame] | 139 | } |