Committed by
Gerrit Code Review
adding device specific counters for meter ids in
the meter service. Change-Id: I38d38a0a85024927f5a74013b2b4d9efa9b32d22
Showing
7 changed files
with
137 additions
and
34 deletions
| ... | @@ -43,7 +43,8 @@ public final class DefaultMeterRequest implements MeterRequest { | ... | @@ -43,7 +43,8 @@ public final class DefaultMeterRequest implements MeterRequest { |
| 43 | 43 | ||
| 44 | private DefaultMeterRequest(DeviceId deviceId, ApplicationId appId, | 44 | private DefaultMeterRequest(DeviceId deviceId, ApplicationId appId, |
| 45 | Meter.Unit unit, boolean burst, | 45 | Meter.Unit unit, boolean burst, |
| 46 | - Collection<Band> bands, MeterContext context, Type op) { | 46 | + Collection<Band> bands, MeterContext context, |
| 47 | + Type op) { | ||
| 47 | this.deviceId = deviceId; | 48 | this.deviceId = deviceId; |
| 48 | this.appId = appId; | 49 | this.appId = appId; |
| 49 | this.unit = unit; | 50 | this.unit = unit; |
| ... | @@ -58,6 +59,7 @@ public final class DefaultMeterRequest implements MeterRequest { | ... | @@ -58,6 +59,7 @@ public final class DefaultMeterRequest implements MeterRequest { |
| 58 | return deviceId; | 59 | return deviceId; |
| 59 | } | 60 | } |
| 60 | 61 | ||
| 62 | + | ||
| 61 | @Override | 63 | @Override |
| 62 | public ApplicationId appId() { | 64 | public ApplicationId appId() { |
| 63 | return appId; | 65 | return appId; |
| ... | @@ -107,6 +109,7 @@ public final class DefaultMeterRequest implements MeterRequest { | ... | @@ -107,6 +109,7 @@ public final class DefaultMeterRequest implements MeterRequest { |
| 107 | private Collection<Band> bands; | 109 | private Collection<Band> bands; |
| 108 | private DeviceId deviceId; | 110 | private DeviceId deviceId; |
| 109 | private MeterContext context; | 111 | private MeterContext context; |
| 112 | + private Optional<MeterId> desiredId = Optional.empty(); | ||
| 110 | 113 | ||
| 111 | 114 | ||
| 112 | @Override | 115 | @Override | ... | ... |
| 1 | +/* | ||
| 2 | + * Copyright 2015 Open Networking Laboratory | ||
| 3 | + * | ||
| 4 | + * Licensed under the Apache License, Version 2.0 (the "License"); | ||
| 5 | + * you may not use this file except in compliance with the License. | ||
| 6 | + * You may obtain a copy of the License at | ||
| 7 | + * | ||
| 8 | + * http://www.apache.org/licenses/LICENSE-2.0 | ||
| 9 | + * | ||
| 10 | + * Unless required by applicable law or agreed to in writing, software | ||
| 11 | + * distributed under the License is distributed on an "AS IS" BASIS, | ||
| 12 | + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| 13 | + * See the License for the specific language governing permissions and | ||
| 14 | + * limitations under the License. | ||
| 15 | + */ | ||
| 16 | +package org.onosproject.net.meter; | ||
| 17 | + | ||
| 18 | +import com.google.common.base.Objects; | ||
| 19 | +import org.onosproject.net.DeviceId; | ||
| 20 | + | ||
| 21 | +/** | ||
| 22 | + * A meter key represents a meter uniquely. | ||
| 23 | + */ | ||
| 24 | +public final class MeterKey { | ||
| 25 | + | ||
| 26 | + private final DeviceId deviceId; | ||
| 27 | + private final MeterId id; | ||
| 28 | + | ||
| 29 | + private MeterKey(DeviceId deviceId, MeterId id) { | ||
| 30 | + this.deviceId = deviceId; | ||
| 31 | + this.id = id; | ||
| 32 | + } | ||
| 33 | + | ||
| 34 | + public DeviceId deviceId() { | ||
| 35 | + return deviceId; | ||
| 36 | + } | ||
| 37 | + | ||
| 38 | + public MeterId meterId() { | ||
| 39 | + return id; | ||
| 40 | + } | ||
| 41 | + | ||
| 42 | + @Override | ||
| 43 | + public boolean equals(Object o) { | ||
| 44 | + if (this == o) { | ||
| 45 | + return true; | ||
| 46 | + } | ||
| 47 | + if (o == null || getClass() != o.getClass()) { | ||
| 48 | + return false; | ||
| 49 | + } | ||
| 50 | + MeterKey meterKey = (MeterKey) o; | ||
| 51 | + return Objects.equal(deviceId, meterKey.deviceId) && | ||
| 52 | + Objects.equal(id, meterKey.id); | ||
| 53 | + } | ||
| 54 | + | ||
| 55 | + @Override | ||
| 56 | + public int hashCode() { | ||
| 57 | + return Objects.hashCode(deviceId, id); | ||
| 58 | + } | ||
| 59 | + | ||
| 60 | + public static MeterKey key(DeviceId deviceId, MeterId id) { | ||
| 61 | + return new MeterKey(deviceId, id); | ||
| 62 | + } | ||
| 63 | +} |
| ... | @@ -16,6 +16,7 @@ | ... | @@ -16,6 +16,7 @@ |
| 16 | package org.onosproject.net.meter; | 16 | package org.onosproject.net.meter; |
| 17 | 17 | ||
| 18 | import org.onosproject.event.ListenerService; | 18 | import org.onosproject.event.ListenerService; |
| 19 | +import org.onosproject.net.DeviceId; | ||
| 19 | 20 | ||
| 20 | import java.util.Collection; | 21 | import java.util.Collection; |
| 21 | 22 | ||
| ... | @@ -46,10 +47,11 @@ public interface MeterService | ... | @@ -46,10 +47,11 @@ public interface MeterService |
| 46 | /** | 47 | /** |
| 47 | * Fetch the meter by the meter id. | 48 | * Fetch the meter by the meter id. |
| 48 | * | 49 | * |
| 50 | + * @param deviceId a device id | ||
| 49 | * @param id a meter id | 51 | * @param id a meter id |
| 50 | * @return a meter | 52 | * @return a meter |
| 51 | */ | 53 | */ |
| 52 | - Meter getMeter(MeterId id); | 54 | + Meter getMeter(DeviceId deviceId, MeterId id); |
| 53 | 55 | ||
| 54 | /** | 56 | /** |
| 55 | * Fetches all the meters. | 57 | * Fetches all the meters. | ... | ... |
| ... | @@ -57,12 +57,12 @@ public interface MeterStore extends Store<MeterEvent, MeterStoreDelegate> { | ... | @@ -57,12 +57,12 @@ public interface MeterStore extends Store<MeterEvent, MeterStoreDelegate> { |
| 57 | void updateMeterState(Meter meter); | 57 | void updateMeterState(Meter meter); |
| 58 | 58 | ||
| 59 | /** | 59 | /** |
| 60 | - * Obtains a meter matching the given meter id. | 60 | + * Obtains a meter matching the given meter key. |
| 61 | * | 61 | * |
| 62 | - * @param meterId a meter id | 62 | + * @param key a meter key |
| 63 | * @return a meter | 63 | * @return a meter |
| 64 | */ | 64 | */ |
| 65 | - Meter getMeter(MeterId meterId); | 65 | + Meter getMeter(MeterKey key); |
| 66 | 66 | ||
| 67 | /** | 67 | /** |
| 68 | * Returns all meters stored in the store. | 68 | * Returns all meters stored in the store. | ... | ... |
| ... | @@ -15,6 +15,7 @@ | ... | @@ -15,6 +15,7 @@ |
| 15 | */ | 15 | */ |
| 16 | package org.onosproject.incubator.net.meter.impl; | 16 | package org.onosproject.incubator.net.meter.impl; |
| 17 | 17 | ||
| 18 | +import com.google.common.collect.Maps; | ||
| 18 | import org.apache.felix.scr.annotations.Activate; | 19 | import org.apache.felix.scr.annotations.Activate; |
| 19 | import org.apache.felix.scr.annotations.Component; | 20 | import org.apache.felix.scr.annotations.Component; |
| 20 | import org.apache.felix.scr.annotations.Deactivate; | 21 | import org.apache.felix.scr.annotations.Deactivate; |
| ... | @@ -27,6 +28,7 @@ import org.onosproject.net.meter.Meter; | ... | @@ -27,6 +28,7 @@ import org.onosproject.net.meter.Meter; |
| 27 | import org.onosproject.net.meter.MeterEvent; | 28 | import org.onosproject.net.meter.MeterEvent; |
| 28 | import org.onosproject.net.meter.MeterFailReason; | 29 | import org.onosproject.net.meter.MeterFailReason; |
| 29 | import org.onosproject.net.meter.MeterId; | 30 | import org.onosproject.net.meter.MeterId; |
| 31 | +import org.onosproject.net.meter.MeterKey; | ||
| 30 | import org.onosproject.net.meter.MeterListener; | 32 | import org.onosproject.net.meter.MeterListener; |
| 31 | import org.onosproject.net.meter.MeterOperation; | 33 | import org.onosproject.net.meter.MeterOperation; |
| 32 | import org.onosproject.net.meter.MeterProvider; | 34 | import org.onosproject.net.meter.MeterProvider; |
| ... | @@ -61,7 +63,7 @@ public class MeterManager extends AbstractListenerProviderRegistry<MeterEvent, M | ... | @@ -61,7 +63,7 @@ public class MeterManager extends AbstractListenerProviderRegistry<MeterEvent, M |
| 61 | MeterProvider, MeterProviderService> | 63 | MeterProvider, MeterProviderService> |
| 62 | implements MeterService, MeterProviderRegistry { | 64 | implements MeterService, MeterProviderRegistry { |
| 63 | 65 | ||
| 64 | - private final String meterIdentifier = "meter-id-counter"; | 66 | + private static final String METERCOUNTERIDENTIFIER = "meter-id-counter-%s"; |
| 65 | private final Logger log = getLogger(getClass()); | 67 | private final Logger log = getLogger(getClass()); |
| 66 | private final MeterStoreDelegate delegate = new InternalMeterStoreDelegate(); | 68 | private final MeterStoreDelegate delegate = new InternalMeterStoreDelegate(); |
| 67 | 69 | ||
| ... | @@ -71,15 +73,16 @@ public class MeterManager extends AbstractListenerProviderRegistry<MeterEvent, M | ... | @@ -71,15 +73,16 @@ public class MeterManager extends AbstractListenerProviderRegistry<MeterEvent, M |
| 71 | @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY) | 73 | @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY) |
| 72 | protected MeterStore store; | 74 | protected MeterStore store; |
| 73 | 75 | ||
| 74 | - private AtomicCounter meterIdCounter; | 76 | + private Map<DeviceId, AtomicCounter> meterIdCounters |
| 77 | + = Maps.newConcurrentMap(); | ||
| 75 | 78 | ||
| 76 | private TriConsumer<MeterRequest, MeterStoreResult, Throwable> onComplete; | 79 | private TriConsumer<MeterRequest, MeterStoreResult, Throwable> onComplete; |
| 77 | 80 | ||
| 78 | @Activate | 81 | @Activate |
| 79 | public void activate() { | 82 | public void activate() { |
| 80 | - meterIdCounter = storageService.atomicCounterBuilder() | 83 | + /*meterIdCounter = storageService.atomicCounterBuilder() |
| 81 | - .withName(meterIdentifier) | 84 | + .withName(METERCOUNTERIDENTIFIER) |
| 82 | - .build(); | 85 | + .build();*/ |
| 83 | 86 | ||
| 84 | store.setDelegate(delegate); | 87 | store.setDelegate(delegate); |
| 85 | 88 | ||
| ... | @@ -115,11 +118,13 @@ public class MeterManager extends AbstractListenerProviderRegistry<MeterEvent, M | ... | @@ -115,11 +118,13 @@ public class MeterManager extends AbstractListenerProviderRegistry<MeterEvent, M |
| 115 | @Override | 118 | @Override |
| 116 | public Meter submit(MeterRequest request) { | 119 | public Meter submit(MeterRequest request) { |
| 117 | 120 | ||
| 121 | + MeterId id = allocateMeterId(request.deviceId()); | ||
| 122 | + | ||
| 118 | Meter.Builder mBuilder = DefaultMeter.builder() | 123 | Meter.Builder mBuilder = DefaultMeter.builder() |
| 119 | .forDevice(request.deviceId()) | 124 | .forDevice(request.deviceId()) |
| 120 | .fromApp(request.appId()) | 125 | .fromApp(request.appId()) |
| 121 | .withBands(request.bands()) | 126 | .withBands(request.bands()) |
| 122 | - .withId(allocateMeterId()) | 127 | + .withId(id) |
| 123 | .withUnit(request.unit()); | 128 | .withUnit(request.unit()); |
| 124 | 129 | ||
| 125 | if (request.isBurst()) { | 130 | if (request.isBurst()) { |
| ... | @@ -152,8 +157,9 @@ public class MeterManager extends AbstractListenerProviderRegistry<MeterEvent, M | ... | @@ -152,8 +157,9 @@ public class MeterManager extends AbstractListenerProviderRegistry<MeterEvent, M |
| 152 | } | 157 | } |
| 153 | 158 | ||
| 154 | @Override | 159 | @Override |
| 155 | - public Meter getMeter(MeterId id) { | 160 | + public Meter getMeter(DeviceId deviceId, MeterId id) { |
| 156 | - return store.getMeter(id); | 161 | + MeterKey key = MeterKey.key(deviceId, id); |
| 162 | + return store.getMeter(key); | ||
| 157 | } | 163 | } |
| 158 | 164 | ||
| 159 | @Override | 165 | @Override |
| ... | @@ -161,9 +167,21 @@ public class MeterManager extends AbstractListenerProviderRegistry<MeterEvent, M | ... | @@ -161,9 +167,21 @@ public class MeterManager extends AbstractListenerProviderRegistry<MeterEvent, M |
| 161 | return store.getAllMeters(); | 167 | return store.getAllMeters(); |
| 162 | } | 168 | } |
| 163 | 169 | ||
| 164 | - private MeterId allocateMeterId() { | 170 | + private MeterId allocateMeterId(DeviceId deviceId) { |
| 165 | - // FIXME: This will break one day. | 171 | + long id = meterIdCounters.compute(deviceId, (k, v) -> { |
| 166 | - return MeterId.meterId((int) meterIdCounter.incrementAndGet()); | 172 | + if (v == null) { |
| 173 | + return allocateCounter(k); | ||
| 174 | + } | ||
| 175 | + return v; | ||
| 176 | + }).incrementAndGet(); | ||
| 177 | + | ||
| 178 | + return MeterId.meterId((int) id); | ||
| 179 | + } | ||
| 180 | + | ||
| 181 | + private AtomicCounter allocateCounter(DeviceId deviceId) { | ||
| 182 | + return storageService.atomicCounterBuilder() | ||
| 183 | + .withName(String.format(METERCOUNTERIDENTIFIER, deviceId)) | ||
| 184 | + .build(); | ||
| 167 | } | 185 | } |
| 168 | 186 | ||
| 169 | private class InternalMeterProviderService | 187 | private class InternalMeterProviderService | ... | ... |
| ... | @@ -130,7 +130,7 @@ public class MeterManagerTest { | ... | @@ -130,7 +130,7 @@ public class MeterManagerTest { |
| 130 | m2 = DefaultMeter.builder() | 130 | m2 = DefaultMeter.builder() |
| 131 | .forDevice(did("2")) | 131 | .forDevice(did("2")) |
| 132 | .fromApp(APP_ID) | 132 | .fromApp(APP_ID) |
| 133 | - .withId(MeterId.meterId(2)) | 133 | + .withId(MeterId.meterId(1)) |
| 134 | .withUnit(Meter.Unit.KB_PER_SEC) | 134 | .withUnit(Meter.Unit.KB_PER_SEC) |
| 135 | .withBands(Collections.singletonList(band)) | 135 | .withBands(Collections.singletonList(band)) |
| 136 | .build(); | 136 | .build(); |
| ... | @@ -167,7 +167,7 @@ public class MeterManagerTest { | ... | @@ -167,7 +167,7 @@ public class MeterManagerTest { |
| 167 | 167 | ||
| 168 | assertTrue("The meter was not added", manager.getAllMeters().size() == 1); | 168 | assertTrue("The meter was not added", manager.getAllMeters().size() == 1); |
| 169 | 169 | ||
| 170 | - assertThat(manager.getMeter(MeterId.meterId(1)), is(m1)); | 170 | + assertThat(manager.getMeter(did("1"), MeterId.meterId(1)), is(m1)); |
| 171 | } | 171 | } |
| 172 | 172 | ||
| 173 | @Test | 173 | @Test |
| ... | @@ -175,7 +175,7 @@ public class MeterManagerTest { | ... | @@ -175,7 +175,7 @@ public class MeterManagerTest { |
| 175 | manager.submit(m1Request.add()); | 175 | manager.submit(m1Request.add()); |
| 176 | manager.withdraw(m1Request.remove(), m1.id()); | 176 | manager.withdraw(m1Request.remove(), m1.id()); |
| 177 | 177 | ||
| 178 | - assertThat(manager.getMeter(MeterId.meterId(1)).state(), | 178 | + assertThat(manager.getMeter(did("1"), MeterId.meterId(1)).state(), |
| 179 | is(MeterState.PENDING_REMOVE)); | 179 | is(MeterState.PENDING_REMOVE)); |
| 180 | 180 | ||
| 181 | providerService.pushMeterMetrics(m1.deviceId(), Collections.emptyList()); | 181 | providerService.pushMeterMetrics(m1.deviceId(), Collections.emptyList()); |
| ... | @@ -184,7 +184,16 @@ public class MeterManagerTest { | ... | @@ -184,7 +184,16 @@ public class MeterManagerTest { |
| 184 | 184 | ||
| 185 | } | 185 | } |
| 186 | 186 | ||
| 187 | + @Test | ||
| 188 | + public void testMultipleDevice() { | ||
| 189 | + manager.submit(m1Request.add()); | ||
| 190 | + manager.submit(m2Request.add()); | ||
| 187 | 191 | ||
| 192 | + assertTrue("The meters were not added", manager.getAllMeters().size() == 2); | ||
| 193 | + | ||
| 194 | + assertThat(manager.getMeter(did("1"), MeterId.meterId(1)), is(m1)); | ||
| 195 | + assertThat(manager.getMeter(did("2"), MeterId.meterId(1)), is(m2)); | ||
| 196 | + } | ||
| 188 | 197 | ||
| 189 | public class TestApplicationId extends DefaultApplicationId { | 198 | public class TestApplicationId extends DefaultApplicationId { |
| 190 | public TestApplicationId(int id, String name) { | 199 | public TestApplicationId(int id, String name) { | ... | ... |
| ... | @@ -33,6 +33,7 @@ import org.onosproject.net.meter.Meter; | ... | @@ -33,6 +33,7 @@ import org.onosproject.net.meter.Meter; |
| 33 | import org.onosproject.net.meter.MeterEvent; | 33 | import org.onosproject.net.meter.MeterEvent; |
| 34 | import org.onosproject.net.meter.MeterFailReason; | 34 | import org.onosproject.net.meter.MeterFailReason; |
| 35 | import org.onosproject.net.meter.MeterId; | 35 | import org.onosproject.net.meter.MeterId; |
| 36 | +import org.onosproject.net.meter.MeterKey; | ||
| 36 | import org.onosproject.net.meter.MeterOperation; | 37 | import org.onosproject.net.meter.MeterOperation; |
| 37 | import org.onosproject.net.meter.MeterState; | 38 | import org.onosproject.net.meter.MeterState; |
| 38 | import org.onosproject.net.meter.MeterStore; | 39 | import org.onosproject.net.meter.MeterStore; |
| ... | @@ -78,12 +79,12 @@ public class DistributedMeterStore extends AbstractStore<MeterEvent, MeterStoreD | ... | @@ -78,12 +79,12 @@ public class DistributedMeterStore extends AbstractStore<MeterEvent, MeterStoreD |
| 78 | @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY) | 79 | @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY) |
| 79 | private ClusterService clusterService; | 80 | private ClusterService clusterService; |
| 80 | 81 | ||
| 81 | - private ConsistentMap<MeterId, MeterData> meters; | 82 | + private ConsistentMap<MeterKey, MeterData> meters; |
| 82 | private NodeId local; | 83 | private NodeId local; |
| 83 | 84 | ||
| 84 | private MapEventListener mapListener = new InternalMapEventListener(); | 85 | private MapEventListener mapListener = new InternalMapEventListener(); |
| 85 | 86 | ||
| 86 | - private Map<MeterId, CompletableFuture<MeterStoreResult>> futures = | 87 | + private Map<MeterKey, CompletableFuture<MeterStoreResult>> futures = |
| 87 | Maps.newConcurrentMap(); | 88 | Maps.newConcurrentMap(); |
| 88 | 89 | ||
| 89 | @Activate | 90 | @Activate |
| ... | @@ -92,9 +93,10 @@ public class DistributedMeterStore extends AbstractStore<MeterEvent, MeterStoreD | ... | @@ -92,9 +93,10 @@ public class DistributedMeterStore extends AbstractStore<MeterEvent, MeterStoreD |
| 92 | local = clusterService.getLocalNode().id(); | 93 | local = clusterService.getLocalNode().id(); |
| 93 | 94 | ||
| 94 | 95 | ||
| 95 | - meters = storageService.<MeterId, MeterData>consistentMapBuilder() | 96 | + meters = storageService.<MeterKey, MeterData>consistentMapBuilder() |
| 96 | .withName(METERSTORE) | 97 | .withName(METERSTORE) |
| 97 | .withSerializer(Serializer.using(Arrays.asList(KryoNamespaces.API), | 98 | .withSerializer(Serializer.using(Arrays.asList(KryoNamespaces.API), |
| 99 | + MeterKey.class, | ||
| 98 | MeterData.class, | 100 | MeterData.class, |
| 99 | DefaultMeter.class, | 101 | DefaultMeter.class, |
| 100 | DefaultBand.class, | 102 | DefaultBand.class, |
| ... | @@ -120,11 +122,12 @@ public class DistributedMeterStore extends AbstractStore<MeterEvent, MeterStoreD | ... | @@ -120,11 +122,12 @@ public class DistributedMeterStore extends AbstractStore<MeterEvent, MeterStoreD |
| 120 | @Override | 122 | @Override |
| 121 | public CompletableFuture<MeterStoreResult> storeMeter(Meter meter) { | 123 | public CompletableFuture<MeterStoreResult> storeMeter(Meter meter) { |
| 122 | CompletableFuture<MeterStoreResult> future = new CompletableFuture<>(); | 124 | CompletableFuture<MeterStoreResult> future = new CompletableFuture<>(); |
| 123 | - futures.put(meter.id(), future); | 125 | + MeterKey key = MeterKey.key(meter.deviceId(), meter.id()); |
| 126 | + futures.put(key, future); | ||
| 124 | MeterData data = new MeterData(meter, null, local); | 127 | MeterData data = new MeterData(meter, null, local); |
| 125 | 128 | ||
| 126 | try { | 129 | try { |
| 127 | - meters.put(meter.id(), data); | 130 | + meters.put(key, data); |
| 128 | } catch (StorageException e) { | 131 | } catch (StorageException e) { |
| 129 | future.completeExceptionally(e); | 132 | future.completeExceptionally(e); |
| 130 | } | 133 | } |
| ... | @@ -136,14 +139,15 @@ public class DistributedMeterStore extends AbstractStore<MeterEvent, MeterStoreD | ... | @@ -136,14 +139,15 @@ public class DistributedMeterStore extends AbstractStore<MeterEvent, MeterStoreD |
| 136 | @Override | 139 | @Override |
| 137 | public CompletableFuture<MeterStoreResult> deleteMeter(Meter meter) { | 140 | public CompletableFuture<MeterStoreResult> deleteMeter(Meter meter) { |
| 138 | CompletableFuture<MeterStoreResult> future = new CompletableFuture<>(); | 141 | CompletableFuture<MeterStoreResult> future = new CompletableFuture<>(); |
| 139 | - futures.put(meter.id(), future); | 142 | + MeterKey key = MeterKey.key(meter.deviceId(), meter.id()); |
| 143 | + futures.put(key, future); | ||
| 140 | 144 | ||
| 141 | MeterData data = new MeterData(meter, null, local); | 145 | MeterData data = new MeterData(meter, null, local); |
| 142 | 146 | ||
| 143 | // update the state of the meter. It will be pruned by observing | 147 | // update the state of the meter. It will be pruned by observing |
| 144 | // that it has been removed from the dataplane. | 148 | // that it has been removed from the dataplane. |
| 145 | try { | 149 | try { |
| 146 | - if (meters.computeIfPresent(meter.id(), (k, v) -> data) == null) { | 150 | + if (meters.computeIfPresent(key, (k, v) -> data) == null) { |
| 147 | future.complete(MeterStoreResult.success()); | 151 | future.complete(MeterStoreResult.success()); |
| 148 | } | 152 | } |
| 149 | } catch (StorageException e) { | 153 | } catch (StorageException e) { |
| ... | @@ -157,11 +161,12 @@ public class DistributedMeterStore extends AbstractStore<MeterEvent, MeterStoreD | ... | @@ -157,11 +161,12 @@ public class DistributedMeterStore extends AbstractStore<MeterEvent, MeterStoreD |
| 157 | @Override | 161 | @Override |
| 158 | public CompletableFuture<MeterStoreResult> updateMeter(Meter meter) { | 162 | public CompletableFuture<MeterStoreResult> updateMeter(Meter meter) { |
| 159 | CompletableFuture<MeterStoreResult> future = new CompletableFuture<>(); | 163 | CompletableFuture<MeterStoreResult> future = new CompletableFuture<>(); |
| 160 | - futures.put(meter.id(), future); | 164 | + MeterKey key = MeterKey.key(meter.deviceId(), meter.id()); |
| 165 | + futures.put(key, future); | ||
| 161 | 166 | ||
| 162 | MeterData data = new MeterData(meter, null, local); | 167 | MeterData data = new MeterData(meter, null, local); |
| 163 | try { | 168 | try { |
| 164 | - if (meters.computeIfPresent(meter.id(), (k, v) -> data) == null) { | 169 | + if (meters.computeIfPresent(key, (k, v) -> data) == null) { |
| 165 | future.complete(MeterStoreResult.fail(MeterFailReason.INVALID_METER)); | 170 | future.complete(MeterStoreResult.fail(MeterFailReason.INVALID_METER)); |
| 166 | } | 171 | } |
| 167 | } catch (StorageException e) { | 172 | } catch (StorageException e) { |
| ... | @@ -172,7 +177,8 @@ public class DistributedMeterStore extends AbstractStore<MeterEvent, MeterStoreD | ... | @@ -172,7 +177,8 @@ public class DistributedMeterStore extends AbstractStore<MeterEvent, MeterStoreD |
| 172 | 177 | ||
| 173 | @Override | 178 | @Override |
| 174 | public void updateMeterState(Meter meter) { | 179 | public void updateMeterState(Meter meter) { |
| 175 | - meters.computeIfPresent(meter.id(), (id, v) -> { | 180 | + MeterKey key = MeterKey.key(meter.deviceId(), meter.id()); |
| 181 | + meters.computeIfPresent(key, (k, v) -> { | ||
| 176 | DefaultMeter m = (DefaultMeter) v.meter(); | 182 | DefaultMeter m = (DefaultMeter) v.meter(); |
| 177 | m.setState(meter.state()); | 183 | m.setState(meter.state()); |
| 178 | m.setProcessedPackets(meter.packetsSeen()); | 184 | m.setProcessedPackets(meter.packetsSeen()); |
| ... | @@ -185,8 +191,8 @@ public class DistributedMeterStore extends AbstractStore<MeterEvent, MeterStoreD | ... | @@ -185,8 +191,8 @@ public class DistributedMeterStore extends AbstractStore<MeterEvent, MeterStoreD |
| 185 | } | 191 | } |
| 186 | 192 | ||
| 187 | @Override | 193 | @Override |
| 188 | - public Meter getMeter(MeterId meterId) { | 194 | + public Meter getMeter(MeterKey key) { |
| 189 | - MeterData data = Versioned.valueOrElse(meters.get(meterId), null); | 195 | + MeterData data = Versioned.valueOrElse(meters.get(key), null); |
| 190 | return data == null ? null : data.meter(); | 196 | return data == null ? null : data.meter(); |
| 191 | } | 197 | } |
| 192 | 198 | ||
| ... | @@ -198,14 +204,16 @@ public class DistributedMeterStore extends AbstractStore<MeterEvent, MeterStoreD | ... | @@ -198,14 +204,16 @@ public class DistributedMeterStore extends AbstractStore<MeterEvent, MeterStoreD |
| 198 | 204 | ||
| 199 | @Override | 205 | @Override |
| 200 | public void failedMeter(MeterOperation op, MeterFailReason reason) { | 206 | public void failedMeter(MeterOperation op, MeterFailReason reason) { |
| 201 | - meters.computeIfPresent(op.meter().id(), (k, v) -> | 207 | + MeterKey key = MeterKey.key(op.meter().deviceId(), op.meter().id()); |
| 208 | + meters.computeIfPresent(key, (k, v) -> | ||
| 202 | new MeterData(v.meter(), reason, v.origin())); | 209 | new MeterData(v.meter(), reason, v.origin())); |
| 203 | } | 210 | } |
| 204 | 211 | ||
| 205 | @Override | 212 | @Override |
| 206 | public void deleteMeterNow(Meter m) { | 213 | public void deleteMeterNow(Meter m) { |
| 207 | - futures.remove(m.id()); | 214 | + MeterKey key = MeterKey.key(m.deviceId(), m.id()); |
| 208 | - meters.remove(m.id()); | 215 | + futures.remove(key); |
| 216 | + meters.remove(key); | ||
| 209 | } | 217 | } |
| 210 | 218 | ||
| 211 | private class InternalMapEventListener implements MapEventListener<MeterId, MeterData> { | 219 | private class InternalMapEventListener implements MapEventListener<MeterId, MeterData> { | ... | ... |
-
Please register or login to post a comment