package org.opendaylight.controller.clustering.it.provider;
import com.google.common.util.concurrent.Futures;
+import com.google.common.util.concurrent.SettableFuture;
+import java.util.HashMap;
+import java.util.Map;
import java.util.concurrent.Future;
+import org.opendaylight.controller.clustering.it.provider.impl.FlappingSingletonService;
import org.opendaylight.controller.clustering.it.provider.impl.GetConstantService;
+import org.opendaylight.controller.clustering.it.provider.impl.PublishNotificationsTask;
+import org.opendaylight.controller.clustering.it.provider.impl.RoutedGetConstantService;
+import org.opendaylight.controller.clustering.it.provider.impl.SingletonGetConstantService;
+import org.opendaylight.controller.clustering.it.provider.impl.YnlListener;
+import org.opendaylight.controller.md.sal.binding.api.NotificationPublishService;
+import org.opendaylight.controller.md.sal.binding.api.NotificationService;
import org.opendaylight.controller.md.sal.dom.api.DOMRpcImplementationRegistration;
import org.opendaylight.controller.md.sal.dom.api.DOMRpcProviderService;
import org.opendaylight.controller.sal.binding.api.BindingAwareBroker;
import org.opendaylight.controller.sal.binding.api.RpcProviderRegistry;
+import org.opendaylight.controller.sal.core.api.model.SchemaService;
+import org.opendaylight.mdsal.binding.dom.codec.api.BindingNormalizedNodeSerializer;
import org.opendaylight.mdsal.singleton.common.api.ClusterSingletonServiceProvider;
+import org.opendaylight.mdsal.singleton.common.api.ClusterSingletonServiceRegistration;
import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.AddShardReplicaInput;
import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.BecomeModuleLeaderInput;
import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.BecomePrefixLeaderInput;
import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.SubscribeYnlInput;
import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.UnregisterBoundConstantInput;
import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.UnregisterFlappingSingletonOutput;
+import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.UnregisterFlappingSingletonOutputBuilder;
import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.UnsubscribeDdtlOutput;
import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.UnsubscribeDtclOutput;
import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.UnsubscribeYnlInput;
import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.UnsubscribeYnlOutput;
import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.WriteTransactionsInput;
+import org.opendaylight.yangtools.concepts.ListenerRegistration;
+import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
import org.opendaylight.yangtools.yang.common.RpcError;
import org.opendaylight.yangtools.yang.common.RpcError.ErrorType;
import org.opendaylight.yangtools.yang.common.RpcResult;
private final RpcProviderRegistry rpcRegistry;
private final BindingAwareBroker.RpcRegistration<OdlMdsalLowlevelControlService> registration;
+ private final BindingNormalizedNodeSerializer bindingNormalizedNodeSerializer;
+ private final NotificationPublishService notificationPublishService;
+ private final NotificationService notificationService;
+ private final SchemaService schemaService;
private final ClusterSingletonServiceProvider singletonService;
private final DOMRpcProviderService domRpcService;
+ private Map<InstanceIdentifier<?>, DOMRpcImplementationRegistration<RoutedGetConstantService>> routedRegistrations =
+ new HashMap<>();
+
+ private Map<String, ListenerRegistration<YnlListener>> ynlRegistrations = new HashMap<>();
+
private DOMRpcImplementationRegistration<GetConstantService> globalGetConstantRegistration = null;
+ private ClusterSingletonServiceRegistration getSingletonConstantRegistration;
+ private FlappingSingletonService flappingSingletonService;
public MdsalLowLevelTestProvider(final RpcProviderRegistry rpcRegistry,
final DOMRpcProviderService domRpcService,
- final ClusterSingletonServiceProvider singletonService) {
+ final ClusterSingletonServiceProvider singletonService,
+ final SchemaService schemaService,
+ final BindingNormalizedNodeSerializer bindingNormalizedNodeSerializer,
+ final NotificationPublishService notificationPublishService,
+ final NotificationService notificationService) {
this.rpcRegistry = rpcRegistry;
this.domRpcService = domRpcService;
this.singletonService = singletonService;
+ this.schemaService = schemaService;
+ this.bindingNormalizedNodeSerializer = bindingNormalizedNodeSerializer;
+ this.notificationPublishService = notificationPublishService;
+ this.notificationService = notificationService;
registration = rpcRegistry.addRpcImplementation(OdlMdsalLowlevelControlService.class, this);
}
@Override
public Future<RpcResult<Void>> unregisterSingletonConstant() {
- return null;
+ LOG.debug("unregister-singleton-constant");
+
+ if (getSingletonConstantRegistration == null) {
+ LOG.debug("No get-singleton-constant registration present.");
+ final RpcError rpcError = RpcResultBuilder
+ .newError(ErrorType.APPLICATION, "missing-registration", "No get-singleton-constant rpc registration present.");
+ final RpcResult<Void> result = RpcResultBuilder.<Void>failed().withRpcError(rpcError).build();
+ return Futures.immediateFuture(result);
+ }
+
+ try {
+ getSingletonConstantRegistration.close();
+ getSingletonConstantRegistration = null;
+
+ return Futures.immediateFuture(RpcResultBuilder.<Void>success().build());
+ } catch (final Exception e) {
+ LOG.debug("There was a problem closing the singleton constant service", e);
+ final RpcError rpcError = RpcResultBuilder
+ .newError(ErrorType.APPLICATION, "error-closing", "There was a problem closing get-singleton-constant");
+ final RpcResult<Void> result = RpcResultBuilder.<Void>failed().withRpcError(rpcError).build();
+ return Futures.immediateFuture(result);
+ }
}
@Override
- public Future<RpcResult<Void>> publishNotifications(PublishNotificationsInput input) {
- return null;
+ public Future<RpcResult<Void>> publishNotifications(final PublishNotificationsInput input) {
+ LOG.debug("publish-notifications, input: {}", input);
+
+ final PublishNotificationsTask task = new PublishNotificationsTask(notificationPublishService, input.getId(),
+ input.getSeconds(), input.getNotificationsPerSecond());
+
+ final SettableFuture<RpcResult<Void>> settableFuture = SettableFuture.create();
+ task.start(settableFuture);
+
+ return settableFuture;
}
@Override
}
@Override
- public Future<RpcResult<Void>> subscribeYnl(SubscribeYnlInput input) {
- return null;
+ public Future<RpcResult<Void>> subscribeYnl(final SubscribeYnlInput input) {
+
+ LOG.debug("subscribe-ynl, input: {}", input);
+
+ if (ynlRegistrations.containsKey(input.getId())) {
+ final RpcError error = RpcResultBuilder.newError(ErrorType.RPC, "Registration present.",
+ "There is already ynl listener registered for this id: " + input.getId());
+ return Futures.immediateFuture(RpcResultBuilder.<Void>failed().withRpcError(error).build());
+ }
+
+ ynlRegistrations.put(input.getId(),
+ notificationService.registerNotificationListener(new YnlListener(input.getId())));
+
+ return Futures.immediateFuture(RpcResultBuilder.<Void>success().build());
}
@Override
}
@Override
- public Future<RpcResult<Void>> unregisterBoundConstant(UnregisterBoundConstantInput input) {
- return null;
+ public Future<RpcResult<Void>> unregisterBoundConstant(final UnregisterBoundConstantInput input) {
+ LOG.debug("unregister-bound-constant, {}", input);
+
+ final DOMRpcImplementationRegistration<RoutedGetConstantService> registration =
+ routedRegistrations.remove(input.getContext());
+
+ if (registration == null) {
+ LOG.debug("No get-contexted-constant registration for context: {}", input.getContext());
+ final RpcError rpcError = RpcResultBuilder
+ .newError(ErrorType.APPLICATION, "missing-registration", "No get-constant rpc registration present.");
+ final RpcResult<Void> result = RpcResultBuilder.<Void>failed().withRpcError(rpcError).build();
+ return Futures.immediateFuture(result);
+ }
+
+ registration.close();
+ return Futures.immediateFuture(RpcResultBuilder.<Void>success().build());
}
@Override
- public Future<RpcResult<Void>> registerSingletonConstant(RegisterSingletonConstantInput input) {
- return null;
+ public Future<RpcResult<Void>> registerSingletonConstant(final RegisterSingletonConstantInput input) {
+
+ LOG.debug("Received register-singleton-constant rpc, input: {}", input);
+
+ if (input.getConstant() == null) {
+ final RpcError error = RpcResultBuilder.newError(
+ ErrorType.RPC, "Invalid input.", "Constant value is null");
+ return Futures.immediateFuture(RpcResultBuilder.<Void>failed().withRpcError(error).build());
+ }
+
+ getSingletonConstantRegistration =
+ SingletonGetConstantService.registerNew(singletonService, domRpcService, input.getConstant());
+
+ return Futures.immediateFuture(RpcResultBuilder.<Void>success().build());
}
@Override
@Override
public Future<RpcResult<UnregisterFlappingSingletonOutput>> unregisterFlappingSingleton() {
- return null;
+ LOG.debug("unregister-flapping-singleton received.");
+
+ if (flappingSingletonService == null) {
+ final RpcError rpcError = RpcResultBuilder
+ .newError(ErrorType.APPLICATION, "missing-registration", "No flapping-singleton registration present.");
+ final RpcResult<UnregisterFlappingSingletonOutput> result =
+ RpcResultBuilder.<UnregisterFlappingSingletonOutput>failed().withRpcError(rpcError).build();
+ return Futures.immediateFuture(result);
+ }
+
+ final long flapCount = flappingSingletonService.setInactive();
+ flappingSingletonService = null;
+
+ final UnregisterFlappingSingletonOutput output =
+ new UnregisterFlappingSingletonOutputBuilder().setFlapCount(flapCount).build();
+
+ return Futures.immediateFuture(RpcResultBuilder.success(output).build());
}
@Override
}
@Override
- public Future<RpcResult<Void>> registerBoundConstant(RegisterBoundConstantInput input) {
- return null;
+ public Future<RpcResult<Void>> registerBoundConstant(final RegisterBoundConstantInput input) {
+ LOG.debug("register-bound-constant: {}", input);
+
+ if (input.getContext() == null) {
+ final RpcError error = RpcResultBuilder.newError(
+ ErrorType.RPC, "Invalid input.", "Context value is null");
+ return Futures.immediateFuture(RpcResultBuilder.<Void>failed().withRpcError(error).build());
+ }
+
+ if (input.getConstant() == null) {
+ final RpcError error = RpcResultBuilder.newError(
+ ErrorType.RPC, "Invalid input.", "Constant value is null");
+ return Futures.immediateFuture(RpcResultBuilder.<Void>failed().withRpcError(error).build());
+ }
+
+ if (routedRegistrations.containsKey(input.getContext())) {
+ final RpcError error = RpcResultBuilder.newError(ErrorType.RPC, "Registration present.",
+ "There is already a rpc registered for context: " + input.getContext());
+ return Futures.immediateFuture(RpcResultBuilder.<Void>failed().withRpcError(error).build());
+ }
+
+ final DOMRpcImplementationRegistration<RoutedGetConstantService> registration =
+ RoutedGetConstantService.registerNew(bindingNormalizedNodeSerializer, domRpcService,
+ input.getConstant(), input.getContext());
+
+ routedRegistrations.put(input.getContext(), registration);
+ return Futures.immediateFuture(RpcResultBuilder.<Void>success().build());
}
@Override
public Future<RpcResult<Void>> registerFlappingSingleton() {
- return null;
+ LOG.debug("Received register-flapping-singleton.");
+
+ if (flappingSingletonService != null) {
+ final RpcError error = RpcResultBuilder.newError(
+ ErrorType.RPC, "Registration present.", "flappin-singleton already registered");
+ return Futures.immediateFuture(RpcResultBuilder.<Void>failed().withRpcError(error).build());
+ }
+
+ flappingSingletonService = new FlappingSingletonService(singletonService);
+
+ return Futures.immediateFuture(RpcResultBuilder.<Void>success().build());
}
@Override
}
@Override
- public Future<RpcResult<UnsubscribeYnlOutput>> unsubscribeYnl(UnsubscribeYnlInput input) {
- return null;
+ public Future<RpcResult<UnsubscribeYnlOutput>> unsubscribeYnl(final UnsubscribeYnlInput input) {
+ LOG.debug("Received unsubscribe-ynl, input: {}", input);
+
+ if (!ynlRegistrations.containsKey(input.getId())) {
+ final RpcError rpcError = RpcResultBuilder
+ .newError(ErrorType.APPLICATION, "missing-registration", "No ynl listener with this id registered.");
+ final RpcResult<UnsubscribeYnlOutput> result =
+ RpcResultBuilder.<UnsubscribeYnlOutput>failed().withRpcError(rpcError).build();
+ return Futures.immediateFuture(result);
+ }
+
+ final ListenerRegistration<YnlListener> registration = ynlRegistrations.remove(input.getId());
+ final UnsubscribeYnlOutput output = registration.getInstance().getOutput();
+
+ registration.close();
+
+ return Futures.immediateFuture(RpcResultBuilder.<UnsubscribeYnlOutput>success().withResult(output).build());
}
@Override