d633bc53cf83c53a48a06bbc547ab4db230478e7
[bgpcep.git] / pcep / tunnel / tunnel-provider / src / main / java / org / opendaylight / bgpcep / pcep / tunnel / provider / UpdateTunnelInstructionExecutor.java
1 /*
2  * Copyright (c) 2015 Cisco Systems, Inc. and others.  All rights reserved.
3  *
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
7  */
8 package org.opendaylight.bgpcep.pcep.tunnel.provider;
9
10 import static java.util.Objects.requireNonNull;
11
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.concurrent.ExecutionException;
16 import org.opendaylight.bgpcep.pcep.topology.spi.AbstractInstructionExecutor;
17 import org.opendaylight.bgpcep.programming.topology.TopologyProgrammingUtil;
18 import org.opendaylight.bgpcep.programming.tunnel.TunnelProgrammingUtil;
19 import org.opendaylight.mdsal.binding.api.DataBroker;
20 import org.opendaylight.mdsal.binding.api.ReadTransaction;
21 import org.opendaylight.mdsal.binding.api.RpcConsumerRegistry;
22 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev200720.AdministrativeStatus;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev200720.Arguments3Builder;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev200720.PcepUpdateTunnelInput1;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev200720.lsp.object.LspBuilder;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.bandwidth.object.BandwidthBuilder;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.classtype.object.ClassTypeBuilder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.lspa.object.LspaBuilder;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev220730.OperationResult;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev220730.UpdateLsp;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev220730.UpdateLspInput;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev220730.UpdateLspInputBuilder;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.pcep.programming.rev181109.PcepUpdateTunnelInput;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.pcep.rev181109.Link1;
36 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
37 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Link;
38 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
39 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
40 import org.opendaylight.yangtools.yang.common.RpcResult;
41 import org.slf4j.Logger;
42 import org.slf4j.LoggerFactory;
43
44 final class UpdateTunnelInstructionExecutor extends AbstractInstructionExecutor {
45     private static final Logger LOG = LoggerFactory.getLogger(UpdateTunnelInstructionExecutor.class);
46     private final PcepUpdateTunnelInput updateTunnelInput;
47     private final DataBroker dataProvider;
48     private final UpdateLsp updateLsp;
49
50     UpdateTunnelInstructionExecutor(final PcepUpdateTunnelInput updateTunnelInput, final DataBroker dataProvider,
51             final RpcConsumerRegistry rpcConsumerRegistry) {
52         super(updateTunnelInput);
53         this.updateTunnelInput = updateTunnelInput;
54         this.dataProvider = dataProvider;
55         updateLsp = rpcConsumerRegistry.getRpc(UpdateLsp.class);
56     }
57
58     @Override
59     protected ListenableFuture<OperationResult> invokeOperation() {
60         final InstanceIdentifier<Topology> tii = TopologyProgrammingUtil.topologyForInput(updateTunnelInput);
61         final InstanceIdentifier<Link> lii = TunnelProgrammingUtil.linkIdentifier(tii, updateTunnelInput);
62         try (ReadTransaction t = dataProvider.newReadOnlyTransaction()) {
63             final Link link;
64             final Node node;
65             try {
66                 // The link has to exist
67                 link = t.read(LogicalDatastoreType.OPERATIONAL, lii).get().orElseThrow();
68                 // The source node has to exist
69                 node = TunelProgrammingUtil.sourceNode(t, tii, link).orElseThrow();
70             } catch (final InterruptedException | ExecutionException e) {
71                 LOG.debug("Link or node does not exist.", e);
72                 return TunelProgrammingUtil.RESULT;
73             }
74             return Futures.transform(updateLsp.invoke(buildUpdateInput(link, node)), RpcResult::getResult,
75                 MoreExecutors.directExecutor());
76         }
77     }
78
79     private UpdateLspInput buildUpdateInput(final Link link, final Node node) {
80         final UpdateLspInputBuilder ab = new UpdateLspInputBuilder()
81             .setName(link.augmentation(Link1.class).getSymbolicPathName())
82             .setNode(requireNonNull(TunelProgrammingUtil.supportingNode(node)));
83
84         final var args = new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev220730
85             .update.lsp.args.ArgumentsBuilder()
86                 .setBandwidth(new BandwidthBuilder().setBandwidth(updateTunnelInput.getBandwidth()).build())
87                 .setClassType(new ClassTypeBuilder().setClassType(updateTunnelInput.getClassType()).build())
88                 .setEro(TunelProgrammingUtil.buildEro(updateTunnelInput.getExplicitHops()))
89                 .setLspa(new LspaBuilder(updateTunnelInput).build());
90
91         final AdministrativeStatus adminStatus = updateTunnelInput.augmentation(PcepUpdateTunnelInput1.class)
92                 .getAdministrativeStatus();
93         if (adminStatus != null) {
94             args.addAugmentation(new Arguments3Builder().setLsp(new LspBuilder()
95                     .setAdministrative(adminStatus == AdministrativeStatus.Active).build()).build());
96         }
97         return ab.setArguments(args.build()).build();
98     }
99 }