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