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
8 package org.opendaylight.openflowplugin.impl.services.sal;
10 import static java.util.Objects.requireNonNull;
12 import com.google.common.util.concurrent.Futures;
13 import com.google.common.util.concurrent.ListenableFuture;
14 import com.google.common.util.concurrent.MoreExecutors;
15 import java.util.ArrayList;
16 import java.util.List;
17 import java.util.stream.Collectors;
18 import org.opendaylight.openflowplugin.impl.util.BarrierUtil;
19 import org.opendaylight.openflowplugin.impl.util.MeterUtil;
20 import org.opendaylight.openflowplugin.impl.util.PathUtil;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.transaction.rev150304.FlowCapableTransactionService;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeRef;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.AddMeterInput;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.AddMeterInputBuilder;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.AddMeterOutput;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.RemoveMeterInput;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.RemoveMeterInputBuilder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.RemoveMeterOutput;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.SalMeterService;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.UpdateMeterInput;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.UpdateMeterInputBuilder;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.UpdateMeterOutput;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.meter.update.OriginalMeterBuilder;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.meter.update.UpdatedMeterBuilder;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.Meter;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.MeterRef;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.AddMetersBatchInput;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.AddMetersBatchOutput;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.BatchMeterInputUpdateGrouping;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.RemoveMetersBatchInput;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.RemoveMetersBatchOutput;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.SalMetersBatchService;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.UpdateMetersBatchInput;
45 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.UpdateMetersBatchOutput;
46 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.add.meters.batch.input.BatchAddMeters;
47 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.batch.meter.output.list.grouping.BatchFailedMetersOutput;
48 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.remove.meters.batch.input.BatchRemoveMeters;
49 import org.opendaylight.yang.gen.v1.urn.opendaylight.meters.service.rev160316.update.meters.batch.input.BatchUpdateMeters;
50 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
51 import org.opendaylight.yangtools.yang.common.RpcResult;
52 import org.slf4j.Logger;
53 import org.slf4j.LoggerFactory;
56 * Default implementation of {@link SalMetersBatchService} - delegates work to {@link SalMeterService}.
58 public class SalMetersBatchServiceImpl implements SalMetersBatchService {
59 private static final Logger LOG = LoggerFactory.getLogger(SalMetersBatchServiceImpl.class);
61 private final SalMeterService salMeterService;
62 private final FlowCapableTransactionService transactionService;
64 public SalMetersBatchServiceImpl(final SalMeterService salMeterService,
65 final FlowCapableTransactionService transactionService) {
66 this.salMeterService = requireNonNull(salMeterService);
67 this.transactionService = requireNonNull(transactionService);
71 public ListenableFuture<RpcResult<UpdateMetersBatchOutput>> updateMetersBatch(final UpdateMetersBatchInput input) {
72 final List<BatchUpdateMeters> batchUpdateMeters = input.getBatchUpdateMeters();
73 LOG.trace("Updating meters @ {} : {}", PathUtil.extractNodeId(input.getNode()), batchUpdateMeters.size());
75 final ArrayList<ListenableFuture<RpcResult<UpdateMeterOutput>>> resultsLot = new ArrayList<>();
76 for (BatchUpdateMeters batchMeter : batchUpdateMeters) {
77 final UpdateMeterInput updateMeterInput = new UpdateMeterInputBuilder(input)
78 .setOriginalMeter(new OriginalMeterBuilder(batchMeter.getOriginalBatchedMeter()).build())
79 .setUpdatedMeter(new UpdatedMeterBuilder(batchMeter.getUpdatedBatchedMeter()).build())
80 .setMeterRef(createMeterRef(input.getNode(), batchMeter))
81 .setNode(input.getNode())
83 resultsLot.add(salMeterService.updateMeter(updateMeterInput));
86 final Iterable<Meter> meters = batchUpdateMeters.stream()
87 .map(BatchMeterInputUpdateGrouping::getUpdatedBatchedMeter)
88 .collect(Collectors.toList());
90 final ListenableFuture<RpcResult<List<BatchFailedMetersOutput>>> commonResult =
91 Futures.transform(Futures.allAsList(resultsLot),
92 MeterUtil.createCumulativeFunction(meters, batchUpdateMeters.size()),
93 MoreExecutors.directExecutor());
95 ListenableFuture<RpcResult<UpdateMetersBatchOutput>> updateMetersBulkFuture =
96 Futures.transform(commonResult, MeterUtil.METER_UPDATE_TRANSFORM, MoreExecutors.directExecutor());
98 if (input.getBarrierAfter()) {
99 updateMetersBulkFuture = BarrierUtil.chainBarrier(updateMetersBulkFuture, input.getNode(),
100 transactionService, MeterUtil.METER_UPDATE_COMPOSING_TRANSFORM);
103 return updateMetersBulkFuture;
107 public ListenableFuture<RpcResult<AddMetersBatchOutput>> addMetersBatch(final AddMetersBatchInput input) {
108 LOG.trace("Adding meters @ {} : {}", PathUtil.extractNodeId(input.getNode()), input.getBatchAddMeters().size());
109 final ArrayList<ListenableFuture<RpcResult<AddMeterOutput>>> resultsLot = new ArrayList<>();
110 for (BatchAddMeters addMeter : input.nonnullBatchAddMeters().values()) {
111 final AddMeterInput addMeterInput = new AddMeterInputBuilder(addMeter)
112 .setMeterRef(createMeterRef(input.getNode(), addMeter))
113 .setNode(input.getNode())
115 resultsLot.add(salMeterService.addMeter(addMeterInput));
118 final ListenableFuture<RpcResult<List<BatchFailedMetersOutput>>> commonResult =
119 Futures.transform(Futures.allAsList(resultsLot),
120 MeterUtil.createCumulativeFunction(input.nonnullBatchAddMeters().values()),
121 MoreExecutors.directExecutor());
123 ListenableFuture<RpcResult<AddMetersBatchOutput>> addMetersBulkFuture =
124 Futures.transform(commonResult, MeterUtil.METER_ADD_TRANSFORM, MoreExecutors.directExecutor());
126 if (input.getBarrierAfter()) {
127 addMetersBulkFuture = BarrierUtil.chainBarrier(addMetersBulkFuture, input.getNode(),
128 transactionService, MeterUtil.METER_ADD_COMPOSING_TRANSFORM);
131 return addMetersBulkFuture;
135 public ListenableFuture<RpcResult<RemoveMetersBatchOutput>> removeMetersBatch(final RemoveMetersBatchInput input) {
136 LOG.trace("Removing meters @ {} : {}",
137 PathUtil.extractNodeId(input.getNode()),
138 input.getBatchRemoveMeters().size());
139 final ArrayList<ListenableFuture<RpcResult<RemoveMeterOutput>>> resultsLot = new ArrayList<>();
140 for (BatchRemoveMeters addMeter : input.nonnullBatchRemoveMeters().values()) {
141 final RemoveMeterInput removeMeterInput = new RemoveMeterInputBuilder(addMeter)
142 .setMeterRef(createMeterRef(input.getNode(), addMeter))
143 .setNode(input.getNode())
145 resultsLot.add(salMeterService.removeMeter(removeMeterInput));
148 final ListenableFuture<RpcResult<List<BatchFailedMetersOutput>>> commonResult =
149 Futures.transform(Futures.allAsList(resultsLot),
150 MeterUtil.createCumulativeFunction(input.nonnullBatchRemoveMeters().values()),
151 MoreExecutors.directExecutor());
153 ListenableFuture<RpcResult<RemoveMetersBatchOutput>> removeMetersBulkFuture =
154 Futures.transform(commonResult, MeterUtil.METER_REMOVE_TRANSFORM, MoreExecutors.directExecutor());
156 if (input.getBarrierAfter()) {
157 removeMetersBulkFuture = BarrierUtil.chainBarrier(removeMetersBulkFuture, input.getNode(),
158 transactionService, MeterUtil.METER_REMOVE_COMPOSING_TRANSFORM);
161 return removeMetersBulkFuture;
164 private static MeterRef createMeterRef(final NodeRef nodeRef, final Meter batchMeter) {
165 return MeterUtil.buildMeterPath((InstanceIdentifier<Node>) nodeRef.getValue(), batchMeter.getMeterId());
168 private static MeterRef createMeterRef(final NodeRef nodeRef, final BatchUpdateMeters batchMeter) {
169 return MeterUtil.buildMeterPath((InstanceIdentifier<Node>) nodeRef.getValue(),
170 batchMeter.getUpdatedBatchedMeter().getMeterId());