acccbbc169f7fad5ae993f9ff7ad6b796781191f
[bgpcep.git] / pcep / 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 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.controller.md.sal.binding.api.DataBroker;
20 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
21 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
22 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.AdministrativeStatus;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.Arguments3;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.Arguments3Builder;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.PcepUpdateTunnelInput1;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.object.LspBuilder;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.bandwidth.object.BandwidthBuilder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.classtype.object.ClassTypeBuilder;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lspa.object.LspaBuilder;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.NetworkTopologyPcepService;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.OperationResult;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.UpdateLspInput;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.UpdateLspInputBuilder;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.UpdateLspOutput;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.pcep.programming.rev131030.PcepUpdateTunnelInput;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.pcep.rev130820.Link1;
38 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
39 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Link;
40 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
41 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
42 import org.opendaylight.yangtools.yang.common.RpcResult;
43 import org.slf4j.Logger;
44 import org.slf4j.LoggerFactory;
45
46 final class UpdateTunnelInstructionExecutor extends AbstractInstructionExecutor {
47     private static final Logger LOG = LoggerFactory.getLogger(UpdateTunnelInstructionExecutor.class);
48     private final PcepUpdateTunnelInput updateTunnelInput;
49     private final DataBroker dataProvider;
50     private final NetworkTopologyPcepService topologyService;
51
52     UpdateTunnelInstructionExecutor(final PcepUpdateTunnelInput updateTunnelInput, final DataBroker dataProvider,
53         final NetworkTopologyPcepService topologyService) {
54         super(updateTunnelInput);
55         this.updateTunnelInput = updateTunnelInput;
56         this.dataProvider = dataProvider;
57         this.topologyService = topologyService;
58     }
59
60     @Override
61     protected ListenableFuture<OperationResult> invokeOperation() {
62         final InstanceIdentifier<Topology> tii = TopologyProgrammingUtil.topologyForInput(this.updateTunnelInput);
63         final InstanceIdentifier<Link> lii = TunnelProgrammingUtil.linkIdentifier(tii, this.updateTunnelInput);
64         try (final ReadOnlyTransaction t = this.dataProvider.newReadOnlyTransaction()) {
65             final Link link;
66             final Node node;
67             try {
68                 // The link has to exist
69                 link = t.read(LogicalDatastoreType.OPERATIONAL, lii).checkedGet().get();
70                 // The source node has to exist
71                 node = TunelProgrammingUtil.sourceNode(t, tii, link).get();
72             } catch (IllegalStateException | ReadFailedException e) {
73                 LOG.debug("Link or node does not exist.", e);
74                 return TunelProgrammingUtil.RESULT;
75             }
76             return Futures.transform(
77                 (ListenableFuture<RpcResult<UpdateLspOutput>>) this.topologyService.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.getAugmentation(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.rev131024.update.lsp.args.ArgumentsBuilder args =
88             new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.update.lsp.args.ArgumentsBuilder();
89         args.setBandwidth(new BandwidthBuilder().setBandwidth(this.updateTunnelInput.getBandwidth()).build());
90         args.setClassType(new ClassTypeBuilder().setClassType(this.updateTunnelInput.getClassType()).build());
91         args.setEro(TunelProgrammingUtil.buildEro(this.updateTunnelInput.getExplicitHops()));
92         args.setLspa(new LspaBuilder(this.updateTunnelInput).build());
93
94         final AdministrativeStatus adminStatus = this.updateTunnelInput.getAugmentation(PcepUpdateTunnelInput1.class).getAdministrativeStatus();
95         if (adminStatus != null) {
96             args.addAugmentation(Arguments3.class, new Arguments3Builder().setLsp(new LspBuilder().
97                 setAdministrative(adminStatus == AdministrativeStatus.Active).build()).build());
98         }
99         ab.setArguments(args.build());
100         return ab.build();
101     }
102 }