Adding purgeOnDisconnect support to the meter subsystem and adding vlanId match

criteria to EAPOL trap flows.

* Adding purgeOnDisconnect property to MeterManager
* DeviceListener implementation on MeterManager
* Adding purgeMeter(DeviceId deviceId) method to MeterStore
* Calling the above method when DEVICE_AVAILABILITY_CHANGE is received
* Adding vlanId match criteria to EAPOL trap flows (OltPipeline change)

Change-Id: Ibb254302efe94edf1fd596f74a6eef6587410475
(cherry picked from commit 91b38543d822a0d9d092f9b3ff7760b1a206226a)

core/api/src/main/java/org/onosproject/net/meter/MeterStore.java

@@ -102,13 +102,14 @@ public interface MeterStore extends Store<MeterEvent, MeterStoreDelegate> {
      * Notifies the delegate that the meter failed to allow it
      * to nofity the app.
      *
-     * @param op a failed meter operation
+     * @param op     a failed meter operation
      * @param reason a failure reason
      */
     void failedMeter(MeterOperation op, MeterFailReason reason);
 
     /**
      * Delete this meter immediately.
+     *
      * @param m a meter
      */
     void deleteMeterNow(Meter m);
@@ -134,8 +135,15 @@ public interface MeterStore extends Store<MeterEvent, MeterStoreDelegate> {
      * Frees the given meter id.
      *
      * @param deviceId the device id
-     * @param meterId the id to be freed
+     * @param meterId  the id to be freed
      */
     void freeMeterId(DeviceId deviceId, MeterId meterId);
 
+    /**
+     * Removes all meters of given device from store.
+     *
+     * @param deviceId the device id
+     */
+    void purgeMeter(DeviceId deviceId);
+
 }

core/net/src/main/java/org/onosproject/net/meter/impl/MeterManager.java

@@ -23,10 +23,13 @@ import org.apache.felix.scr.annotations.Property;
 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.Tools;
 import org.onlab.util.TriConsumer;
 import org.onosproject.cfg.ComponentConfigService;
 import org.onosproject.mastership.MastershipService;
 import org.onosproject.net.DeviceId;
+import org.onosproject.net.device.DeviceEvent;
+import org.onosproject.net.device.DeviceListener;
 import org.onosproject.net.device.DeviceService;
 import org.onosproject.net.driver.DriverService;
 import org.onosproject.net.meter.DefaultMeter;
@@ -87,6 +90,7 @@ public class MeterManager
 
     private final Logger log = getLogger(getClass());
     private final MeterStoreDelegate delegate = new InternalMeterStoreDelegate();
+    private final DeviceListener deviceListener = new InternalDeviceListener();
 
     @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY)
     private MeterStore store;
@@ -108,6 +112,10 @@ public class MeterManager
             label = "Frequency (in seconds) for polling meters via fallback provider")
     private int fallbackMeterPollFrequency = DEFAULT_POLL_FREQUENCY;
 
+    @Property(name = "purgeOnDisconnection", boolValue = false,
+            label = "Purge entries associated with a device when the device goes offline")
+    private boolean purgeOnDisconnection = false;
+
     private TriConsumer<MeterRequest, MeterStoreResult, Throwable> onComplete;
 
     private ExecutorService executorService;
@@ -119,6 +127,7 @@ public class MeterManager
         store.setDelegate(delegate);
         cfgService.registerProperties(getClass());
         eventDispatcher.addSink(MeterEvent.class, listenerRegistry);
+        deviceService.addListener(deviceListener);
 
         onComplete = (request, result, error) -> {
                 request.context().ifPresent(c -> {
@@ -155,6 +164,7 @@ public class MeterManager
         defaultProvider.terminate();
         store.unsetDelegate(delegate);
         eventDispatcher.removeSink(MeterEvent.class);
+        deviceService.removeListener(deviceListener);
         cfgService.unregisterProperties(getClass(), false);
         executorService.shutdown();
         log.info("Stopped");
@@ -167,6 +177,17 @@ public class MeterManager
      */
     private void readComponentConfiguration(ComponentContext context) {
         Dictionary<?, ?> properties = context.getProperties();
+        Boolean flag;
+
+        flag = Tools.isPropertyEnabled(properties, "purgeOnDisconnection");
+        if (flag == null) {
+            log.info("PurgeOnDisconnection is not configured," +
+                    "using current value of {}", purgeOnDisconnection);
+        } else {
+            purgeOnDisconnection = flag;
+            log.info("Configured. PurgeOnDisconnection is {}",
+                    purgeOnDisconnection ? "enabled" : "disabled");
+        }
 
         String s = get(properties, "fallbackMeterPollFrequency");
         try {
@@ -396,4 +417,24 @@ public class MeterManager
         }
     }
 
+    private class InternalDeviceListener implements DeviceListener {
+
+        @Override
+        public void event(DeviceEvent event) {
+            switch (event.type()) {
+                case DEVICE_REMOVED:
+                case DEVICE_AVAILABILITY_CHANGED:
+                    DeviceId deviceId = event.subject().id();
+                    if (!deviceService.isAvailable(deviceId)) {
+                        if (purgeOnDisconnection) {
+                            store.purgeMeter(deviceId);
+                        }
+                    }
+                    break;
+                default:
+                    break;
+            }
+        }
+    }
+
 }

drivers/default/src/main/java/org/onosproject/driver/pipeline/OltPipeline.java

@@ -789,7 +789,9 @@ public class OltPipeline extends AbstractHandlerBehaviour implements Pipeliner {
         Instruction meter = filter.meta().metered();
         Instruction writeMetadata = filter.meta().writeMetadata();
 
-        TrafficSelector selector = buildSelector(filter.key(), ethType);
+        Criterion vlanId = filterForCriterion(filter.conditions(), Criterion.Type.VLAN_VID);
+
+        TrafficSelector selector = buildSelector(filter.key(), ethType, vlanId);
         TrafficTreatment treatment = buildTreatment(output, meter, writeMetadata);
         buildAndApplyRule(filter, selector, treatment);
 
@@ -898,19 +900,15 @@ public class OltPipeline extends AbstractHandlerBehaviour implements Pipeliner {
 
     private TrafficSelector buildSelector(Criterion... criteria) {
 
-
         TrafficSelector.Builder sBuilder = DefaultTrafficSelector.builder();
 
-        for (Criterion c : criteria) {
-            sBuilder.add(c);
-        }
+        Arrays.stream(criteria).filter(Objects::nonNull).forEach(sBuilder::add);
 
         return sBuilder.build();
     }
 
     private TrafficTreatment buildTreatment(Instruction... instructions) {
 
-
         TrafficTreatment.Builder tBuilder = DefaultTrafficTreatment.builder();
 
         Arrays.stream(instructions).filter(Objects::nonNull).forEach(tBuilder::add);

incubator/store/src/main/java/org/onosproject/incubator/store/meter/impl/DistributedMeterStore.java

@@ -67,7 +67,9 @@ import org.onosproject.store.service.Versioned;
 import org.slf4j.Logger;
 
 import java.util.Collection;
+import java.util.List;
 import java.util.Map;
+import java.util.Objects;
 import java.util.Set;
 import java.util.concurrent.CompletableFuture;
 import java.util.stream.Collectors;
@@ -335,6 +337,19 @@ public class DistributedMeterStore extends AbstractStore<MeterEvent, MeterStoreD
         notifyDelegate(new MeterEvent(MeterEvent.Type.METER_REMOVED, m));
     }
 
+    @Override
+    public void purgeMeter(DeviceId deviceId) {
+
+        List<Versioned<MeterData>> metersPendingRemove = meters.stream()
+                .filter(e -> Objects.equals(e.getKey().deviceId(), deviceId))
+                .map(Map.Entry::getValue)
+                .collect(Collectors.toList());
+
+        metersPendingRemove.forEach(versionedMeterKey
+                -> deleteMeterNow(versionedMeterKey.value().meter()));
+
+    }
+
     @Override
     public long getMaxMeters(MeterFeaturesKey key) {
         MeterFeatures features = Versioned.valueOrElse(meterFeatures.get(key), null);