Jonathan Hart | aa38097 | 2014-04-03 10:24:46 -0700 | [diff] [blame] | 1 | package net.onrc.onos.core.intent.runtime; |
Brian O'Connor | 7f8e301 | 2014-02-15 23:59:29 -0800 | [diff] [blame] | 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; |
Brian O'Connor | 8a52cdd | 2014-02-25 15:35:25 -0800 | [diff] [blame] | 8 | import java.util.LinkedList; |
Brian O'Connor | 7f8e301 | 2014-02-15 23:59:29 -0800 | [diff] [blame] | 9 | import java.util.List; |
| 10 | import java.util.Map; |
| 11 | import java.util.Set; |
| 12 | |
| 13 | import net.floodlightcontroller.util.MACAddress; |
Jonathan Hart | aa38097 | 2014-04-03 10:24:46 -0700 | [diff] [blame] | 14 | import net.onrc.onos.core.intent.FlowEntry; |
| 15 | import net.onrc.onos.core.intent.Intent; |
| 16 | import net.onrc.onos.core.intent.IntentOperation; |
Jonathan Hart | a99ec67 | 2014-04-03 11:30:34 -0700 | [diff] [blame] | 17 | import net.onrc.onos.core.intent.IntentOperation.Operator; |
Jonathan Hart | aa38097 | 2014-04-03 10:24:46 -0700 | [diff] [blame] | 18 | import net.onrc.onos.core.intent.IntentOperationList; |
| 19 | import net.onrc.onos.core.intent.PathIntent; |
| 20 | import net.onrc.onos.core.intent.ShortestPathIntent; |
Jonathan Hart | 472062d | 2014-04-03 10:56:48 -0700 | [diff] [blame] | 21 | import net.onrc.onos.core.topology.LinkEvent; |
| 22 | //import net.onrc.onos.core.topology.NetworkGraph; |
| 23 | |
Brian O'Connor | 8a52cdd | 2014-02-25 15:35:25 -0800 | [diff] [blame] | 24 | import org.slf4j.Logger; |
| 25 | import org.slf4j.LoggerFactory; |
| 26 | |
Brian O'Connor | 7f8e301 | 2014-02-15 23:59:29 -0800 | [diff] [blame] | 27 | /** |
Brian O'Connor | 7f8e301 | 2014-02-15 23:59:29 -0800 | [diff] [blame] | 28 | * @author Brian O'Connor <bocon@onlab.us> |
Brian O'Connor | 7f8e301 | 2014-02-15 23:59:29 -0800 | [diff] [blame] | 29 | */ |
| 30 | |
| 31 | public class PlanCalcRuntime { |
Toshio Koide | d9fa2a8 | 2014-02-19 17:35:18 -0800 | [diff] [blame] | 32 | |
Ray Milkey | 269ffb9 | 2014-04-03 14:43:30 -0700 | [diff] [blame] | 33 | // NetworkGraph graph; |
Ray Milkey | ec83894 | 2014-04-09 11:28:43 -0700 | [diff] [blame] | 34 | private static final Logger log = LoggerFactory.getLogger(PlanCalcRuntime.class); |
Brian O'Connor | 12861f7 | 2014-02-19 20:40:32 -0800 | [diff] [blame] | 35 | |
Brian O'Connor | 488e5ed | 2014-02-20 19:50:01 -0800 | [diff] [blame] | 36 | public PlanCalcRuntime(/*NetworkGraph graph*/) { |
Ray Milkey | 269ffb9 | 2014-04-03 14:43:30 -0700 | [diff] [blame] | 37 | // this.graph = graph; |
Brian O'Connor | 12861f7 | 2014-02-19 20:40:32 -0800 | [diff] [blame] | 38 | } |
| 39 | |
| 40 | public List<Set<FlowEntry>> computePlan(IntentOperationList intentOps) { |
Ray Milkey | 269ffb9 | 2014-04-03 14:43:30 -0700 | [diff] [blame] | 41 | long start = System.nanoTime(); |
| 42 | List<Collection<FlowEntry>> flowEntries = computeFlowEntries(intentOps); |
| 43 | long step1 = System.nanoTime(); |
| 44 | List<Set<FlowEntry>> plan = buildPhases(flowEntries); |
| 45 | long step2 = System.nanoTime(); |
Pavlin Radoslavov | 964f8ae | 2014-04-18 16:44:14 -0700 | [diff] [blame] | 46 | log.debug("MEASUREMENT: Compute flow entries: {} ns, Build phases: {} ns", |
Ray Milkey | 269ffb9 | 2014-04-03 14:43:30 -0700 | [diff] [blame] | 47 | (step1 - start), (step2 - step1)); |
| 48 | return plan; |
Brian O'Connor | 12861f7 | 2014-02-19 20:40:32 -0800 | [diff] [blame] | 49 | } |
| 50 | |
Brian O'Connor | 8a52cdd | 2014-02-25 15:35:25 -0800 | [diff] [blame] | 51 | private List<Collection<FlowEntry>> computeFlowEntries(IntentOperationList intentOps) { |
Ray Milkey | 269ffb9 | 2014-04-03 14:43:30 -0700 | [diff] [blame] | 52 | List<Collection<FlowEntry>> flowEntries = new LinkedList<>(); |
| 53 | for (IntentOperation i : intentOps) { |
| 54 | if (!(i.intent instanceof PathIntent)) { |
| 55 | log.warn("Not a path intent: {}", i); |
| 56 | continue; |
| 57 | } |
| 58 | PathIntent intent = (PathIntent) i.intent; |
| 59 | Intent parent = intent.getParentIntent(); |
| 60 | long srcPort, dstPort; |
| 61 | long lastDstSw = -1, lastDstPort = -1; |
| 62 | MACAddress srcMac, dstMac; |
| 63 | if (parent instanceof ShortestPathIntent) { |
| 64 | ShortestPathIntent pathIntent = (ShortestPathIntent) parent; |
| 65 | // Switch srcSwitch = graph.getSwitch(pathIntent.getSrcSwitchDpid()); |
| 66 | // srcPort = srcSwitch.getPort(pathIntent.getSrcPortNumber()); |
| 67 | srcPort = pathIntent.getSrcPortNumber(); |
| 68 | srcMac = MACAddress.valueOf(pathIntent.getSrcMac()); |
| 69 | dstMac = MACAddress.valueOf(pathIntent.getDstMac()); |
| 70 | // Switch dstSwitch = graph.getSwitch(pathIntent.getDstSwitchDpid()); |
| 71 | lastDstSw = pathIntent.getDstSwitchDpid(); |
| 72 | // lastDstPort = dstSwitch.getPort(pathIntent.getDstPortNumber()); |
| 73 | lastDstPort = pathIntent.getDstPortNumber(); |
| 74 | } else { |
| 75 | log.warn("Unsupported Intent: {}", parent); |
| 76 | continue; |
| 77 | } |
| 78 | List<FlowEntry> entries = new ArrayList<>(); |
| 79 | for (LinkEvent linkEvent : intent.getPath()) { |
| 80 | // Link link = graph.getLink(linkEvent.getSrc().getDpid(), |
| 81 | // linkEvent.getSrc().getNumber(), |
| 82 | // linkEvent.getDst().getDpid(), |
| 83 | // linkEvent.getDst().getNumber()); |
| 84 | // Switch sw = link.getSrcSwitch(); |
| 85 | long sw = linkEvent.getSrc().getDpid(); |
| 86 | // dstPort = link.getSrcPort(); |
| 87 | dstPort = linkEvent.getSrc().getNumber(); |
| 88 | FlowEntry fe = new FlowEntry(sw, srcPort, dstPort, srcMac, dstMac, i.operator); |
| 89 | entries.add(fe); |
| 90 | // srcPort = link.getDstPort(); |
| 91 | srcPort = linkEvent.getDst().getNumber(); |
| 92 | } |
| 93 | if (lastDstSw >= 0 && lastDstPort >= 0) { |
| 94 | //Switch sw = lastDstPort.getSwitch(); |
| 95 | long sw = lastDstSw; |
| 96 | dstPort = lastDstPort; |
| 97 | FlowEntry fe = new FlowEntry(sw, srcPort, dstPort, srcMac, dstMac, i.operator); |
| 98 | entries.add(fe); |
| 99 | } |
| 100 | // install flow entries in reverse order |
| 101 | Collections.reverse(entries); |
| 102 | flowEntries.add(entries); |
| 103 | } |
| 104 | return flowEntries; |
Brian O'Connor | 12861f7 | 2014-02-19 20:40:32 -0800 | [diff] [blame] | 105 | } |
Toshio Koide | d9fa2a8 | 2014-02-19 17:35:18 -0800 | [diff] [blame] | 106 | |
Brian O'Connor | 8a52cdd | 2014-02-25 15:35:25 -0800 | [diff] [blame] | 107 | private List<Set<FlowEntry>> buildPhases(List<Collection<FlowEntry>> flowEntries) { |
Ray Milkey | 269ffb9 | 2014-04-03 14:43:30 -0700 | [diff] [blame] | 108 | Map<FlowEntry, Integer> map = new HashMap<>(); |
| 109 | List<Set<FlowEntry>> plan = new ArrayList<>(); |
| 110 | for (Collection<FlowEntry> c : flowEntries) { |
| 111 | for (FlowEntry e : c) { |
| 112 | Integer i = map.get(e); |
| 113 | if (i == null) { |
| 114 | i = Integer.valueOf(0); |
| 115 | } |
| 116 | switch (e.getOperator()) { |
| 117 | case ADD: |
| 118 | i += 1; |
| 119 | break; |
| 120 | case REMOVE: |
| 121 | i -= 1; |
| 122 | break; |
| 123 | default: |
| 124 | break; |
| 125 | } |
| 126 | map.put(e, i); |
| 127 | // System.out.println(e + " " + e.getOperator()); |
| 128 | } |
| 129 | } |
Toshio Koide | bf87566 | 2014-02-24 12:19:15 -0800 | [diff] [blame] | 130 | |
Ray Milkey | 269ffb9 | 2014-04-03 14:43:30 -0700 | [diff] [blame] | 131 | // really simple first iteration of plan |
| 132 | //TODO: optimize the map in phases |
| 133 | Set<FlowEntry> phase = new HashSet<>(); |
Pavlin Radoslavov | 45a0ad3 | 2014-04-10 12:16:40 -0700 | [diff] [blame] | 134 | for (Map.Entry<FlowEntry, Integer> entry : map.entrySet()) { |
| 135 | FlowEntry e = entry.getKey(); |
| 136 | Integer i = entry.getValue(); |
Ray Milkey | 269ffb9 | 2014-04-03 14:43:30 -0700 | [diff] [blame] | 137 | if (i == 0) { |
| 138 | continue; |
| 139 | } else if (i > 0) { |
| 140 | e.setOperator(Operator.ADD); |
| 141 | } else if (i < 0) { |
| 142 | e.setOperator(Operator.REMOVE); |
| 143 | } |
| 144 | phase.add(e); |
| 145 | } |
| 146 | plan.add(phase); |
Toshio Koide | bf87566 | 2014-02-24 12:19:15 -0800 | [diff] [blame] | 147 | |
Ray Milkey | 269ffb9 | 2014-04-03 14:43:30 -0700 | [diff] [blame] | 148 | return plan; |
Brian O'Connor | 12861f7 | 2014-02-19 20:40:32 -0800 | [diff] [blame] | 149 | } |
Brian O'Connor | 7f8e301 | 2014-02-15 23:59:29 -0800 | [diff] [blame] | 150 | } |