/* * Copyright 2016-present Open Networking Laboratory * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package org.onosproject.bandwidthmgr; import static com.google.common.base.Preconditions.checkNotNull; import static org.onosproject.net.config.NetworkConfigEvent.Type.CONFIG_ADDED; import static org.onosproject.net.config.NetworkConfigEvent.Type.CONFIG_REMOVED; import static org.onosproject.net.config.NetworkConfigEvent.Type.CONFIG_UPDATED; import java.util.LinkedHashSet; import java.util.Set; import org.apache.felix.scr.annotations.Activate; import org.apache.felix.scr.annotations.Component; import org.apache.felix.scr.annotations.Deactivate; import org.apache.felix.scr.annotations.Reference; import org.apache.felix.scr.annotations.ReferenceCardinality; import org.apache.felix.scr.annotations.Service; import org.onlab.util.KryoNamespace; import org.onosproject.net.ConnectPoint; import org.onosproject.net.LinkKey; import org.onosproject.net.config.NetworkConfigEvent; import org.onosproject.net.config.NetworkConfigListener; import org.onosproject.net.config.NetworkConfigService; import org.onosproject.bandwidthmgr.api.BandwidthMgmtStore; import org.onosproject.pcep.api.TeLinkConfig; import org.onosproject.store.serializers.KryoNamespaces; import org.onosproject.store.service.ConsistentMap; import org.onosproject.store.service.Serializer; import org.onosproject.store.service.StorageService; import org.slf4j.Logger; import org.slf4j.LoggerFactory; /** * Manages the pool of available labels to devices, links and tunnels. */ @Component(immediate = true) @Service public class DistributedBandwidthMgmtStore implements BandwidthMgmtStore { private static final Logger log = LoggerFactory.getLogger(BandwidthManager.class); @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY) protected NetworkConfigService netCfgService; @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY) protected StorageService storageService; private InternalConfigListener cfgListener = new InternalConfigListener(); private ConsistentMap<LinkKey, Double> teCost; // Locally maintain unreserved bandwidth of each link. private ConsistentMap<LinkKey, Set<Double>> unResvBw; // Mapping tunnel with link key with local reserved bandwidth private ConsistentMap<LinkKey, Double> localReservedBw; private static final Serializer SERIALIZER = Serializer .using(new KryoNamespace.Builder().register(KryoNamespaces.API) .register(KryoNamespaces.API) .register(LinkKey.class) .register(ConnectPoint.class) .build()); @Activate protected void activate() { netCfgService.addListener(cfgListener); localReservedBw = storageService.<LinkKey, Double>consistentMapBuilder() .withName("local-reserved-bandwith") .withSerializer(SERIALIZER) .build(); unResvBw = storageService.<LinkKey, Set<Double>>consistentMapBuilder() .withName("onos-unreserved-bandwidth") .withSerializer(SERIALIZER) .build(); teCost = storageService.<LinkKey, Double>consistentMapBuilder() .withName("onos-tecost") .withSerializer(SERIALIZER) .build(); log.info("Started"); } @Deactivate protected void deactivate() { netCfgService.removeListener(cfgListener); log.info("Stopped"); } @Override public Double getTeCost(LinkKey linkKey) { if (teCost.get(linkKey) != null) { return teCost.get(linkKey).value(); } return null; } @Override public boolean allocLocalReservedBw(LinkKey linkkey, Double bandwidth) { Double allocatedBw = null; if (localReservedBw.get(linkkey) != null) { allocatedBw = localReservedBw.get(linkkey).value(); } if (allocatedBw != null) { localReservedBw.put(linkkey, (allocatedBw + bandwidth)); } else { localReservedBw.put(linkkey, bandwidth); } return true; } @Override public boolean releaseLocalReservedBw(LinkKey linkkey, Double bandwidth) { Double allocatedBw = null; if (localReservedBw.get(linkkey) != null) { allocatedBw = localReservedBw.get(linkkey).value(); } if (allocatedBw == null || allocatedBw < bandwidth) { return false; } Double releasedBw = allocatedBw - bandwidth; if (releasedBw == 0.0) { localReservedBw.remove(linkkey); } else { localReservedBw.put(linkkey, releasedBw); } return true; } @Override public Double getAllocatedLocalReservedBw(LinkKey linkkey) { return localReservedBw.get(linkkey) != null ? localReservedBw.get(linkkey).value() : null; } @Override public boolean addUnreservedBw(LinkKey linkkey, Set<Double> bandwidth) { unResvBw.put(linkkey, bandwidth); return true; } @Override public boolean removeUnreservedBw(LinkKey linkkey) { unResvBw.remove(linkkey); return true; } @Override public Set<Double> getUnreservedBw(LinkKey linkkey) { checkNotNull(linkkey); return unResvBw.get(linkkey) != null ? unResvBw.get(linkkey).value() : null; } private class InternalConfigListener implements NetworkConfigListener { @Override public void event(NetworkConfigEvent event) { if (event.configClass().equals(TeLinkConfig.class)) { if ((event.type() != CONFIG_ADDED) && (event.type() != CONFIG_UPDATED) && (event.type() != CONFIG_REMOVED)) { return; } LinkKey linkKey = (LinkKey) event.subject(); switch (event.type()) { case CONFIG_ADDED: case CONFIG_UPDATED: TeLinkConfig cfg = netCfgService.getConfig(linkKey, TeLinkConfig.class); if (cfg == null) { log.error("Unable to get the configuration of the link."); return; } Set<Double> unresvBw = new LinkedHashSet<>(); unresvBw.add(cfg.unResvBandwidth()); addUnreservedBw(linkKey, unresvBw); if (cfg.teCost() != 0) { teCost.put(linkKey, (double) cfg.teCost()); } break; case CONFIG_REMOVED: removeUnreservedBw(linkKey); localReservedBw.remove(linkKey); teCost.remove(linkKey); break; default: break; } } } } }