1 package org.opendaylight.controller.forwardingrulesmanager.consumer.impl;
3 import java.util.ArrayList;
4 import java.util.Collection;
5 import java.util.Collections;
6 import java.util.EnumSet;
7 import java.util.HashMap;
8 import java.util.HashSet;
9 import java.util.Iterator;
10 import java.util.List;
12 import java.util.Map.Entry;
14 import java.util.concurrent.ConcurrentHashMap;
15 import java.util.concurrent.ConcurrentMap;
17 import org.opendaylight.controller.clustering.services.CacheConfigException;
18 import org.opendaylight.controller.clustering.services.CacheExistException;
19 import org.opendaylight.controller.clustering.services.IClusterContainerServices;
20 import org.opendaylight.controller.clustering.services.IClusterServices;
21 import org.opendaylight.controller.md.sal.common.api.data.DataCommitHandler;
22 import org.opendaylight.controller.md.sal.common.api.data.DataCommitHandler.DataCommitTransaction;
23 import org.opendaylight.controller.md.sal.common.api.data.DataModification;
24 import org.opendaylight.controller.sal.common.util.Rpcs;
25 import org.opendaylight.controller.sal.core.IContainer;
26 import org.opendaylight.controller.sal.core.Node;
27 import org.opendaylight.controller.sal.utils.GlobalConstants;
28 import org.opendaylight.controller.sal.utils.Status;
29 import org.opendaylight.controller.sal.utils.StatusCode;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.config.rev131024.Meters;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.config.rev131024.meters.Meter;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.config.rev131024.meters.MeterKey;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.AddMeterInputBuilder;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.MeterAdded;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.MeterRemoved;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.MeterUpdated;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.RemoveMeterInputBuilder;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.SalMeterListener;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.SalMeterService;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.UpdateMeterInputBuilder;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.meter.update.UpdatedMeterBuilder;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.MeterId;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.band.type.BandType;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.band.type.band.type.Drop;
45 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.band.type.band.type.DscpRemark;
46 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.band.type.band.type.Experimenter;
47 import org.opendaylight.yangtools.concepts.Registration;
48 import org.opendaylight.yangtools.yang.binding.DataObject;
49 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
50 import org.opendaylight.yangtools.yang.binding.NotificationListener;
51 import org.opendaylight.yangtools.yang.common.RpcError;
52 import org.opendaylight.yangtools.yang.common.RpcResult;
53 import org.slf4j.Logger;
54 import org.slf4j.LoggerFactory;
56 public class MeterConsumerImpl implements IForwardingRulesManager {
57 protected static final Logger logger = LoggerFactory.getLogger(MeterConsumerImpl.class);
58 private final MeterEventListener meterEventListener = new MeterEventListener();
59 private Registration<NotificationListener> meterListener;
60 private SalMeterService meterService;
61 private MeterDataCommitHandler commitHandler;
63 private ConcurrentMap<MeterKey, Meter> originalSwMeterView;
64 @SuppressWarnings("unused")
65 private ConcurrentMap<MeterKey, Meter> installedSwMeterView;
66 @SuppressWarnings("unused")
67 private ConcurrentMap<Node, List<Meter>> nodeMeters;
68 @SuppressWarnings("unused")
69 private ConcurrentMap<MeterKey, Meter> inactiveMeters;
70 @SuppressWarnings("unused")
71 private IContainer container;
72 private IClusterContainerServices clusterMeterContainerService = null;
74 public MeterConsumerImpl() {
75 InstanceIdentifier<? extends DataObject> path = InstanceIdentifier.builder(Meters.class).toInstance();
76 meterService = FRMConsumerImpl.getProviderSession().getRpcService(SalMeterService.class);
77 clusterMeterContainerService = FRMConsumerImpl.getClusterContainerService();
78 container = FRMConsumerImpl.getContainer();
80 if (!(cacheStartup())) {
81 logger.error("Unable to allocate/retrieve meter cache");
82 System.out.println("Unable to allocate/retrieve meter cache");
85 if (null == meterService) {
86 logger.error("Consumer SAL Meter Service is down or NULL. FRM may not function as intended");
87 System.out.println("Consumer SAL Meter Service is down or NULL.");
91 // For switch/plugin events
92 meterListener = FRMConsumerImpl.getNotificationService().registerNotificationListener(meterEventListener);
94 if (null == meterListener) {
95 logger.error("Listener to listen on meter data modifcation events");
96 System.out.println("Listener to listen on meter data modifcation events.");
100 commitHandler = new MeterDataCommitHandler();
101 FRMConsumerImpl.getDataProviderService().registerCommitHandler(path, commitHandler);
104 private boolean allocateMeterCaches() {
105 if (this.clusterMeterContainerService == null) {
106 logger.warn("Meter: Un-initialized clusterMeterContainerService, can't create cache");
111 clusterMeterContainerService.createCache("frm.originalSwMeterView",
112 EnumSet.of(IClusterServices.cacheMode.TRANSACTIONAL));
114 clusterMeterContainerService.createCache("frm.installedSwMeterView",
115 EnumSet.of(IClusterServices.cacheMode.TRANSACTIONAL));
117 clusterMeterContainerService.createCache("frm.inactiveMeters",
118 EnumSet.of(IClusterServices.cacheMode.TRANSACTIONAL));
120 clusterMeterContainerService.createCache("frm.nodeMeters",
121 EnumSet.of(IClusterServices.cacheMode.TRANSACTIONAL));
123 // TODO for cluster mode
125 * clusterMeterContainerService.createCache(WORK_STATUS_CACHE,
126 * EnumSet.of(IClusterServices.cacheMode.NON_TRANSACTIONAL,
127 * IClusterServices.cacheMode.ASYNC));
129 * clusterMeterContainerService.createCache(WORK_ORDER_CACHE,
130 * EnumSet.of(IClusterServices.cacheMode.NON_TRANSACTIONAL,
131 * IClusterServices.cacheMode.ASYNC));
134 } catch (CacheConfigException cce) {
135 logger.error("Meter CacheConfigException");
138 } catch (CacheExistException cce) {
139 logger.error(" Meter CacheExistException");
145 private void nonClusterMeterObjectCreate() {
146 originalSwMeterView = new ConcurrentHashMap<MeterKey, Meter>();
147 installedSwMeterView = new ConcurrentHashMap<MeterKey, Meter>();
148 nodeMeters = new ConcurrentHashMap<Node, List<Meter>>();
149 inactiveMeters = new ConcurrentHashMap<MeterKey, Meter>();
152 @SuppressWarnings({ "unchecked" })
153 private boolean retrieveMeterCaches() {
154 ConcurrentMap<?, ?> map;
156 if (this.clusterMeterContainerService == null) {
157 logger.warn("Meter: un-initialized clusterMeterContainerService, can't retrieve cache");
158 nonClusterMeterObjectCreate();
162 map = clusterMeterContainerService.getCache("frm.originalSwMeterView");
164 originalSwMeterView = (ConcurrentMap<MeterKey, Meter>) map;
166 logger.error("Retrieval of cache(originalSwMeterView) failed");
170 map = clusterMeterContainerService.getCache("frm.installedSwMeterView");
172 installedSwMeterView = (ConcurrentMap<MeterKey, Meter>) map;
174 logger.error("Retrieval of cache(installedSwMeterView) failed");
178 map = clusterMeterContainerService.getCache("frm.inactiveMeters");
180 inactiveMeters = (ConcurrentMap<MeterKey, Meter>) map;
182 logger.error("Retrieval of cache(inactiveMeters) failed");
186 map = clusterMeterContainerService.getCache("frm.nodeMeters");
188 nodeMeters = (ConcurrentMap<Node, List<Meter>>) map;
190 logger.error("Retrieval of cache(nodeMeter) failed");
197 private boolean cacheStartup() {
198 if (allocateMeterCaches()) {
199 if (retrieveMeterCaches()) {
208 * Adds Meter to the southbound plugin and our internal database
213 private Status addMeter(InstanceIdentifier<?> path, Meter meterAddDataObject) {
214 MeterKey meterKey = meterAddDataObject.getKey();
216 if (null != meterKey && validateMeter(meterAddDataObject, FRMUtil.operation.ADD).isSuccess()) {
217 AddMeterInputBuilder meterBuilder = new AddMeterInputBuilder();
218 meterBuilder.setContainerName(meterAddDataObject.getContainerName());
219 meterBuilder.setFlags(meterAddDataObject.getFlags());
220 meterBuilder.setMeterBandHeaders(meterAddDataObject.getMeterBandHeaders());
221 meterBuilder.setMeterId(new MeterId(meterAddDataObject.getId()));
222 meterBuilder.setNode(meterAddDataObject.getNode());
223 meterService.addMeter(meterBuilder.build());
225 return new Status(StatusCode.BADREQUEST, "Meter Key or attribute validation failed");
228 return new Status(StatusCode.SUCCESS);
232 * Update Meter to the southbound plugin and our internal database
238 private Status updateMeter(InstanceIdentifier<?> path, Meter meterUpdateDataObject) {
239 MeterKey meterKey = meterUpdateDataObject.getKey();
240 UpdatedMeterBuilder updateMeterBuilder = null;
242 if (null != meterKey && validateMeter(meterUpdateDataObject, FRMUtil.operation.UPDATE).isSuccess()) { UpdateMeterInputBuilder updateMeterInputBuilder = new UpdateMeterInputBuilder();
243 updateMeterBuilder = new UpdatedMeterBuilder();
244 updateMeterBuilder.fieldsFrom(meterUpdateDataObject);
245 updateMeterInputBuilder.setUpdatedMeter(updateMeterBuilder.build());
246 meterService.updateMeter(updateMeterInputBuilder.build());
248 return new Status(StatusCode.BADREQUEST, "Meter Key or attribute validation failed");
251 return new Status(StatusCode.SUCCESS);
255 * Remove Meter to the southbound plugin and our internal database
261 private Status removeMeter(InstanceIdentifier<?> path, Meter meterRemoveDataObject) {
262 MeterKey meterKey = meterRemoveDataObject.getKey();
264 if (null != meterKey && validateMeter(meterRemoveDataObject, FRMUtil.operation.DELETE).isSuccess()) {
265 RemoveMeterInputBuilder meterBuilder = new RemoveMeterInputBuilder();
266 meterBuilder.setContainerName(meterRemoveDataObject.getContainerName());
267 meterBuilder.setNode(meterRemoveDataObject.getNode());
268 meterBuilder.setFlags(meterRemoveDataObject.getFlags());
269 meterBuilder.setMeterBandHeaders(meterRemoveDataObject.getMeterBandHeaders());
270 meterBuilder.setMeterId(meterRemoveDataObject.getMeterId());
271 meterBuilder.setNode(meterRemoveDataObject.getNode());
272 meterService.removeMeter(meterBuilder.build());
274 return new Status(StatusCode.BADREQUEST, "Meter Key or attribute validation failed");
277 return new Status(StatusCode.SUCCESS);
280 public Status validateMeter(Meter meter, FRMUtil.operation operation) {
281 String containerName;
283 Status returnStatus = null;
286 containerName = meter.getContainerName();
288 if (null == containerName) {
289 containerName = GlobalConstants.DEFAULT.toString();
290 } else if (!FRMUtil.isNameValid(containerName)) {
291 logger.error("Container Name is invalid %s" + containerName);
292 returnStatus = new Status(StatusCode.BADREQUEST, "Container Name is invalid");
296 meterName = meter.getMeterName();
297 if (!FRMUtil.isNameValid(meterName)) {
298 logger.error("Meter Name is invalid %s" + meterName);
299 returnStatus = new Status(StatusCode.BADREQUEST, "Meter Name is invalid");
303 for (int i = 0; i < meter.getMeterBandHeaders().getMeterBandHeader().size(); i++) {
304 if (!meter.getFlags().isMeterBurst()) {
305 if (0 < meter.getMeterBandHeaders().getMeterBandHeader().get(i).getBurstSize()) {
306 logger.error("Burst size should only be associated when Burst FLAG is set");
307 returnStatus = new Status(StatusCode.BADREQUEST,
308 "Burst size should only be associated when Burst FLAG is set");
314 if (null != returnStatus && !returnStatus.isSuccess()) {
317 BandType setBandType = null;
318 DscpRemark dscpRemark = null;
319 for (int i = 0; i < meter.getMeterBandHeaders().getMeterBandHeader().size(); i++) {
320 setBandType = meter.getMeterBandHeaders().getMeterBandHeader().get(i).getBandType();
321 if (setBandType instanceof DscpRemark) {
322 dscpRemark = (DscpRemark) setBandType;
323 if (0 > dscpRemark.getRate()) {
326 } else if (setBandType instanceof Drop) {
327 if (0 < dscpRemark.getPercLevel()) {
328 logger.error("Number of drop Precedence level");
330 } else if (setBandType instanceof Experimenter) {
336 return new Status(StatusCode.SUCCESS);
339 final class InternalTransaction implements DataCommitTransaction<InstanceIdentifier<?>, DataObject> {
341 private final DataModification<InstanceIdentifier<?>, DataObject> modification;
344 public DataModification<InstanceIdentifier<?>, DataObject> getModification() {
348 public InternalTransaction(DataModification<InstanceIdentifier<?>, DataObject> modification) {
349 this.modification = modification;
352 Map<InstanceIdentifier<?>, Meter> additions = new HashMap<>();
353 Map<InstanceIdentifier<?>, Meter> updates = new HashMap<>();
354 Set<InstanceIdentifier<?>> removals = new HashSet<>();
357 * We create a plan which flows will be added, which will be updated and
358 * which will be removed based on our internal state.
361 void prepareUpdate() {
363 Set<Entry<InstanceIdentifier<?>, DataObject>> addMeter = modification.getCreatedConfigurationData().entrySet();
364 for (Entry<InstanceIdentifier<?>, DataObject> entry : addMeter) {
365 if (entry.getValue() instanceof Meter) {
366 Meter meter = (Meter) entry.getValue();
367 additions.put(entry.getKey(), meter);
372 Set<Entry<InstanceIdentifier<?>, DataObject>> updateMeter = modification.getUpdatedConfigurationData().entrySet();
373 for (Entry<InstanceIdentifier<?>, DataObject> entry : updateMeter) {
374 if (entry.getValue() instanceof Meter) {
375 Meter meter = (Meter) entry.getValue();
376 ///will be fixed once getUpdatedConfigurationData returns only updated data not created data with it.
377 if (!additions.containsKey(entry.getKey())) {
378 updates.put(entry.getKey(), meter);
383 removals = modification.getRemovedConfigurationData();
387 * We are OK to go with execution of plan
391 public RpcResult<Void> finish() throws IllegalStateException {
393 RpcResult<Void> rpcStatus = commitToPlugin(this);
394 // We return true if internal transaction is successful.
395 // return Rpcs.getRpcResult(true, null, Collections.emptySet());
401 * We should rollback our preparation
405 public RpcResult<Void> rollback() throws IllegalStateException {
406 // NOOP - we did not modified any internal state during
407 // requestCommit phase
408 // return Rpcs.getRpcResult(true, null, Collections.emptySet());
409 return Rpcs.getRpcResult(true, null, Collections.<RpcError>emptySet());
415 private RpcResult<Void> commitToPlugin(InternalTransaction transaction) {
416 for (Entry<InstanceIdentifier<?>, Meter> entry : transaction.additions.entrySet()) {
418 if (!addMeter(entry.getKey(), entry.getValue()).isSuccess()) {
419 return Rpcs.getRpcResult(false, null, Collections.<RpcError>emptySet());
422 for (Entry<InstanceIdentifier<?>, Meter> entry : transaction.updates.entrySet()) {
424 if (!updateMeter(entry.getKey(), entry.getValue()).isSuccess()) {
425 return Rpcs.getRpcResult(false, null, Collections.<RpcError>emptySet());
429 for (InstanceIdentifier<?> meterId : transaction.removals) {
430 DataObject removeValue = transaction.getModification().getOriginalConfigurationData().get(meterId);
432 if(removeValue instanceof Meter) {
433 if(!removeMeter(meterId, (Meter)removeValue).isSuccess()) {
434 return Rpcs.getRpcResult(false, null, Collections.<RpcError>emptySet());
439 return Rpcs.getRpcResult(true, null, Collections.<RpcError>emptySet());
442 private final class MeterDataCommitHandler implements DataCommitHandler<InstanceIdentifier<?>, DataObject> {
444 public org.opendaylight.controller.md.sal.common.api.data.DataCommitHandler.DataCommitTransaction<InstanceIdentifier<?>, DataObject> requestCommit(
445 DataModification<InstanceIdentifier<?>, DataObject> modification) {
446 // We should verify transaction
447 InternalTransaction transaction = new InternalTransaction(modification);
448 transaction.prepareUpdate();
453 final class MeterEventListener implements SalMeterListener {
455 List<MeterAdded> addedMeter = new ArrayList<>();
456 List<MeterRemoved> removeMeter = new ArrayList<>();
457 List<MeterUpdated> updatedMeter = new ArrayList<>();
460 public void onMeterAdded(MeterAdded notification) {
461 // TODO Auto-generated method stub
466 public void onMeterRemoved(MeterRemoved notification) {
467 // TODO Auto-generated method stub
472 public void onMeterUpdated(MeterUpdated notification) {
473 // TODO Auto-generated method stub
479 public List<DataObject> get() {
481 List<DataObject> orderedList = new ArrayList<DataObject>();
482 Collection<Meter> meterList = originalSwMeterView.values();
483 for (Iterator<Meter> iterator = meterList.iterator(); iterator.hasNext();) {
484 orderedList.add(iterator.next());
490 public DataObject getWithName(String name, Node n) {
491 if (this instanceof MeterConsumerImpl) {
492 Collection<Meter> meterList = originalSwMeterView.values();
493 for (Iterator<Meter> iterator = meterList.iterator(); iterator.hasNext();) {
494 Meter meter = iterator.next();
495 if (meter.getNode().equals(n) && meter.getMeterName().equals(name)) {