blob: 85ea16c68b3e92cb33305ba061f45598a72b21f8 [file] [log] [blame]
package net.onrc.onos.intent;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import net.onrc.onos.ofcontroller.networkgraph.Link;
import net.onrc.onos.ofcontroller.networkgraph.LinkEvent;
import net.onrc.onos.ofcontroller.networkgraph.PortEvent.SwitchPort;
/**
* @author Toshio Koide (t-koide@onlab.us)
*/
public class PathIntentMap extends IntentMap {
private HashMap<Long, HashMap<Long, HashSet<PathIntent>>> intents;
public PathIntentMap() {
intents = new HashMap<>();
}
private HashSet<PathIntent> get(SwitchPort swPort) {
Long dpid = swPort.getDpid();
Long port = swPort.getNumber();
HashMap<Long, HashSet<PathIntent>> portToIntents = intents.get(dpid);
if (portToIntents == null) {
portToIntents = new HashMap<>();
intents.put(dpid, portToIntents);
}
HashSet<PathIntent> targetIntents = portToIntents.get(port);
if (targetIntents == null) {
targetIntents = new HashSet<>();
portToIntents.put(port, targetIntents);
}
return targetIntents;
}
private void put(SwitchPort swPort, PathIntent intent) {
get(swPort).add(intent);
}
@Override
protected void putIntent(Intent intent) {
if (!(intent instanceof PathIntent)) return; // TODO throw exception
super.putIntent(intent);
PathIntent pathIntent = (PathIntent) intent;
for (LinkEvent linkEvent: pathIntent.getPath()) {
put(linkEvent.getSrc(), (PathIntent) intent);
put(linkEvent.getDst(), (PathIntent) intent);
}
}
@Override
protected void removeIntent(String intentId) {
PathIntent intent = (PathIntent) getIntent(intentId);
for (LinkEvent linkEvent: intent.getPath()) {
get(linkEvent.getSrc()).remove(intent);
get(linkEvent.getDst()).remove(intent);
}
super.removeIntent(intentId);
}
public Collection<PathIntent> getIntentsByLink(LinkEvent linkEvent) {
return getIntentsByPort(
linkEvent.getSrc().getDpid(),
linkEvent.getSrc().getNumber());
}
public Collection<PathIntent> getIntentsByPort(Long dpid, Long port) {
HashMap<Long, HashSet<PathIntent>> portToIntents = intents.get(dpid);
if (portToIntents != null) {
HashSet<PathIntent> targetIntents = portToIntents.get(port);
if (targetIntents != null) {
return Collections.unmodifiableCollection(targetIntents);
}
}
return new HashSet<>();
}
public Collection<PathIntent> getIntentsByDpid(Long dpid) {
HashSet<PathIntent> result = new HashSet<>();
HashMap<Long, HashSet<PathIntent>> portToIntents = intents.get(dpid);
if (portToIntents != null) {
for (HashSet<PathIntent> targetIntents: portToIntents.values()) {
result.addAll(targetIntents);
}
}
return result;
}
/**
* calculate available bandwidth of specified link
* @param link
* @return
*/
public Double getAvailableBandwidth(Link link) {
if (link == null) return null;
Double bandwidth = link.getCapacity();
LinkEvent linkEvent = new LinkEvent(link);
if (!bandwidth.isInfinite()) {
for (PathIntent intent: getIntentsByLink(linkEvent)) {
Double intentBandwidth = intent.getBandwidth();
if (intentBandwidth == null || intentBandwidth.isInfinite() || intentBandwidth.isNaN())
continue;
bandwidth -= intentBandwidth;
}
}
return bandwidth;
}
}