2 * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
9 package org.opendaylight.openflowplugin.impl.services;
11 import com.google.common.base.Function;
12 import com.google.common.base.Preconditions;
13 import com.google.common.collect.Iterables;
14 import com.google.common.util.concurrent.Futures;
15 import com.google.common.util.concurrent.JdkFutureAdapters;
16 import com.google.common.util.concurrent.ListenableFuture;
17 import java.util.ArrayList;
18 import java.util.List;
19 import java.util.concurrent.Future;
20 import javax.annotation.Nullable;
21 import org.opendaylight.openflowplugin.impl.util.BarrierUtil;
22 import org.opendaylight.openflowplugin.impl.util.MeterUtil;
23 import org.opendaylight.openflowplugin.impl.util.PathUtil;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.transaction.rev150304.FlowCapableTransactionService;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeRef;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.AddMeterInput;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.AddMeterInputBuilder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.AddMeterOutput;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.RemoveMeterInput;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.RemoveMeterInputBuilder;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.RemoveMeterOutput;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.SalMeterService;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.UpdateMeterInput;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.UpdateMeterInputBuilder;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.UpdateMeterOutput;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.meter.update.OriginalMeterBuilder;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.meter.update.UpdatedMeterBuilder;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.Meter;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.MeterRef;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.AddMetersBatchInput;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.AddMetersBatchOutput;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.RemoveMetersBatchInput;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.RemoveMetersBatchOutput;
45 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.SalMetersBatchService;
46 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.UpdateMetersBatchInput;
47 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.UpdateMetersBatchOutput;
48 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.add.meters.batch.input.BatchAddMeters;
49 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.batch.meter.output.list.grouping.BatchFailedMetersOutput;
50 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.remove.meters.batch.input.BatchRemoveMeters;
51 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.update.meters.batch.input.BatchUpdateMeters;
52 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
53 import org.opendaylight.yangtools.yang.common.RpcResult;
54 import org.slf4j.Logger;
55 import org.slf4j.LoggerFactory;
58 * default implementation of {@link SalMetersBatchService} - delegates work to {@link SalMeterService}
60 public class SalMetersBatchServiceImpl implements SalMetersBatchService {
62 private static final Logger LOG = LoggerFactory.getLogger(SalMetersBatchServiceImpl.class);
64 private final SalMeterService salMeterService;
65 private final FlowCapableTransactionService transactionService;
67 public SalMetersBatchServiceImpl(final SalMeterService salMeterService, final FlowCapableTransactionService transactionService) {
68 this.salMeterService = Preconditions.checkNotNull(salMeterService);
69 this.transactionService = Preconditions.checkNotNull(transactionService);
73 public Future<RpcResult<UpdateMetersBatchOutput>> updateMetersBatch(final UpdateMetersBatchInput input) {
74 final List<BatchUpdateMeters> batchUpdateMeters = input.getBatchUpdateMeters();
75 LOG.trace("Updating meters @ {} : {}", PathUtil.extractNodeId(input.getNode()), batchUpdateMeters.size());
77 final ArrayList<ListenableFuture<RpcResult<UpdateMeterOutput>>> resultsLot = new ArrayList<>();
78 for (BatchUpdateMeters batchMeter : batchUpdateMeters) {
79 final UpdateMeterInput updateMeterInput = new UpdateMeterInputBuilder(input)
80 .setOriginalMeter(new OriginalMeterBuilder(batchMeter.getOriginalBatchedMeter()).build())
81 .setUpdatedMeter(new UpdatedMeterBuilder(batchMeter.getUpdatedBatchedMeter()).build())
82 .setMeterRef(createMeterRef(input.getNode(), batchMeter))
83 .setNode(input.getNode())
85 resultsLot.add(JdkFutureAdapters.listenInPoolThread(salMeterService.updateMeter(updateMeterInput)));
88 final Iterable<Meter> meters = Iterables.transform(batchUpdateMeters, new Function<BatchUpdateMeters, Meter>() {
91 public Meter apply(@Nullable final BatchUpdateMeters input) {
92 return input.getUpdatedBatchedMeter();
97 final ListenableFuture<RpcResult<List<BatchFailedMetersOutput>>> commonResult =
98 Futures.transform(Futures.allAsList(resultsLot), MeterUtil.<UpdateMeterOutput>createCumulativeFunction(
99 meters, batchUpdateMeters.size()));
101 ListenableFuture<RpcResult<UpdateMetersBatchOutput>> updateMetersBulkFuture =
102 Futures.transform(commonResult, MeterUtil.METER_UPDATE_TRANSFORM);
104 if (input.isBarrierAfter()) {
105 updateMetersBulkFuture = BarrierUtil.chainBarrier(updateMetersBulkFuture, input.getNode(),
106 transactionService, MeterUtil.METER_UPDATE_COMPOSING_TRANSFORM);
109 return updateMetersBulkFuture;
113 public Future<RpcResult<AddMetersBatchOutput>> addMetersBatch(final AddMetersBatchInput input) {
114 LOG.trace("Adding meters @ {} : {}", PathUtil.extractNodeId(input.getNode()), input.getBatchAddMeters().size());
115 final ArrayList<ListenableFuture<RpcResult<AddMeterOutput>>> resultsLot = new ArrayList<>();
116 for (BatchAddMeters addMeter : input.getBatchAddMeters()) {
117 final AddMeterInput addMeterInput = new AddMeterInputBuilder(addMeter)
118 .setMeterRef(createMeterRef(input.getNode(), addMeter))
119 .setNode(input.getNode())
121 resultsLot.add(JdkFutureAdapters.listenInPoolThread(salMeterService.addMeter(addMeterInput)));
124 final ListenableFuture<RpcResult<List<BatchFailedMetersOutput>>> commonResult =
125 Futures.transform(Futures.allAsList(resultsLot),
126 MeterUtil.<AddMeterOutput>createCumulativeFunction(input.getBatchAddMeters()));
128 ListenableFuture<RpcResult<AddMetersBatchOutput>> addMetersBulkFuture =
129 Futures.transform(commonResult, MeterUtil.METER_ADD_TRANSFORM);
131 if (input.isBarrierAfter()) {
132 addMetersBulkFuture = BarrierUtil.chainBarrier(addMetersBulkFuture, input.getNode(),
133 transactionService, MeterUtil.METER_ADD_COMPOSING_TRANSFORM);
136 return addMetersBulkFuture;
140 public Future<RpcResult<RemoveMetersBatchOutput>> removeMetersBatch(final RemoveMetersBatchInput input) {
141 LOG.trace("Removing meters @ {} : {}", PathUtil.extractNodeId(input.getNode()), input.getBatchRemoveMeters().size());
142 final ArrayList<ListenableFuture<RpcResult<RemoveMeterOutput>>> resultsLot = new ArrayList<>();
143 for (BatchRemoveMeters addMeter : input.getBatchRemoveMeters()) {
144 final RemoveMeterInput removeMeterInput = new RemoveMeterInputBuilder(addMeter)
145 .setMeterRef(createMeterRef(input.getNode(), addMeter))
146 .setNode(input.getNode())
148 resultsLot.add(JdkFutureAdapters.listenInPoolThread(salMeterService.removeMeter(removeMeterInput)));
151 final ListenableFuture<RpcResult<List<BatchFailedMetersOutput>>> commonResult =
152 Futures.transform(Futures.allAsList(resultsLot),
153 MeterUtil.<RemoveMeterOutput>createCumulativeFunction(input.getBatchRemoveMeters()));
155 ListenableFuture<RpcResult<RemoveMetersBatchOutput>> removeMetersBulkFuture =
156 Futures.transform(commonResult, MeterUtil.METER_REMOVE_TRANSFORM);
158 if (input.isBarrierAfter()) {
159 removeMetersBulkFuture = BarrierUtil.chainBarrier(removeMetersBulkFuture, input.getNode(),
160 transactionService, MeterUtil.METER_REMOVE_COMPOSING_TRANSFORM);
163 return removeMetersBulkFuture;
166 private static MeterRef createMeterRef(final NodeRef nodeRef, final Meter batchMeter) {
167 return MeterUtil.buildMeterPath((InstanceIdentifier<Node>) nodeRef.getValue(), batchMeter.getMeterId());
170 private static MeterRef createMeterRef(final NodeRef nodeRef, final BatchUpdateMeters batchMeter) {
171 return MeterUtil.buildMeterPath((InstanceIdentifier<Node>) nodeRef.getValue(),
172 batchMeter.getUpdatedBatchedMeter().getMeterId());