Modernize RPC registration
[bgpcep.git] / pcep / tunnel / tunnel-provider / src / main / java / org / opendaylight / bgpcep / pcep / tunnel / provider / TunnelProgramming.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.tunnel.provider;
9
10 import static java.util.Objects.requireNonNull;
11
12 import com.google.common.annotations.VisibleForTesting;
13 import com.google.common.util.concurrent.Futures;
14 import com.google.common.util.concurrent.ListenableFuture;
15 import java.util.List;
16 import java.util.Set;
17 import org.eclipse.jdt.annotation.NonNull;
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.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.pcep.programming.rev181109.PcepCreateP2pTunnel;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.pcep.programming.rev181109.PcepCreateP2pTunnelInput;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.pcep.programming.rev181109.PcepCreateP2pTunnelOutput;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.pcep.programming.rev181109.PcepCreateP2pTunnelOutputBuilder;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.pcep.programming.rev181109.PcepDestroyTunnel;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.pcep.programming.rev181109.PcepDestroyTunnelInput;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.pcep.programming.rev181109.PcepDestroyTunnelOutput;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.pcep.programming.rev181109.PcepDestroyTunnelOutputBuilder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.pcep.programming.rev181109.PcepUpdateTunnel;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.pcep.programming.rev181109.PcepUpdateTunnelInput;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.pcep.programming.rev181109.PcepUpdateTunnelOutput;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.tunnel.pcep.programming.rev181109.PcepUpdateTunnelOutputBuilder;
33 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
34 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
35 import org.opendaylight.yangtools.concepts.Registration;
36 import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
37 import org.opendaylight.yangtools.yang.common.RpcResult;
38 import org.slf4j.Logger;
39 import org.slf4j.LoggerFactory;
40
41 public final class TunnelProgramming implements AutoCloseable {
42     private static final Logger LOG = LoggerFactory.getLogger(TunnelProgramming.class);
43
44     private final InstructionScheduler scheduler;
45     private final TunnelProviderDependencies dependencies;
46
47     TunnelProgramming(final @NonNull InstructionScheduler scheduler,
48             final @NonNull TunnelProviderDependencies dependencies) {
49         this.scheduler = requireNonNull(scheduler);
50         this.dependencies = requireNonNull(dependencies);
51     }
52
53     Registration register(final KeyedInstanceIdentifier<Topology, TopologyKey> topologyPath) {
54         return dependencies.getRpcProviderRegistry().registerRpcImplementations(List.of(
55             (PcepCreateP2pTunnel) this::pcepCreateP2pTunnel,
56             (PcepDestroyTunnel) this::pcepDestroyTunnel,
57             (PcepUpdateTunnel) this::pcepUpdateTunnel), Set.of(topologyPath));
58     }
59
60     @VisibleForTesting
61     ListenableFuture<RpcResult<PcepCreateP2pTunnelOutput>> pcepCreateP2pTunnel(
62             final PcepCreateP2pTunnelInput p2pTunnelInput) {
63         return Futures.immediateFuture(SuccessfulRpcResult.create(new PcepCreateP2pTunnelOutputBuilder()
64             .setResult(AbstractInstructionExecutor.schedule(scheduler, new CreateTunnelInstructionExecutor(
65                 p2pTunnelInput, dependencies.getDataBroker(), dependencies.getRpcConsumerRegistry())))
66             .build()));
67     }
68
69     @VisibleForTesting
70     ListenableFuture<RpcResult<PcepDestroyTunnelOutput>> pcepDestroyTunnel(
71             final PcepDestroyTunnelInput destroyTunnelInput) {
72         return Futures.immediateFuture(SuccessfulRpcResult.create(new PcepDestroyTunnelOutputBuilder()
73             .setResult(AbstractInstructionExecutor.schedule(scheduler, new DestroyTunnelInstructionExecutor(
74                 destroyTunnelInput, dependencies.getDataBroker(), dependencies.getRpcConsumerRegistry())))
75             .build()));
76     }
77
78     @VisibleForTesting
79     ListenableFuture<RpcResult<PcepUpdateTunnelOutput>> pcepUpdateTunnel(
80             final PcepUpdateTunnelInput updateTunnelInput) {
81         return Futures.immediateFuture(SuccessfulRpcResult.create(new PcepUpdateTunnelOutputBuilder()
82             .setResult(AbstractInstructionExecutor.schedule(scheduler, new UpdateTunnelInstructionExecutor(
83                 updateTunnelInput, dependencies.getDataBroker(), dependencies.getRpcConsumerRegistry())))
84             .build()));
85     }
86
87     @Override
88     public void close() {
89         LOG.debug("Shutting down instruction scheduler {}", this);
90     }
91 }