2 * Copyright (c) 2017 Cisco Systems, Inc. and others. All rights reserved.
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
9 package org.opendaylight.controller.clustering.it.provider;
11 import com.google.common.util.concurrent.Futures;
12 import java.util.concurrent.Future;
13 import org.opendaylight.controller.clustering.it.provider.impl.GetConstantService;
14 import org.opendaylight.controller.md.sal.dom.api.DOMRpcImplementationRegistration;
15 import org.opendaylight.controller.md.sal.dom.api.DOMRpcProviderService;
16 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker;
17 import org.opendaylight.controller.sal.binding.api.RpcProviderRegistry;
18 import org.opendaylight.mdsal.singleton.common.api.ClusterSingletonServiceProvider;
19 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.AddShardReplicaInput;
20 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.BecomeModuleLeaderInput;
21 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.BecomePrefixLeaderInput;
22 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.IsClientAbortedOutput;
23 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.OdlMdsalLowlevelControlService;
24 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.ProduceTransactionsInput;
25 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.PublishNotificationsInput;
26 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.RegisterBoundConstantInput;
27 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.RegisterConstantInput;
28 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.RegisterDefaultConstantInput;
29 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.RegisterSingletonConstantInput;
30 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.RemoveShardReplicaInput;
31 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.SubscribeYnlInput;
32 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.UnregisterBoundConstantInput;
33 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.UnregisterFlappingSingletonOutput;
34 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.UnsubscribeDdtlOutput;
35 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.UnsubscribeDtclOutput;
36 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.UnsubscribeYnlInput;
37 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.UnsubscribeYnlOutput;
38 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.WriteTransactionsInput;
39 import org.opendaylight.yangtools.yang.common.RpcError;
40 import org.opendaylight.yangtools.yang.common.RpcError.ErrorType;
41 import org.opendaylight.yangtools.yang.common.RpcResult;
42 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
43 import org.slf4j.Logger;
44 import org.slf4j.LoggerFactory;
46 public class MdsalLowLevelTestProvider implements OdlMdsalLowlevelControlService {
48 private static final Logger LOG = LoggerFactory.getLogger(MdsalLowLevelTestProvider.class);
50 private final RpcProviderRegistry rpcRegistry;
51 private final BindingAwareBroker.RpcRegistration<OdlMdsalLowlevelControlService> registration;
52 private final ClusterSingletonServiceProvider singletonService;
53 private final DOMRpcProviderService domRpcService;
55 private DOMRpcImplementationRegistration<GetConstantService> globalGetConstantRegistration = null;
57 public MdsalLowLevelTestProvider(final RpcProviderRegistry rpcRegistry,
58 final DOMRpcProviderService domRpcService,
59 final ClusterSingletonServiceProvider singletonService) {
60 this.rpcRegistry = rpcRegistry;
61 this.domRpcService = domRpcService;
62 this.singletonService = singletonService;
64 registration = rpcRegistry.addRpcImplementation(OdlMdsalLowlevelControlService.class, this);
68 public Future<RpcResult<Void>> unregisterSingletonConstant() {
73 public Future<RpcResult<Void>> publishNotifications(PublishNotificationsInput input) {
78 public Future<RpcResult<Void>> subscribeDtcl() {
83 public Future<RpcResult<Void>> writeTransactions(WriteTransactionsInput input) {
88 public Future<RpcResult<IsClientAbortedOutput>> isClientAborted() {
93 public Future<RpcResult<Void>> becomeModuleLeader(BecomeModuleLeaderInput input) {
98 public Future<RpcResult<Void>> removeShardReplica(RemoveShardReplicaInput input) {
103 public Future<RpcResult<Void>> subscribeYnl(SubscribeYnlInput input) {
108 public Future<RpcResult<Void>> becomePrefixLeader(BecomePrefixLeaderInput input) {
113 public Future<RpcResult<Void>> unregisterBoundConstant(UnregisterBoundConstantInput input) {
118 public Future<RpcResult<Void>> registerSingletonConstant(RegisterSingletonConstantInput input) {
123 public Future<RpcResult<Void>> registerDefaultConstant(RegisterDefaultConstantInput input) {
128 public Future<RpcResult<Void>> unregisterConstant() {
130 if (globalGetConstantRegistration == null) {
131 final RpcError rpcError = RpcResultBuilder
132 .newError(ErrorType.APPLICATION, "missing-registration", "No get-constant rpc registration present.");
133 final RpcResult<Void> result = RpcResultBuilder.<Void>failed().withRpcError(rpcError).build();
134 return Futures.immediateFuture(result);
137 globalGetConstantRegistration.close();
138 globalGetConstantRegistration = null;
140 return Futures.immediateFuture(RpcResultBuilder.<Void>success().build());
144 public Future<RpcResult<UnregisterFlappingSingletonOutput>> unregisterFlappingSingleton() {
149 public Future<RpcResult<Void>> addShardReplica(AddShardReplicaInput input) {
154 public Future<RpcResult<Void>> subscribeDdtl() {
159 public Future<RpcResult<Void>> registerBoundConstant(RegisterBoundConstantInput input) {
164 public Future<RpcResult<Void>> registerFlappingSingleton() {
169 public Future<RpcResult<UnsubscribeDtclOutput>> unsubscribeDtcl() {
174 public Future<RpcResult<Void>> deconfigureIdIntsShard() {
179 public Future<RpcResult<UnsubscribeYnlOutput>> unsubscribeYnl(UnsubscribeYnlInput input) {
184 public Future<RpcResult<Void>> produceTransactions(ProduceTransactionsInput input) {
189 public Future<RpcResult<Void>> registerConstant(final RegisterConstantInput input) {
191 LOG.debug("Received register-constant rpc, input: {}", input);
193 if (input.getConstant() == null) {
194 final RpcError error = RpcResultBuilder.newError(
195 ErrorType.RPC, "Invalid input.", "Constant value is null");
196 return Futures.immediateFuture(RpcResultBuilder.<Void>failed().withRpcError(error).build());
199 if (globalGetConstantRegistration != null) {
200 final RpcError error = RpcResultBuilder.newError(ErrorType.RPC, "Registration present.",
201 "There is already a get-constant rpc registered.");
202 return Futures.immediateFuture(RpcResultBuilder.<Void>failed().withRpcError(error).build());
205 globalGetConstantRegistration = GetConstantService.registerNew(domRpcService, input.getConstant());
206 return Futures.immediateFuture(RpcResultBuilder.<Void>success().build());
210 public Future<RpcResult<Void>> unregisterDefaultConstant() {
215 public Future<RpcResult<UnsubscribeDdtlOutput>> unsubscribeDdtl() {