Modernize RPC registration
[bgpcep.git] / pcep / topology / topology-provider / src / main / java / org / opendaylight / bgpcep / pcep / topology / provider / TopologyProgramming.java
1 /*
2  * Copyright (c) 2013 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.topology.provider;
9
10 import static java.util.Objects.requireNonNull;
11
12 import com.google.common.annotations.VisibleForTesting;
13 import com.google.common.base.Preconditions;
14 import com.google.common.util.concurrent.Futures;
15 import com.google.common.util.concurrent.ListenableFuture;
16 import java.util.List;
17 import java.util.Set;
18 import org.opendaylight.bgpcep.pcep.topology.spi.AbstractInstructionExecutor;
19 import org.opendaylight.bgpcep.programming.spi.InstructionScheduler;
20 import org.opendaylight.bgpcep.programming.spi.SuccessfulRpcResult;
21 import org.opendaylight.mdsal.binding.api.RpcProviderService;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.programming.rev181109.SubmitAddLsp;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.programming.rev181109.SubmitAddLspInput;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.programming.rev181109.SubmitAddLspOutput;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.programming.rev181109.SubmitAddLspOutputBuilder;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.programming.rev181109.SubmitEnsureLspOperational;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.programming.rev181109.SubmitEnsureLspOperationalInput;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.programming.rev181109.SubmitEnsureLspOperationalOutput;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.programming.rev181109.SubmitEnsureLspOperationalOutputBuilder;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.programming.rev181109.SubmitRemoveLsp;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.programming.rev181109.SubmitRemoveLspInput;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.programming.rev181109.SubmitRemoveLspOutput;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.programming.rev181109.SubmitRemoveLspOutputBuilder;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.programming.rev181109.SubmitTriggerSync;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.programming.rev181109.SubmitTriggerSyncInput;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.programming.rev181109.SubmitTriggerSyncOutput;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.programming.rev181109.SubmitTriggerSyncOutputBuilder;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.programming.rev181109.SubmitUpdateLsp;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.programming.rev181109.SubmitUpdateLspInput;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.programming.rev181109.SubmitUpdateLspOutput;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.programming.rev181109.SubmitUpdateLspOutputBuilder;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev220730.EnsureLspOperationalInputBuilder;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev220730.OperationResult;
44 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
45 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
46 import org.opendaylight.yangtools.concepts.Registration;
47 import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
48 import org.opendaylight.yangtools.yang.common.RpcResult;
49
50 final class TopologyProgramming {
51     private final InstructionScheduler scheduler;
52     private final ServerSessionManager manager;
53
54     TopologyProgramming(final InstructionScheduler scheduler, final ServerSessionManager manager) {
55         this.scheduler = requireNonNull(scheduler);
56         this.manager = requireNonNull(manager);
57     }
58
59     Registration register(final RpcProviderService rpcProviderService,
60             final KeyedInstanceIdentifier<Topology, TopologyKey> path) {
61         return rpcProviderService.registerRpcImplementations(List.of(
62             (SubmitAddLsp) this::submitAddLsp,
63             (SubmitRemoveLsp) this::submitRemoveLsp,
64             (SubmitUpdateLsp) this::submitUpdateLsp,
65             (SubmitEnsureLspOperational) this::submitEnsureLspOperational,
66             (SubmitTriggerSync) this::submitTriggerSync), Set.of(path));
67     }
68
69     @VisibleForTesting
70     ListenableFuture<RpcResult<SubmitAddLspOutput>> submitAddLsp(final SubmitAddLspInput input) {
71         Preconditions.checkArgument(input.getNode() != null);
72         Preconditions.checkArgument(input.getName() != null);
73
74         return Futures.immediateFuture(SuccessfulRpcResult.create(new SubmitAddLspOutputBuilder()
75             .setResult(AbstractInstructionExecutor.schedule(scheduler, new AbstractInstructionExecutor(input) {
76                 @Override
77                 protected ListenableFuture<OperationResult> invokeOperation() {
78                     return manager.addLsp(input);
79                 }
80             }))
81             .build()));
82     }
83
84     @VisibleForTesting
85     ListenableFuture<RpcResult<SubmitRemoveLspOutput>> submitRemoveLsp(final SubmitRemoveLspInput input) {
86         Preconditions.checkArgument(input.getNode() != null);
87         Preconditions.checkArgument(input.getName() != null);
88
89         return Futures.immediateFuture(SuccessfulRpcResult.create(new SubmitRemoveLspOutputBuilder()
90             .setResult(AbstractInstructionExecutor.schedule(scheduler, new AbstractInstructionExecutor(input) {
91                 @Override
92                 protected ListenableFuture<OperationResult> invokeOperation() {
93                     return manager.removeLsp(input);
94                 }
95             }))
96             .build()));
97     }
98
99     @VisibleForTesting
100     ListenableFuture<RpcResult<SubmitUpdateLspOutput>> submitUpdateLsp(final SubmitUpdateLspInput input) {
101         Preconditions.checkArgument(input.getNode() != null);
102         Preconditions.checkArgument(input.getName() != null);
103
104         return Futures.immediateFuture(SuccessfulRpcResult.create(new SubmitUpdateLspOutputBuilder()
105             .setResult(AbstractInstructionExecutor.schedule(scheduler, new AbstractInstructionExecutor(input) {
106                 @Override
107                 protected ListenableFuture<OperationResult> invokeOperation() {
108                     return manager.updateLsp(input);
109                 }
110             }))
111             .build()));
112     }
113
114     @VisibleForTesting
115     ListenableFuture<RpcResult<SubmitEnsureLspOperationalOutput>> submitEnsureLspOperational(
116             final SubmitEnsureLspOperationalInput input) {
117         Preconditions.checkArgument(input.getNode() != null);
118         Preconditions.checkArgument(input.getName() != null);
119         Preconditions.checkArgument(input.getArguments() != null);
120
121         // FIXME: can we validate this early?
122         // Preconditions.checkArgument(input.getArguments().getOperational() != null);
123
124         return Futures.immediateFuture(SuccessfulRpcResult.create(new SubmitEnsureLspOperationalOutputBuilder()
125             .setResult(AbstractInstructionExecutor.schedule(scheduler, new AbstractInstructionExecutor(input) {
126                 @Override
127                 protected ListenableFuture<OperationResult> invokeOperation() {
128                     EnsureLspOperationalInputBuilder ensureLspOperationalInputBuilder =
129                         new EnsureLspOperationalInputBuilder();
130                     ensureLspOperationalInputBuilder.fieldsFrom(input);
131                     return manager.ensureLspOperational(ensureLspOperationalInputBuilder.build());
132                 }
133             }))
134             .build()));
135     }
136
137
138     @VisibleForTesting
139     ListenableFuture<RpcResult<SubmitTriggerSyncOutput>> submitTriggerSync(final SubmitTriggerSyncInput input) {
140         Preconditions.checkArgument(input.getNode() != null);
141
142         return Futures.immediateFuture(SuccessfulRpcResult.create(new SubmitTriggerSyncOutputBuilder()
143             .setResult(AbstractInstructionExecutor.schedule(scheduler, new AbstractInstructionExecutor(input) {
144                 @Override
145                 protected ListenableFuture<OperationResult> invokeOperation() {
146                     return manager.triggerSync(input);
147                 }
148             }))
149             .build()));
150     }
151 }