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.HashMap;
14 import java.util.concurrent.Future;
15 import org.opendaylight.controller.clustering.it.provider.impl.GetConstantService;
16 import org.opendaylight.controller.clustering.it.provider.impl.RoutedGetConstantService;
17 import org.opendaylight.controller.md.sal.binding.impl.BindingToNormalizedNodeCodec;
18 import org.opendaylight.controller.md.sal.binding.impl.BindingToNormalizedNodeCodecFactory;
19 import org.opendaylight.controller.md.sal.dom.api.DOMRpcImplementationRegistration;
20 import org.opendaylight.controller.md.sal.dom.api.DOMRpcProviderService;
21 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker;
22 import org.opendaylight.controller.sal.binding.api.RpcProviderRegistry;
23 import org.opendaylight.controller.sal.core.api.model.SchemaService;
24 import org.opendaylight.mdsal.binding.dom.codec.api.BindingNormalizedNodeSerializer;
25 import org.opendaylight.mdsal.singleton.common.api.ClusterSingletonServiceProvider;
26 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.AddShardReplicaInput;
27 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.BecomeModuleLeaderInput;
28 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.BecomePrefixLeaderInput;
29 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.IsClientAbortedOutput;
30 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.OdlMdsalLowlevelControlService;
31 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.ProduceTransactionsInput;
32 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.PublishNotificationsInput;
33 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.RegisterBoundConstantInput;
34 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.RegisterConstantInput;
35 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.RegisterDefaultConstantInput;
36 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.RegisterSingletonConstantInput;
37 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.RemoveShardReplicaInput;
38 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.SubscribeYnlInput;
39 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.UnregisterBoundConstantInput;
40 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.UnregisterFlappingSingletonOutput;
41 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.UnsubscribeDdtlOutput;
42 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.UnsubscribeDtclOutput;
43 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.UnsubscribeYnlInput;
44 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.UnsubscribeYnlOutput;
45 import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.WriteTransactionsInput;
46 import org.opendaylight.yangtools.concepts.ListenerRegistration;
47 import org.opendaylight.yangtools.sal.binding.generator.impl.GeneratedClassLoadingStrategy;
48 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
49 import org.opendaylight.yangtools.yang.common.RpcError;
50 import org.opendaylight.yangtools.yang.common.RpcError.ErrorType;
51 import org.opendaylight.yangtools.yang.common.RpcResult;
52 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
53 import org.opendaylight.yangtools.yang.model.api.SchemaContextListener;
54 import org.slf4j.Logger;
55 import org.slf4j.LoggerFactory;
57 public class MdsalLowLevelTestProvider implements OdlMdsalLowlevelControlService {
59 private static final Logger LOG = LoggerFactory.getLogger(MdsalLowLevelTestProvider.class);
61 private final RpcProviderRegistry rpcRegistry;
62 private final BindingAwareBroker.RpcRegistration<OdlMdsalLowlevelControlService> registration;
63 private final BindingNormalizedNodeSerializer bindingNormalizedNodeSerializer;
64 private final SchemaService schemaService;
65 private final ClusterSingletonServiceProvider singletonService;
66 private final DOMRpcProviderService domRpcService;
68 private Map<InstanceIdentifier<?>, DOMRpcImplementationRegistration<RoutedGetConstantService>> routedRegistrations =
71 private DOMRpcImplementationRegistration<GetConstantService> globalGetConstantRegistration = null;
73 public MdsalLowLevelTestProvider(final RpcProviderRegistry rpcRegistry,
74 final DOMRpcProviderService domRpcService,
75 final ClusterSingletonServiceProvider singletonService,
76 final SchemaService schemaService,
77 final BindingNormalizedNodeSerializer bindingNormalizedNodeSerializer) {
78 this.rpcRegistry = rpcRegistry;
79 this.domRpcService = domRpcService;
80 this.singletonService = singletonService;
81 this.schemaService = schemaService;
82 this.bindingNormalizedNodeSerializer = bindingNormalizedNodeSerializer;
84 registration = rpcRegistry.addRpcImplementation(OdlMdsalLowlevelControlService.class, this);
88 public Future<RpcResult<Void>> unregisterSingletonConstant() {
93 public Future<RpcResult<Void>> publishNotifications(PublishNotificationsInput input) {
98 public Future<RpcResult<Void>> subscribeDtcl() {
103 public Future<RpcResult<Void>> writeTransactions(WriteTransactionsInput input) {
108 public Future<RpcResult<IsClientAbortedOutput>> isClientAborted() {
113 public Future<RpcResult<Void>> becomeModuleLeader(BecomeModuleLeaderInput input) {
118 public Future<RpcResult<Void>> removeShardReplica(RemoveShardReplicaInput input) {
123 public Future<RpcResult<Void>> subscribeYnl(SubscribeYnlInput input) {
128 public Future<RpcResult<Void>> becomePrefixLeader(BecomePrefixLeaderInput input) {
133 public Future<RpcResult<Void>> unregisterBoundConstant(final UnregisterBoundConstantInput input) {
134 LOG.debug("unregister-bound-constant, {}", input);
136 final DOMRpcImplementationRegistration<RoutedGetConstantService> registration =
137 routedRegistrations.remove(input.getContext());
139 if (registration == null) {
140 LOG.debug("No get-contexted-constant registration for context: {}", input.getContext());
141 final RpcError rpcError = RpcResultBuilder
142 .newError(ErrorType.APPLICATION, "missing-registration", "No get-constant rpc registration present.");
143 final RpcResult<Void> result = RpcResultBuilder.<Void>failed().withRpcError(rpcError).build();
144 return Futures.immediateFuture(result);
147 registration.close();
148 return Futures.immediateFuture(RpcResultBuilder.<Void>success().build());
152 public Future<RpcResult<Void>> registerSingletonConstant(RegisterSingletonConstantInput input) {
157 public Future<RpcResult<Void>> registerDefaultConstant(RegisterDefaultConstantInput input) {
162 public Future<RpcResult<Void>> unregisterConstant() {
164 if (globalGetConstantRegistration == null) {
165 final RpcError rpcError = RpcResultBuilder
166 .newError(ErrorType.APPLICATION, "missing-registration", "No get-constant rpc registration present.");
167 final RpcResult<Void> result = RpcResultBuilder.<Void>failed().withRpcError(rpcError).build();
168 return Futures.immediateFuture(result);
171 globalGetConstantRegistration.close();
172 globalGetConstantRegistration = null;
174 return Futures.immediateFuture(RpcResultBuilder.<Void>success().build());
178 public Future<RpcResult<UnregisterFlappingSingletonOutput>> unregisterFlappingSingleton() {
183 public Future<RpcResult<Void>> addShardReplica(AddShardReplicaInput input) {
188 public Future<RpcResult<Void>> subscribeDdtl() {
193 public Future<RpcResult<Void>> registerBoundConstant(final RegisterBoundConstantInput input) {
194 LOG.debug("register-bound-constant: {}", input);
196 if (input.getContext() == null) {
197 final RpcError error = RpcResultBuilder.newError(
198 ErrorType.RPC, "Invalid input.", "Context value is null");
199 return Futures.immediateFuture(RpcResultBuilder.<Void>failed().withRpcError(error).build());
202 if (input.getConstant() == null) {
203 final RpcError error = RpcResultBuilder.newError(
204 ErrorType.RPC, "Invalid input.", "Constant value is null");
205 return Futures.immediateFuture(RpcResultBuilder.<Void>failed().withRpcError(error).build());
208 if (routedRegistrations.containsKey(input.getContext())) {
209 final RpcError error = RpcResultBuilder.newError(ErrorType.RPC, "Registration present.",
210 "There is already a rpc registered for context: " + input.getContext());
211 return Futures.immediateFuture(RpcResultBuilder.<Void>failed().withRpcError(error).build());
214 final DOMRpcImplementationRegistration<RoutedGetConstantService> registration =
215 RoutedGetConstantService.registerNew(bindingNormalizedNodeSerializer, domRpcService,
216 input.getConstant(), input.getContext());
218 routedRegistrations.put(input.getContext(), registration);
219 return Futures.immediateFuture(RpcResultBuilder.<Void>success().build());
223 public Future<RpcResult<Void>> registerFlappingSingleton() {
228 public Future<RpcResult<UnsubscribeDtclOutput>> unsubscribeDtcl() {
233 public Future<RpcResult<Void>> deconfigureIdIntsShard() {
238 public Future<RpcResult<UnsubscribeYnlOutput>> unsubscribeYnl(UnsubscribeYnlInput input) {
243 public Future<RpcResult<Void>> produceTransactions(ProduceTransactionsInput input) {
248 public Future<RpcResult<Void>> registerConstant(final RegisterConstantInput input) {
250 LOG.debug("Received register-constant rpc, input: {}", input);
252 if (input.getConstant() == null) {
253 final RpcError error = RpcResultBuilder.newError(
254 ErrorType.RPC, "Invalid input.", "Constant value is null");
255 return Futures.immediateFuture(RpcResultBuilder.<Void>failed().withRpcError(error).build());
258 if (globalGetConstantRegistration != null) {
259 final RpcError error = RpcResultBuilder.newError(ErrorType.RPC, "Registration present.",
260 "There is already a get-constant rpc registered.");
261 return Futures.immediateFuture(RpcResultBuilder.<Void>failed().withRpcError(error).build());
264 globalGetConstantRegistration = GetConstantService.registerNew(domRpcService, input.getConstant());
265 return Futures.immediateFuture(RpcResultBuilder.<Void>success().build());
269 public Future<RpcResult<Void>> unregisterDefaultConstant() {
274 public Future<RpcResult<UnsubscribeDdtlOutput>> unsubscribeDdtl() {