[SDFAB-542] Meters cleanup and leftovers v2
- Code clean up (unused code, unuseful comments)
- Remove deprecated internal APIs
- Prevent the ejection of the meter pollers
- Prevent the ejection of the mf pollers
- Fix unproper filter of device events
- Fix delete on store which updated existing meters with dummy value
- Fix NPE in TofinoMeterProgrammable caused by default config
- Update unit tests
Change-Id: Ib2767e3ab3cf146693e61b7e1890419c9743d521
diff --git a/core/store/dist/src/main/java/org/onosproject/store/meter/impl/DistributedMeterStore.java b/core/store/dist/src/main/java/org/onosproject/store/meter/impl/DistributedMeterStore.java
index 5b9252d..b94375c 100644
--- a/core/store/dist/src/main/java/org/onosproject/store/meter/impl/DistributedMeterStore.java
+++ b/core/store/dist/src/main/java/org/onosproject/store/meter/impl/DistributedMeterStore.java
@@ -37,7 +37,6 @@
import org.onosproject.net.meter.MeterFailReason;
import org.onosproject.net.meter.MeterFeatures;
import org.onosproject.net.meter.MeterFeaturesFlag;
-import org.onosproject.net.meter.MeterFeaturesKey;
import org.onosproject.net.meter.MeterId;
import org.onosproject.net.meter.MeterKey;
import org.onosproject.net.meter.MeterOperation;
@@ -115,13 +114,10 @@
// Meters id related objects
private static final String AVAILABLEMETERIDSTORE = "onos-meters-available-store";
- // Available meter identifiers
protected ConcurrentMap<MeterTableKey, DistributedSet<MeterKey>> availableMeterIds;
- // Atomic counter map for generation of new identifiers;
private static final String METERIDSTORE = "onos-meters-id-store";
private AtomicCounterMap<MeterTableKey> meterIdGenerators;
- // Serializer related objects
private static final KryoNamespace.Builder APP_KRYO_BUILDER = KryoNamespace.newBuilder()
.register(KryoNamespaces.API)
.register(MeterKey.class)
@@ -169,22 +165,20 @@
@Activate
public void activate() {
- // Init meters map and setup the map listener
meters = storageService.<MeterKey, MeterData>consistentMapBuilder()
.withName(METERSTORE)
.withSerializer(serializer).build();
meters.addListener(metersMapListener);
metersMap = meters.asJavaMap();
- // Init meter features map
+
metersFeatures = storageService.<MeterTableKey, MeterFeatures>eventuallyConsistentMapBuilder()
.withName(METERFEATURESSTORE)
.withTimestampProvider((key, features) -> new WallClockTimestamp())
.withSerializer(APP_KRYO_BUILDER).build();
metersFeatures.addListener(featuresMapListener);
- // Init the map of the available ids set
- // Set will be created when a new Meter Features is pushed to the store
+
availableMeterIds = new ConcurrentHashMap<>();
- // Init atomic map counters
+
meterIdGenerators = storageService.<MeterTableKey>atomicCounterMapBuilder()
.withName(METERIDSTORE)
.withSerializer(Serializer.using(KryoNamespaces.API,
@@ -207,14 +201,11 @@
@Override
public CompletableFuture<MeterStoreResult> addOrUpdateMeter(Meter meter) {
- // Verify integrity of the index
checkArgument(validIndex(meter), "Meter index is not valid");
CompletableFuture<MeterStoreResult> future = new CompletableFuture<>();
MeterKey key = MeterKey.key(meter.deviceId(), meter.meterCellId());
MeterData data = new MeterData(meter, null);
- // Store the future related to the operation
futures.put(key, future);
- // Check if the meter exists
try {
meters.compute(key, (k, v) -> data);
} catch (StorageException e) {
@@ -227,25 +218,25 @@
}
@Override
- public CompletableFuture<MeterStoreResult> storeMeter(Meter meter) {
- return addOrUpdateMeter(meter);
- }
-
- @Override
public CompletableFuture<MeterStoreResult> deleteMeter(Meter meter) {
- // Init steps
CompletableFuture<MeterStoreResult> future = new CompletableFuture<>();
MeterKey key = MeterKey.key(meter.deviceId(), meter.meterCellId());
- // Store the future related to the operation
futures.put(key, future);
- // Create the meter data
- MeterData data = new MeterData(meter, null);
// Update the state of the meter. It will be pruned by observing
// that it has been removed from the dataplane.
try {
- // If it does not exist in the system
- if (meters.computeIfPresent(key, (k, v) -> data) == null) {
- // Complete immediately
+ Versioned<MeterData> versionedData = meters.computeIfPresent(key, (k, v) -> {
+ DefaultMeter m = (DefaultMeter) v.meter();
+ MeterState meterState = m.state();
+ if (meterState == MeterState.PENDING_REMOVE) {
+ return v;
+ }
+ m.setState(meter.state());
+ return new MeterData(m, v.reason().isPresent() ? v.reason().get() : null);
+ });
+ // If it does not exist in the system, completes immediately
+ if (versionedData == null) {
+ futures.remove(key);
future.complete(MeterStoreResult.success());
}
} catch (StorageException e) {
@@ -254,7 +245,6 @@
futures.remove(key);
future.completeExceptionally(e);
}
- // Done, return the future
return future;
}
@@ -300,15 +290,12 @@
e.getMessage(), e);
result = MeterStoreResult.fail(TIMEOUT);
}
-
return result;
}
@Override
public MeterStoreResult deleteMeterFeatures(Collection<MeterFeatures> meterfeatures) {
- // These store operations is treated as one single operation
- // If one of them is failed, Fail is returned
- // But the failed operation will not block the rest.
+ // Same logic of storeMeterFeatures
MeterStoreResult result = MeterStoreResult.success();
for (MeterFeatures mf : meterfeatures) {
try {
@@ -387,19 +374,12 @@
}
@Override
- public void deleteMeterNow(Meter m) {
- // This method is renamed in onos-2.5
- purgeMeter(m);
- }
-
- @Override
public void purgeMeter(Meter m) {
- // Once we receive the ack from the sb
- // create the key and remove definitely the meter
+ // Once we receive the ack from the sb, create the key
+ // remove definitely the meter and free the id
MeterKey key = MeterKey.key(m.deviceId(), m.meterCellId());
try {
if (Versioned.valueOrNull(meters.remove(key)) != null) {
- // Free the id
MeterScope scope;
if (m.meterCellId().type() == PIPELINE_INDEPENDENT) {
PiMeterCellId piMeterCellId = (PiMeterCellId) m.meterCellId();
@@ -417,12 +397,6 @@
}
@Override
- public void purgeMeter(DeviceId deviceId) {
- // This method is renamed in onos-2.5
- purgeMeters(deviceId);
- }
-
- @Override
public void purgeMeters(DeviceId deviceId) {
List<Versioned<MeterData>> metersPendingRemove = meters.stream()
.filter(e -> Objects.equals(e.getKey().deviceId(), deviceId))
@@ -454,20 +428,13 @@
return userDefinedIndexMode;
}
- @Override
- public long getMaxMeters(MeterFeaturesKey key) {
- // Leverage the meter features to know the max id
- // Create a Meter Table key with FeaturesKey's device and global scope
- MeterTableKey meterTableKey = MeterTableKey.key(key.deviceId(), MeterScope.globalScope());
- return getMaxMeters(meterTableKey);
- }
-
- private long getMaxMeters(MeterTableKey key) {
- // Leverage the meter features to know the max id
+ protected long getMaxMeters(MeterTableKey key) {
MeterFeatures features = metersFeatures.get(key);
return features == null ? 0L : features.maxMeter();
}
+ // Validate index using the meter features, useful mainly
+ // when user defined index mode is enabled
private boolean validIndex(Meter meter) {
long index;
MeterTableKey key;
@@ -481,6 +448,7 @@
index = meterId.id();
key = MeterTableKey.key(meter.deviceId(), MeterScope.globalScope());
} else {
+ log.warn("Unable to validate index unsupported cell type {}", meter.meterCellId().type());
return false;
}
@@ -491,52 +459,44 @@
}
private long getStartIndex(MeterTableKey key) {
- // Leverage the meter features to know the start id
- // Since we are using index now
- // if there is no features related to the key
- // -1 is returned
MeterFeatures features = metersFeatures.get(key);
return features == null ? -1L : features.startIndex();
}
private long getEndIndex(MeterTableKey key) {
- // Leverage the meter features to know the max id
- // Since we are using index now
- // if there is no features related to the key
- // -1 is returned
MeterFeatures features = metersFeatures.get(key);
return features == null ? -1L : features.endIndex();
}
- // queryMaxMeters is implemented in FullMetersAvailable behaviour.
+ // queryMaxMeters is implemented in MeterQuery behaviour implementations.
private long queryMaxMeters(DeviceId device) {
- // Get driver handler for this device
DriverHandler handler = driverService.createHandler(device);
- // If creation failed or the device does not have this behavior
if (handler == null || !handler.hasBehaviour(MeterQuery.class)) {
- // We cannot know max meter
return 0L;
}
- // Get the behavior
+
+ // FIXME architecturally this is not right, we should fallback to this
+ // behavior in the providers. Once we do that we can remove this code.
MeterQuery query = handler.behaviour(MeterQuery.class);
- // Insert a new available key set to the map
+ // This results to be necessary because the available ids sets are created
+ // in the meter features map listener if the device does not provide the meter
+ // feature this is the only chance to create this set.
String setName = AVAILABLEMETERIDSTORE + "-" + device + "global";
MeterTableKey meterTableKey = MeterTableKey.key(device, MeterScope.globalScope());
insertAvailableKeySet(meterTableKey, setName);
- // Return as max meter the result of the query
+
return query.getMaxMeters();
}
private boolean updateMeterIdAvailability(MeterTableKey meterTableKey, MeterCellId id,
boolean available) {
- // Retrieve the set first
DistributedSet<MeterKey> keySet = availableMeterIds.get(meterTableKey);
if (keySet == null) {
- // A reusable set should be inserted when a features is pushed
log.warn("Reusable Key set for device: {} scope: {} not found",
meterTableKey.deviceId(), meterTableKey.scope());
return false;
}
+
// According to available, make available or unavailable a meter key
DeviceId deviceId = meterTableKey.deviceId();
return available ? keySet.add(MeterKey.key(deviceId, id)) :
@@ -544,62 +504,46 @@
}
private MeterCellId getNextAvailableId(Set<MeterCellId> availableIds) {
- // If there are no available ids
if (availableIds.isEmpty()) {
- // Just end the cycle
return null;
}
- // If it is the first fit
+
if (reuseStrategy == FIRST_FIT || availableIds.size() == 1) {
return availableIds.iterator().next();
}
- // If it is random, get the size
+
+ // If it is random, get the size and return a random element
int size = availableIds.size();
- // Return a random element
return Iterables.get(availableIds, RandomUtils.nextInt(size));
}
- // Implements reuse strategy
+ // Implements reuse strategy of the meter cell ids
private MeterCellId firstReusableMeterId(MeterTableKey meterTableKey) {
- // Create a Table key and use it to retrieve the reusable meterCellId set
DistributedSet<MeterKey> keySet = availableMeterIds.get(meterTableKey);
if (keySet == null) {
- // A reusable set should be inserted when a features is pushed
log.warn("Reusable Key set for device: {} scope: {} not found",
meterTableKey.deviceId(), meterTableKey.scope());
return null;
}
- // Filter key related to device id, and reduce to meter ids
+
Set<MeterCellId> localAvailableMeterIds = keySet.stream()
.filter(meterKey ->
meterKey.deviceId().equals(meterTableKey.deviceId()))
.map(MeterKey::meterCellId)
.collect(Collectors.toSet());
- // Get next available id
MeterCellId meterId = getNextAvailableId(localAvailableMeterIds);
- // Iterate until there are items
while (meterId != null) {
- // If we are able to reserve the id
if (updateMeterIdAvailability(meterTableKey, meterId, false)) {
- // Just end
return meterId;
}
- // Update the set
localAvailableMeterIds.remove(meterId);
- // Try another time
meterId = getNextAvailableId(localAvailableMeterIds);
}
- // No reusable ids
+ // there are no available ids that can be reused
return null;
}
@Override
- public MeterId allocateMeterId(DeviceId deviceId) {
- // We use global scope for MeterId
- return (MeterId) allocateMeterId(deviceId, MeterScope.globalScope());
- }
-
- @Override
public MeterCellId allocateMeterId(DeviceId deviceId, MeterScope meterScope) {
if (userDefinedIndexMode) {
log.warn("Unable to allocate meter id when user defined index mode is enabled");
@@ -611,19 +555,16 @@
// First, search for reusable key
meterCellId = firstReusableMeterId(meterTableKey);
if (meterCellId != null) {
- // A reusable key is found
return meterCellId;
}
// If there was no reusable meter id we have to generate a new value
// using start and end index as lower and upper bound respectively.
long startIndex = getStartIndex(meterTableKey);
long endIndex = getEndIndex(meterTableKey);
- // If the device does not give us MeterFeatures
+ // If the device does not give us MeterFeatures fallback to queryMeters
if (startIndex == -1L || endIndex == -1L) {
- // MeterFeatures couldn't be retrieved, fallback to queryMeters.
- // Only meaningful to OpenFLow
+ // Only meaningful for OpenFlow today
long maxMeters = queryMaxMeters(deviceId);
- // If we don't know the max, cannot proceed
if (maxMeters == 0L) {
return null;
} else {
@@ -632,18 +573,16 @@
endIndex = maxMeters;
}
}
- // Get a new value
- // If the value is smaller than the start index, get another one
+
do {
id = meterIdGenerators.getAndIncrement(meterTableKey);
} while (id < startIndex);
- // Check with the end index, and if the value is bigger, cannot proceed
if (id > endIndex) {
return null;
}
- // Done, return the value
- // If we are using global scope, return a MeterId
- // Else, return a PiMeterId
+
+ // For backward compatibility if we are using global scope,
+ // return a MeterId, otherwise we create a PiMeterCellId
if (meterScope.isGlobal()) {
return MeterId.meterId(id);
} else {
@@ -658,9 +597,23 @@
freeMeterId(meterTableKey, meterId);
}
- private void freeMeterId(MeterTableKey meterTableKey, MeterCellId meterCellId) {
+ protected void freeMeterId(DeviceId deviceId, MeterCellId meterCellId) {
+ MeterTableKey meterTableKey;
+ if (meterCellId.type() == PIPELINE_INDEPENDENT) {
+ meterTableKey = MeterTableKey.key(deviceId,
+ MeterScope.of(((PiMeterCellId) meterCellId).meterId().id()));
+ } else if (meterCellId.type() == INDEX) {
+ meterTableKey = MeterTableKey.key(deviceId, MeterScope.globalScope());
+ } else {
+ log.warn("Unable to free meter id unsupported cell type {}", meterCellId.type());
+ return;
+ }
+ freeMeterId(meterTableKey, meterCellId);
+ }
+
+ protected void freeMeterId(MeterTableKey meterTableKey, MeterCellId meterCellId) {
if (userDefinedIndexMode) {
- log.warn("Unable to free meter id when user defined index mode is enabled");
+ log.debug("Unable to free meter id when user defined index mode is enabled");
return;
}
long index;
@@ -671,13 +624,13 @@
MeterId meterId = (MeterId) meterCellId;
index = meterId.id();
} else {
+ log.warn("Unable to free meter id unsupported cell type {}", meterCellId.type());
return;
}
// Avoid to free meter not allocated
if (meterIdGenerators.get(meterTableKey) <= index) {
return;
}
- // Update the availability
updateMeterIdAvailability(meterTableKey, meterCellId, true);
}
@@ -728,12 +681,10 @@
}
break;
case REMOVE:
- // Meter removal case
futures.computeIfPresent(key, (k, v) -> {
v.complete(MeterStoreResult.success());
return null;
});
- // Finally notify the delegate
notifyDelegate(new MeterEvent(MeterEvent.Type.METER_REMOVED, data.meter()));
break;
default:
@@ -750,13 +701,11 @@
MeterFeatures meterFeatures = event.value();
switch (event.type()) {
case PUT:
- // Put a new available meter id set to the map
String setName = AVAILABLEMETERIDSTORE + "-" +
meterFeatures.deviceId() + meterFeatures.scope().id();
insertAvailableKeySet(meterTableKey, setName);
break;
case REMOVE:
- // Remove the set
DistributedSet<MeterKey> set = availableMeterIds.remove(meterTableKey);
if (set != null) {
set.destroy();
diff --git a/core/store/dist/src/test/java/org/onosproject/store/meter/impl/DistributedMeterStoreTest.java b/core/store/dist/src/test/java/org/onosproject/store/meter/impl/DistributedMeterStoreTest.java
index 5268527..ee4d75c 100644
--- a/core/store/dist/src/test/java/org/onosproject/store/meter/impl/DistributedMeterStoreTest.java
+++ b/core/store/dist/src/test/java/org/onosproject/store/meter/impl/DistributedMeterStoreTest.java
@@ -37,7 +37,6 @@
import org.onosproject.net.meter.Meter;
import org.onosproject.net.meter.MeterCellId;
import org.onosproject.net.meter.MeterFeatures;
-import org.onosproject.net.meter.MeterFeaturesKey;
import org.onosproject.net.meter.MeterId;
import org.onosproject.net.meter.MeterKey;
import org.onosproject.net.meter.MeterScope;
@@ -64,32 +63,30 @@
* Meter store tests.
*/
public class DistributedMeterStoreTest {
- // Store under testing
+
private DistributedMeterStore meterStore;
- // Device ids used during the tests
private DeviceId did1 = did("1");
private DeviceId did2 = did("2");
private DeviceId did3 = did("3");
private DeviceId did4 = did("4");
- // Meter ids used during the tests
private MeterId mid1 = MeterId.meterId(1);
private MeterId mid2 = MeterId.meterId(2);
private MeterId mid3 = MeterId.meterId(3);
+ private MeterId mid5 = MeterId.meterId(5);
+ private MeterId mid6 = MeterId.meterId(6);
private MeterId mid10 = MeterId.meterId(10);
private MeterCellId cid4 = PiMeterCellId.ofIndirect(
PiMeterId.of("foo"), 4);
private MeterCellId invalidCid = PiMeterCellId.ofIndirect(
PiMeterId.of("foo"), 11);
- // Bands used during the tests
private Band b1 = DefaultBand.builder()
.ofType(Band.Type.DROP)
.withRate(500)
.build();
- // Meters used during the tests
private Meter m1 = DefaultMeter.builder()
.forDevice(did1)
.fromApp(APP_ID)
@@ -97,7 +94,6 @@
.withUnit(Meter.Unit.KB_PER_SEC)
.withBands(Collections.singletonList(b1))
.build();
-
private Meter m2 = DefaultMeter.builder()
.forDevice(did1)
.fromApp(APP_ID_2)
@@ -105,7 +101,6 @@
.withUnit(Meter.Unit.KB_PER_SEC)
.withBands(Collections.singletonList(b1))
.build();
-
private Meter m3 = DefaultMeter.builder()
.forDevice(did2)
.fromApp(APP_ID_2)
@@ -113,7 +108,6 @@
.withUnit(Meter.Unit.KB_PER_SEC)
.withBands(Collections.singletonList(b1))
.build();
-
private Meter m4 = DefaultMeter.builder()
.forDevice(did3)
.fromApp(APP_ID)
@@ -122,7 +116,6 @@
.withBands(Collections.singletonList(b1))
.build();
- // Meter features used during the tests
private MeterFeatures mef1 = DefaultMeterFeatures.builder().forDevice(did1)
.withMaxMeters(3L)
.withBandTypes(new HashSet<>())
@@ -155,33 +148,26 @@
@Before
public void setup() {
- // Init step
meterStore = new DistributedMeterStore();
- // Let's initialize some internal services
+
TestUtils.setField(meterStore, "storageService", new TestStorageService());
TestUtils.setField(meterStore, "driverService", new TestDriverService());
-
- // Inject TestApplicationId into the DistributedMeterStore serializer
KryoNamespace.Builder testKryoBuilder = TestUtils.getField(meterStore, "APP_KRYO_BUILDER");
testKryoBuilder.register(TestApplicationId.class);
Serializer testSerializer = Serializer.using(Lists.newArrayList(testKryoBuilder.build()));
TestUtils.setField(meterStore, "serializer", testSerializer);
- // Activate the store
meterStore.activate();
}
@After
public void tearDown() {
- // Deactivate the store
meterStore.deactivate();
}
private void initMeterStore(boolean enableUserDefinedIndex) {
meterStore.userDefinedIndexMode(enableUserDefinedIndex);
- // Let's store feature for device 1
meterStore.storeMeterFeatures(mef1);
- // Let's store feature for device 2
meterStore.storeMeterFeatures(mef2);
meterStore.storeMeterFeatures(mef3);
}
@@ -191,14 +177,10 @@
*/
@Test
public void testStoreMeterFeatures() {
- // Let's store feature for device 1
- meterStore.storeMeterFeatures(mef1);
- // Verify store meter features
- assertThat(meterStore.getMaxMeters(MeterFeaturesKey.key(did1)), is(3L));
- // Let's store feature for device 1
- meterStore.storeMeterFeatures(mef2);
- // Verify store meter features
- assertThat(meterStore.getMaxMeters(MeterFeaturesKey.key(did2)), is(10L));
+ initMeterStore(false);
+
+ assertThat(meterStore.getMaxMeters(MeterTableKey.key(did1, MeterScope.globalScope())), is(3L));
+ assertThat(meterStore.getMaxMeters(MeterTableKey.key(did2, MeterScope.globalScope())), is(10L));
}
/**
@@ -206,14 +188,11 @@
*/
@Test
public void testDeleteMeterFeatures() {
- // Let's store feature for device 1
- meterStore.storeMeterFeatures(mef1);
- // Verify store meter features
- assertThat(meterStore.getMaxMeters(MeterFeaturesKey.key(did1)), is(3L));
- // Let's delete the features
+ initMeterStore(false);
+ assertThat(meterStore.getMaxMeters(MeterTableKey.key(did1, MeterScope.globalScope())), is(3L));
+
meterStore.deleteMeterFeatures(did1);
- // Verify delete meter features
- assertThat(meterStore.getMaxMeters(MeterFeaturesKey.key(did1)), is(0L));
+ assertThat(meterStore.getMaxMeters(MeterTableKey.key(did1, MeterScope.globalScope())), is(0L));
}
/**
@@ -221,12 +200,10 @@
*/
@Test
public void testAllocateId() {
- // Init the store
initMeterStore(false);
- // Allocate a meter id and verify is equal to mid1
- assertThat(mid1, is(meterStore.allocateMeterId(did1)));
- // Allocate a meter id and verify is equal to mid2
- assertThat(mid2, is(meterStore.allocateMeterId(did1)));
+
+ assertThat(mid1, is(meterStore.allocateMeterId(did1, MeterScope.globalScope())));
+ assertThat(mid2, is(meterStore.allocateMeterId(did1, MeterScope.globalScope())));
}
/**
@@ -234,18 +211,16 @@
*/
@Test
public void testFreeId() {
- // Init the store
initMeterStore(false);
- // Allocate a meter id and verify is equal to mid1
- assertThat(mid1, is(meterStore.allocateMeterId(did1)));
- // Free the above id
+ assertThat(mid1, is(meterStore.allocateMeterId(did1, MeterScope.globalScope())));
+
+ // Verify reuse strategy
meterStore.freeMeterId(did1, mid1);
- // Allocate a meter id and verify is equal to mid1
- assertThat(mid1, is(meterStore.allocateMeterId(did1)));
- // Free an id not allocated
+ assertThat(mid1, is(meterStore.allocateMeterId(did1, MeterScope.globalScope())));
+
+ // Following free does not have effect
meterStore.freeMeterId(did1, mid10);
- // Allocate a meter id and verify is equal to mid2
- assertThat(mid2, is(meterStore.allocateMeterId(did1)));
+ assertThat(mid2, is(meterStore.allocateMeterId(did1, MeterScope.globalScope())));
}
/**
@@ -253,56 +228,39 @@
*/
@Test
public void testReuseId() {
- // Init the store
initMeterStore(false);
- // Reserve id 1
- MeterId meterIdOne = meterStore.allocateMeterId(did2);
- // Free the above id
+
+ MeterCellId meterIdOne = meterStore.allocateMeterId(did2, MeterScope.globalScope());
meterStore.freeMeterId(did2, meterIdOne);
- // Start an async reservation
- CompletableFuture<MeterId> future = CompletableFuture.supplyAsync(
- () -> meterStore.allocateMeterId(did2)
+ CompletableFuture<MeterCellId> future = CompletableFuture.supplyAsync(
+ () -> meterStore.allocateMeterId(did2, MeterScope.globalScope())
);
- // Start another reservation
- MeterId meterIdTwo = meterStore.allocateMeterId(did2);
+ MeterCellId meterIdTwo = meterStore.allocateMeterId(did2, MeterScope.globalScope());
try {
meterIdOne = future.get();
} catch (InterruptedException | ExecutionException e) {
e.printStackTrace();
}
- // Ids should be different, otherwise we had clash in the store
assertNotEquals("Ids should be different", meterIdOne, meterIdTwo);
- // Free the above id
meterStore.freeMeterId(did1, meterIdOne);
- // Free the above id
meterStore.freeMeterId(did1, meterIdTwo);
- // Reserve id 1
- meterIdOne = meterStore.allocateMeterId(did2);
- // Reserve id 2
- meterStore.allocateMeterId(did2);
- // Reserve id 3
- MeterId meterIdThree = meterStore.allocateMeterId(did2);
- // Reserve id 4
- MeterId meterIdFour = meterStore.allocateMeterId(did2);
- // Free the above id
+ meterIdOne = meterStore.allocateMeterId(did2, MeterScope.globalScope());
+ meterStore.allocateMeterId(did2, MeterScope.globalScope());
+ MeterCellId meterIdThree = meterStore.allocateMeterId(did2, MeterScope.globalScope());
+ MeterCellId meterIdFour = meterStore.allocateMeterId(did2, MeterScope.globalScope());
meterStore.freeMeterId(did1, meterIdOne);
- // Free the above id
meterStore.freeMeterId(did1, meterIdThree);
- // Free the above id
meterStore.freeMeterId(did1, meterIdFour);
- // Start an async reservation
future = CompletableFuture.supplyAsync(
- () -> meterStore.allocateMeterId(did2)
+ () -> meterStore.allocateMeterId(did2, MeterScope.globalScope())
);
- // Start another reservation
- MeterId meterAnotherId = meterStore.allocateMeterId(did2);
+ MeterCellId meterAnotherId = meterStore.allocateMeterId(did2, MeterScope.globalScope());
try {
meterAnotherId = future.get();
} catch (InterruptedException | ExecutionException e) {
e.printStackTrace();
}
- // Ids should be different, otherwise we had clash in the store
assertNotEquals("Ids should be different", meterAnotherId, meterIdOne);
}
@@ -311,12 +269,10 @@
*/
@Test
public void testQueryMeters() {
- // Init the store
initMeterStore(false);
- // Let's test queryMeters
- assertThat(mid1, is(meterStore.allocateMeterId(did3)));
- // Let's test queryMeters error
- assertNull(meterStore.allocateMeterId(did4));
+
+ assertThat(mid1, is(meterStore.allocateMeterId(did3, MeterScope.globalScope())));
+ assertNull(meterStore.allocateMeterId(did4, MeterScope.globalScope()));
}
/**
@@ -324,16 +280,12 @@
*/
@Test
public void testMaxMeterError() {
- // Init the store
initMeterStore(false);
- // Reserve id 1
- assertThat(mid1, is(meterStore.allocateMeterId(did1)));
- // Reserve id 2
- assertThat(mid2, is(meterStore.allocateMeterId(did1)));
- // Reserve id 3
- assertThat(mid3, is(meterStore.allocateMeterId(did1)));
- // Max meter error
- assertNull(meterStore.allocateMeterId(did1));
+
+ assertThat(mid1, is(meterStore.allocateMeterId(did1, MeterScope.globalScope())));
+ assertThat(mid2, is(meterStore.allocateMeterId(did1, MeterScope.globalScope())));
+ assertThat(mid3, is(meterStore.allocateMeterId(did1, MeterScope.globalScope())));
+ assertNull(meterStore.allocateMeterId(did1, MeterScope.globalScope()));
}
/**
@@ -341,13 +293,11 @@
*/
@Test
public void testStoreMeter() {
- // Init the store
initMeterStore(false);
- // Simulate the allocation of an id
- MeterId idOne = meterStore.allocateMeterId(did1);
- // Verify the allocation
+
+ MeterCellId idOne = meterStore.allocateMeterId(did1, MeterScope.globalScope());
assertThat(mid1, is(idOne));
- // Let's create a meter
+
Meter meterOne = DefaultMeter.builder()
.forDevice(did1)
.fromApp(APP_ID)
@@ -355,13 +305,9 @@
.withUnit(Meter.Unit.KB_PER_SEC)
.withBands(Collections.singletonList(b1))
.build();
- // Set the state
((DefaultMeter) meterOne).setState(MeterState.PENDING_ADD);
- // Store the meter
- meterStore.storeMeter(meterOne);
- // Let's create meter key
+ meterStore.addOrUpdateMeter(meterOne);
MeterKey meterKey = MeterKey.key(did1, mid1);
- // Verify the store
assertThat(1, is(meterStore.getAllMeters().size()));
assertThat(1, is(meterStore.getAllMeters(did1).size()));
assertThat(m1, is(meterStore.getMeter(meterKey)));
@@ -372,13 +318,11 @@
*/
@Test
public void testDeleteMeter() {
- // Init the store
initMeterStore(false);
- // Simulate the allocation of an id
- MeterId idOne = meterStore.allocateMeterId(did1);
- // Verify the allocation
+
+ MeterCellId idOne = meterStore.allocateMeterId(did1, MeterScope.globalScope());
assertThat(mid1, is(idOne));
- // Let's create a meter
+
Meter meterOne = DefaultMeter.builder()
.forDevice(did1)
.fromApp(APP_ID)
@@ -386,31 +330,23 @@
.withUnit(Meter.Unit.KB_PER_SEC)
.withBands(Collections.singletonList(b1))
.build();
- // Set the state
((DefaultMeter) meterOne).setState(MeterState.PENDING_ADD);
- // Store the meter
- meterStore.storeMeter(meterOne);
- // Set the state
+ meterStore.addOrUpdateMeter(meterOne);
((DefaultMeter) meterOne).setState(MeterState.PENDING_REMOVE);
- // Let's create meter key
MeterKey meterKey = MeterKey.key(did1, mid1);
- // Delete meter
meterStore.deleteMeter(meterOne);
- // Start an async delete, simulating the operation of the provider
CompletableFuture<Void> future = CompletableFuture.runAsync(
- () -> meterStore.deleteMeterNow(meterOne)
+ () -> meterStore.purgeMeter(meterOne)
);
- // Let's wait
try {
future.get();
} catch (InterruptedException | ExecutionException e) {
e.printStackTrace();
}
- // Verify delete
assertThat(0, is(meterStore.getAllMeters().size()));
assertThat(0, is(meterStore.getAllMeters(did1).size()));
assertNull(meterStore.getMeter(meterKey));
- assertThat(mid1, is(meterStore.allocateMeterId(did1)));
+ assertThat(mid1, is(meterStore.allocateMeterId(did1, MeterScope.globalScope())));
}
/**
@@ -418,13 +354,10 @@
*/
@Test
public void testNoDeleteMeter() {
- // Init the store
initMeterStore(false);
- // Simulate the allocation of an id
- MeterId idOne = meterStore.allocateMeterId(did1);
- // Create the key
+
+ MeterCellId idOne = meterStore.allocateMeterId(did1, MeterScope.globalScope());
MeterKey keyOne = MeterKey.key(did1, idOne);
- // Let's create a meter
Meter meterOne = DefaultMeter.builder()
.forDevice(did1)
.fromApp(APP_ID)
@@ -432,11 +365,8 @@
.withUnit(Meter.Unit.KB_PER_SEC)
.withBands(Collections.singletonList(b1))
.build();
- // Set the state
((DefaultMeter) meterOne).setState(MeterState.PENDING_REMOVE);
- // Delete meter
meterStore.deleteMeter(meterOne);
- // Verify No delete
assertThat(0, is(meterStore.getAllMeters().size()));
assertThat(0, is(meterStore.getAllMeters(did1).size()));
assertNull(meterStore.getMeter(keyOne));
@@ -446,11 +376,10 @@
* Test purge meter.
*/
@Test
- public void testPurgeMeter() {
- // add the meter
+ public void testPurgeMeters() {
testStoreMeter();
- meterStore.purgeMeter(did1);
- // Verify delete
+
+ meterStore.purgeMeters(did1);
MeterKey keyOne = MeterKey.key(did1, mid1);
assertThat(0, is(meterStore.getAllMeters().size()));
assertThat(0, is(meterStore.getAllMeters(did1).size()));
@@ -461,20 +390,18 @@
* Test purge meter given device and application.
*/
@Test
- public void testPurgeMeterDeviceAndApp() {
- // Init the store
+ public void testPurgeMetersDeviceAndApp() {
initMeterStore(false);
- // add the meters
+
((DefaultMeter) m1).setState(MeterState.PENDING_ADD);
((DefaultMeter) m2).setState(MeterState.PENDING_ADD);
((DefaultMeter) m3).setState(MeterState.PENDING_ADD);
- meterStore.storeMeter(m1);
- meterStore.storeMeter(m2);
- meterStore.storeMeter(m3);
+ meterStore.addOrUpdateMeter(m1);
+ meterStore.addOrUpdateMeter(m2);
+ meterStore.addOrUpdateMeter(m3);
assertThat(3, is(meterStore.getAllMeters().size()));
meterStore.purgeMeters(did1, APP_ID_2);
- // Verify delete
MeterKey keyTwo = MeterKey.key(did1, mid2);
assertThat(2, is(meterStore.getAllMeters().size()));
assertThat(1, is(meterStore.getAllMeters(did1).size()));
@@ -487,14 +414,10 @@
*/
@Test
public void testGetMetersImmutability() {
- // Init the store
initMeterStore(false);
- // Simulate the allocation of an id
- MeterId idOne = meterStore.allocateMeterId(did1);
- // Verify the allocation
+ MeterCellId idOne = meterStore.allocateMeterId(did1, MeterScope.globalScope());
assertThat(mid1, is(idOne));
- // Let's create a meter
Meter meterOne = DefaultMeter.builder()
.forDevice(did1)
.fromApp(APP_ID)
@@ -502,21 +425,16 @@
.withUnit(Meter.Unit.KB_PER_SEC)
.withBands(Collections.singletonList(b1))
.build();
- // Set the state
((DefaultMeter) meterOne).setState(MeterState.PENDING_ADD);
- // Store the meter
- meterStore.storeMeter(meterOne);
+ meterStore.addOrUpdateMeter(meterOne);
- // Verify the immutability
Collection<Meter> meters = meterStore.getAllMeters();
Collection<Meter> metersDevice = meterStore.getAllMeters(did1);
assertThat(1, is(meters.size()));
assertThat(1, is(metersDevice.size()));
- MeterId idTwo = meterStore.allocateMeterId(did1);
- // Verify the allocation
+ MeterCellId idTwo = meterStore.allocateMeterId(did1, MeterScope.globalScope());
assertThat(mid2, is(idTwo));
- // Let's create a meter
Meter meterTwo = DefaultMeter.builder()
.forDevice(did1)
.fromApp(APP_ID)
@@ -524,14 +442,11 @@
.withUnit(Meter.Unit.KB_PER_SEC)
.withBands(Collections.singletonList(b1))
.build();
- // Set the state
((DefaultMeter) meterTwo).setState(MeterState.PENDING_ADD);
- // Store the meter
- meterStore.storeMeter(meterTwo);
+ meterStore.addOrUpdateMeter(meterTwo);
assertThat(1, is(meters.size()));
assertThat(1, is(metersDevice.size()));
-
meters = meterStore.getAllMeters();
metersDevice = meterStore.getAllMeters(did1);
assertThat(2, is(meters.size()));
@@ -544,6 +459,7 @@
@Test(expected = IllegalArgumentException.class)
public void testInvalidCellId() {
initMeterStore(true);
+
// MF defines an end index equals to 10
Meter meterBad = DefaultMeter.builder()
.forDevice(did3)
@@ -553,7 +469,7 @@
.withBands(Collections.singletonList(b1))
.build();
((DefaultMeter) meterBad).setState(MeterState.PENDING_ADD);
- meterStore.storeMeter(meterBad);
+ meterStore.addOrUpdateMeter(meterBad);
}
/**
@@ -562,6 +478,7 @@
@Test
public void testEnableUserDefinedIndex() {
initMeterStore(false);
+
assertTrue(meterStore.userDefinedIndexMode(true));
}
@@ -571,6 +488,7 @@
@Test
public void testInvalidEnableUserDefinedIndex() {
testStoreMeter();
+
assertFalse(meterStore.userDefinedIndexMode(true));
}
@@ -580,6 +498,7 @@
@Test
public void testDisableUserDefinedIndex() {
initMeterStore(true);
+
assertFalse(meterStore.userDefinedIndexMode(false));
}
@@ -589,7 +508,7 @@
@Test
public void testStoreMeterInUserDefinedIndexMode() {
initMeterStore(true);
- // Let's create a meter
+
Meter meterOne = DefaultMeter.builder()
.forDevice(did3)
.fromApp(APP_ID)
@@ -597,13 +516,9 @@
.withUnit(Meter.Unit.KB_PER_SEC)
.withBands(Collections.singletonList(b1))
.build();
- // Set the state
((DefaultMeter) meterOne).setState(MeterState.PENDING_ADD);
- // Store the meter
- meterStore.storeMeter(meterOne);
- // Let's create meter key
+ meterStore.addOrUpdateMeter(meterOne);
MeterKey meterKey = MeterKey.key(did3, cid4);
- // Verify the store
assertThat(1, is(meterStore.getAllMeters().size()));
assertThat(1, is(meterStore.getAllMeters(did3).size()));
assertThat(m4, is(meterStore.getMeter(meterKey)));
@@ -615,6 +530,7 @@
@Test
public void testInvalidDisableUserDefinedIndex() {
testStoreMeterInUserDefinedIndexMode();
+
assertTrue(meterStore.userDefinedIndexMode(false));
}
@@ -624,7 +540,8 @@
@Test
public void testAllocateIdInUserDefinedIndexMode() {
initMeterStore(true);
- assertNull(meterStore.allocateMeterId(did1));
+
+ assertNull(meterStore.allocateMeterId(did1, MeterScope.globalScope()));
}
/**
@@ -633,7 +550,7 @@
@Test
public void testFreeIdInUserMode() {
initMeterStore(true);
- // Free the id and expect not being available
+
meterStore.freeMeterId(did1, mid1);
MeterTableKey globalKey = MeterTableKey.key(did1, MeterScope.globalScope());
assertNotNull(meterStore.availableMeterIds.get(globalKey));
@@ -646,6 +563,7 @@
@Test
public void testDeleteMeterInUserDefinedIndexMode() {
initMeterStore(true);
+
Meter meterOne = DefaultMeter.builder()
.forDevice(did3)
.fromApp(APP_ID)
@@ -654,7 +572,7 @@
.withBands(Collections.singletonList(b1))
.build();
((DefaultMeter) meterOne).setState(MeterState.PENDING_ADD);
- meterStore.storeMeter(meterOne);
+ meterStore.addOrUpdateMeter(meterOne);
((DefaultMeter) meterOne).setState(MeterState.PENDING_REMOVE);
MeterKey meterKey = MeterKey.key(did3, cid4);
@@ -676,7 +594,6 @@
assertTrue(meterStore.availableMeterIds.get(globalKey).isEmpty());
}
- // Test class for driver service.
private class TestDriverService extends DriverServiceAdapter {
@Override
public DriverHandler createHandler(DeviceId deviceId, String... credentials) {
@@ -684,7 +601,6 @@
}
}
- // Test class for driver handler.
private class TestDriverHandler implements DriverHandler {
@Override
@@ -714,7 +630,6 @@
}
}
- // Test meter query
private class TestMeterQuery implements MeterQuery {
@Override