package org.opendaylight.controller.clustering.it.provider;
+import static akka.actor.ActorRef.noSender;
+import static org.opendaylight.yangtools.yang.common.RpcResultBuilder.newError;
+
+import akka.actor.ActorRef;
+import akka.actor.ActorSystem;
+import akka.actor.PoisonPill;
+import akka.actor.Props;
+import akka.dispatch.OnComplete;
+import akka.pattern.Patterns;
+import com.google.common.base.Optional;
+import com.google.common.base.Strings;
+import com.google.common.util.concurrent.CheckedFuture;
import com.google.common.util.concurrent.Futures;
+import com.google.common.util.concurrent.ListenableFuture;
+import com.google.common.util.concurrent.SettableFuture;
+import java.io.PrintWriter;
+import java.io.StringWriter;
+import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
+import java.util.Objects;
import java.util.concurrent.Future;
+import java.util.concurrent.TimeUnit;
+import org.opendaylight.controller.cluster.ActorSystemProvider;
+import org.opendaylight.controller.cluster.databroker.actors.dds.ClientLocalHistory;
+import org.opendaylight.controller.cluster.databroker.actors.dds.ClientTransaction;
+import org.opendaylight.controller.cluster.databroker.actors.dds.DataStoreClient;
+import org.opendaylight.controller.cluster.databroker.actors.dds.SimpleDataStoreClientActor;
+import org.opendaylight.controller.cluster.datastore.DistributedDataStoreInterface;
+import org.opendaylight.controller.cluster.datastore.utils.ActorContext;
+import org.opendaylight.controller.cluster.datastore.utils.ClusterUtils;
+import org.opendaylight.controller.cluster.raft.client.messages.Shutdown;
+import org.opendaylight.controller.cluster.sharding.DistributedShardFactory;
+import org.opendaylight.controller.clustering.it.provider.impl.DataListenerState;
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.IdIntsDOMDataTreeLIstener;
+import org.opendaylight.controller.clustering.it.provider.impl.IdIntsListener;
+import org.opendaylight.controller.clustering.it.provider.impl.PrefixLeaderHandler;
+import org.opendaylight.controller.clustering.it.provider.impl.PrefixShardHandler;
+import org.opendaylight.controller.clustering.it.provider.impl.ProduceTransactionsHandler;
+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.md.sal.binding.impl.BindingToNormalizedNodeCodec;
-import org.opendaylight.controller.md.sal.binding.impl.BindingToNormalizedNodeCodecFactory;
+import org.opendaylight.controller.clustering.it.provider.impl.WriteTransactionsHandler;
+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.common.api.data.ReadFailedException;
+import org.opendaylight.controller.md.sal.dom.api.DOMDataBroker;
+import org.opendaylight.controller.md.sal.dom.api.DOMDataReadOnlyTransaction;
+import org.opendaylight.controller.md.sal.dom.api.DOMDataTreeChangeListener;
+import org.opendaylight.controller.md.sal.dom.api.DOMDataTreeChangeService;
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.binding.generator.impl.GeneratedClassLoadingStrategy;
+import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
+import org.opendaylight.mdsal.dom.api.DOMDataTreeIdentifier;
+import org.opendaylight.mdsal.dom.api.DOMDataTreeLoopException;
+import org.opendaylight.mdsal.dom.api.DOMDataTreeService;
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.CheckPublishNotificationsInput;
+import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.CheckPublishNotificationsOutput;
+import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.CheckPublishNotificationsOutputBuilder;
+import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.CreatePrefixShardInput;
import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.IsClientAbortedOutput;
import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.OdlMdsalLowlevelControlService;
import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.ProduceTransactionsInput;
-import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.PublishNotificationsInput;
+import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.ProduceTransactionsOutput;
import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.RegisterBoundConstantInput;
import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.RegisterConstantInput;
import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.RegisterDefaultConstantInput;
import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.RegisterSingletonConstantInput;
+import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.RemovePrefixShardInput;
import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.RemoveShardReplicaInput;
+import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.ShutdownPrefixShardReplicaInput;
+import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.ShutdownShardReplicaInput;
+import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.StartPublishNotificationsInput;
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.UnsubscribeDdtlOutputBuilder;
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.UnsubscribeDtclOutputBuilder;
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.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.control.rev170215.WriteTransactionsOutput;
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;
import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
-import org.opendaylight.yangtools.yang.model.api.SchemaContextListener;
+import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
+import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
+import scala.concurrent.duration.FiniteDuration;
public class MdsalLowLevelTestProvider implements OdlMdsalLowlevelControlService {
private static final Logger LOG = LoggerFactory.getLogger(MdsalLowLevelTestProvider.class);
+ private static final org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType CONTROLLER_CONFIG =
+ org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType.CONFIGURATION;
+
+ private static final ListenableFuture<RpcResult<Void>> VOID_SUCCESS = success(null);
private final RpcProviderRegistry rpcRegistry;
private final BindingAwareBroker.RpcRegistration<OdlMdsalLowlevelControlService> registration;
+ private final DistributedShardFactory distributedShardFactory;
+ private final DistributedDataStoreInterface configDataStore;
+ private final DOMDataTreeService domDataTreeService;
private final BindingNormalizedNodeSerializer bindingNormalizedNodeSerializer;
+ private final DOMDataBroker domDataBroker;
+ private final NotificationPublishService notificationPublishService;
+ private final NotificationService notificationService;
private final SchemaService schemaService;
private final ClusterSingletonServiceProvider singletonService;
private final DOMRpcProviderService domRpcService;
+ private final PrefixLeaderHandler prefixLeaderHandler;
+ private final PrefixShardHandler prefixShardHandler;
+ private final DOMDataTreeChangeService domDataTreeChangeService;
+ private final ActorSystem actorSystem;
- private Map<InstanceIdentifier<?>, DOMRpcImplementationRegistration<RoutedGetConstantService>> routedRegistrations =
+ private final Map<InstanceIdentifier<?>, DOMRpcImplementationRegistration<RoutedGetConstantService>> routedRegistrations =
new HashMap<>();
+ private final Map<String, ListenerRegistration<YnlListener>> ynlRegistrations = new HashMap<>();
+
private DOMRpcImplementationRegistration<GetConstantService> globalGetConstantRegistration = null;
private ClusterSingletonServiceRegistration getSingletonConstantRegistration;
private FlappingSingletonService flappingSingletonService;
+ private ListenerRegistration<DOMDataTreeChangeListener> dtclReg;
+ private IdIntsListener idIntsListener;
+ private final Map<String, PublishNotificationsTask> publishNotificationsTasks = new HashMap<>();
+ private ListenerRegistration<IdIntsDOMDataTreeLIstener> ddtlReg;
+ private IdIntsDOMDataTreeLIstener idIntsDdtl;
+
+
public MdsalLowLevelTestProvider(final RpcProviderRegistry rpcRegistry,
final DOMRpcProviderService domRpcService,
final ClusterSingletonServiceProvider singletonService,
final SchemaService schemaService,
- final BindingNormalizedNodeSerializer bindingNormalizedNodeSerializer) {
+ final BindingNormalizedNodeSerializer bindingNormalizedNodeSerializer,
+ final NotificationPublishService notificationPublishService,
+ final NotificationService notificationService,
+ final DOMDataBroker domDataBroker,
+ final DOMDataTreeService domDataTreeService,
+ final DistributedShardFactory distributedShardFactory,
+ final DistributedDataStoreInterface configDataStore,
+ final ActorSystemProvider actorSystemProvider) {
this.rpcRegistry = rpcRegistry;
this.domRpcService = domRpcService;
this.singletonService = singletonService;
this.schemaService = schemaService;
this.bindingNormalizedNodeSerializer = bindingNormalizedNodeSerializer;
+ this.notificationPublishService = notificationPublishService;
+ this.notificationService = notificationService;
+ this.domDataBroker = domDataBroker;
+ this.domDataTreeService = domDataTreeService;
+ this.distributedShardFactory = distributedShardFactory;
+ this.configDataStore = configDataStore;
+ this.actorSystem = actorSystemProvider.getActorSystem();
+
+ this.prefixLeaderHandler = new PrefixLeaderHandler(domDataTreeService, bindingNormalizedNodeSerializer);
+
+ domDataTreeChangeService =
+ (DOMDataTreeChangeService) domDataBroker.getSupportedExtensions().get(DOMDataTreeChangeService.class);
registration = rpcRegistry.addRpcImplementation(OdlMdsalLowlevelControlService.class, this);
+
+ prefixShardHandler = new PrefixShardHandler(distributedShardFactory, domDataTreeService,
+ bindingNormalizedNodeSerializer);
}
@Override
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);
+ return failure(newError(ErrorType.APPLICATION, "missing-registration",
+ "No get-singleton-constant rpc registration present."));
}
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);
+ return failure(newError(ErrorType.APPLICATION, "error-closing",
+ "There was a problem closing get-singleton-constant"));
+ } finally {
+ getSingletonConstantRegistration = null;
}
+
+ return VOID_SUCCESS;
}
@Override
- public Future<RpcResult<Void>> publishNotifications(PublishNotificationsInput input) {
- return null;
+ public Future<RpcResult<Void>> startPublishNotifications(final StartPublishNotificationsInput input) {
+ LOG.debug("publish-notifications, input: {}", input);
+
+ final PublishNotificationsTask task = new PublishNotificationsTask(notificationPublishService, input.getId(),
+ input.getSeconds(), input.getNotificationsPerSecond());
+
+ publishNotificationsTasks.put(input.getId(), task);
+
+ task.start();
+
+ return VOID_SUCCESS;
}
@Override
public Future<RpcResult<Void>> subscribeDtcl() {
- return null;
+ if (dtclReg != null) {
+ return failure(newError(ErrorType.RPC, "Registration present.",
+ "There is already dataTreeChangeListener registered on id-ints list."));
+ }
+
+ idIntsListener = new IdIntsListener();
+
+ dtclReg = domDataTreeChangeService
+ .registerDataTreeChangeListener(
+ new org.opendaylight.controller.md.sal.dom.api.DOMDataTreeIdentifier(
+ CONTROLLER_CONFIG, WriteTransactionsHandler.ID_INT_YID),
+ idIntsListener);
+
+ LOG.debug("ClusteredDOMDataTreeChangeListener registered");
+ return VOID_SUCCESS;
}
@Override
- public Future<RpcResult<Void>> writeTransactions(WriteTransactionsInput input) {
- return null;
+ public Future<RpcResult<WriteTransactionsOutput>> writeTransactions(final WriteTransactionsInput input) {
+ LOG.debug("write-transactions, input: {}", input);
+ return WriteTransactionsHandler.start(domDataBroker, input);
}
@Override
public Future<RpcResult<IsClientAbortedOutput>> isClientAborted() {
+ // FIXME: implement this
return null;
}
@Override
- public Future<RpcResult<Void>> becomeModuleLeader(BecomeModuleLeaderInput input) {
+ public Future<RpcResult<Void>> removeShardReplica(final RemoveShardReplicaInput input) {
+ // FIXME: implement this
return null;
}
@Override
- public Future<RpcResult<Void>> removeShardReplica(RemoveShardReplicaInput input) {
- return null;
+ public Future<RpcResult<Void>> subscribeYnl(final SubscribeYnlInput input) {
+ LOG.debug("subscribe-ynl, input: {}", input);
+
+ if (ynlRegistrations.containsKey(input.getId())) {
+ return failure(newError(ErrorType.RPC, "Registration present.",
+ "There is already ynl listener registered for this id: " + input.getId()));
+ }
+
+ ynlRegistrations.put(input.getId(),
+ notificationService.registerNotificationListener(new YnlListener(input.getId())));
+
+ return success(null);
}
@Override
- public Future<RpcResult<Void>> subscribeYnl(SubscribeYnlInput input) {
- return null;
+ public Future<RpcResult<Void>> removePrefixShard(final RemovePrefixShardInput input) {
+ LOG.debug("remove-prefix-shard, input: {}", input);
+
+ return prefixShardHandler.onRemovePrefixShard(input);
}
@Override
- public Future<RpcResult<Void>> becomePrefixLeader(BecomePrefixLeaderInput input) {
- return null;
+ public Future<RpcResult<Void>> becomePrefixLeader(final BecomePrefixLeaderInput input) {
+ LOG.debug("become-prefix-leader, input: {}", input);
+
+ return prefixLeaderHandler.makeLeaderLocal(input);
}
@Override
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);
+ return failure(newError(ErrorType.APPLICATION, "missing-registration",
+ "No get-constant rpc registration present."));
}
registration.close();
- return Futures.immediateFuture(RpcResultBuilder.<Void>success().build());
+ return VOID_SUCCESS;
}
@Override
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());
+ return failure(newError(ErrorType.RPC, "Invalid input.", "Constant value is null"));
}
getSingletonConstantRegistration =
SingletonGetConstantService.registerNew(singletonService, domRpcService, input.getConstant());
- return Futures.immediateFuture(RpcResultBuilder.<Void>success().build());
+ return VOID_SUCCESS;
}
@Override
- public Future<RpcResult<Void>> registerDefaultConstant(RegisterDefaultConstantInput input) {
+ public Future<RpcResult<Void>> registerDefaultConstant(final RegisterDefaultConstantInput input) {
+ // FIXME: implement this
return null;
}
public Future<RpcResult<Void>> unregisterConstant() {
if (globalGetConstantRegistration == null) {
- 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);
+ return failure(newError(ErrorType.APPLICATION, "missing-registration",
+ "No get-constant rpc registration present."));
}
globalGetConstantRegistration.close();
globalGetConstantRegistration = null;
- return Futures.immediateFuture(RpcResultBuilder.<Void>success().build());
+ return VOID_SUCCESS;
}
@Override
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);
+ return failure(newError(ErrorType.APPLICATION, "missing-registration",
+ "No flapping-singleton registration present."));
}
final long flapCount = flappingSingletonService.setInactive();
flappingSingletonService = null;
- final UnregisterFlappingSingletonOutput output =
- new UnregisterFlappingSingletonOutputBuilder().setFlapCount(flapCount).build();
-
- return Futures.immediateFuture(RpcResultBuilder.success(output).build());
+ return success(new UnregisterFlappingSingletonOutputBuilder().setFlapCount(flapCount).build());
}
@Override
- public Future<RpcResult<Void>> addShardReplica(AddShardReplicaInput input) {
+ public Future<RpcResult<Void>> addShardReplica(final AddShardReplicaInput input) {
+ // FIXME: implement this
return null;
}
@Override
public Future<RpcResult<Void>> subscribeDdtl() {
- return null;
+ if (ddtlReg != null) {
+ return failure(newError(ErrorType.RPC, "Registration present.",
+ "There is already dataTreeChangeListener registered on id-ints list."));
+ }
+
+ idIntsDdtl = new IdIntsDOMDataTreeLIstener();
+
+ try {
+ ddtlReg = domDataTreeService.registerListener(idIntsDdtl,
+ Collections.singleton(new DOMDataTreeIdentifier(LogicalDatastoreType.CONFIGURATION,
+ ProduceTransactionsHandler.ID_INT_YID)), true, Collections.emptyList());
+ } catch (DOMDataTreeLoopException e) {
+ LOG.error("Failed to register DOMDataTreeListener.", e);
+ return failure(newError(ErrorType.APPLICATION, "register-failed", e.getMessage()));
+ }
+
+ LOG.debug("DOMDataTreeListener registered");
+ return VOID_SUCCESS;
}
@Override
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());
+ return failure(newError(ErrorType.RPC, "Invalid input.", "Context value is null"));
}
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());
+ return failure(newError(ErrorType.RPC, "Invalid input.", "Constant value is null"));
}
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());
+ return failure(newError(ErrorType.RPC, "Registration present.",
+ "There is already a rpc registered for context: " + input.getContext()));
}
final DOMRpcImplementationRegistration<RoutedGetConstantService> registration =
input.getConstant(), input.getContext());
routedRegistrations.put(input.getContext(), registration);
- return Futures.immediateFuture(RpcResultBuilder.<Void>success().build());
+ return VOID_SUCCESS;
}
@Override
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());
+ return failure(RpcResultBuilder.newError(ErrorType.RPC, "Registration present.",
+ "flapping-singleton already registered"));
}
flappingSingletonService = new FlappingSingletonService(singletonService);
-
- return Futures.immediateFuture(RpcResultBuilder.<Void>success().build());
+ return VOID_SUCCESS;
}
@Override
public Future<RpcResult<UnsubscribeDtclOutput>> unsubscribeDtcl() {
- return null;
+ LOG.debug("Received unsubscribe-dtcl");
+
+ if (idIntsListener == null || dtclReg == null) {
+ return failure(newError(ErrorType.RPC, "Dtcl missing.", "No DataTreeChangeListener registered."));
+ }
+
+ final ListenableFuture<DataListenerState> future = idIntsListener.tryFinishProcessing(dtclReg);
+ dtclReg = null;
+
+ return Futures.withFallback(Futures.transform(future, this::dtclOutput),
+ t -> failure(newError(ErrorType.RPC, "resource-denied-transport", "Failed to finish processing",
+ "clustering-it", "clustering-it", t)));
+ }
+
+ private RpcResult<UnsubscribeDtclOutput> dtclOutput(final DataListenerState state) {
+ if (state.changeCount() == 0) {
+ return failed(newError(ErrorType.APPLICATION, "No notification received.",
+ "id-ints listener has not received any notifications."));
+ }
+
+ final DOMDataReadOnlyTransaction rTx = domDataBroker.newReadOnlyTransaction();
+ final Optional<NormalizedNode<?, ?>> readResult;
+
+ try {
+ readResult = rTx.read(CONTROLLER_CONFIG, WriteTransactionsHandler.ID_INT_YID).checkedGet();
+ } catch (final ReadFailedException e) {
+ return failed(newError(ErrorType.APPLICATION, "Read failed.", "Final read from id-ints failed."));
+ } finally {
+ rTx.close();
+ }
+
+ final NormalizedNode<?, ?> expected = state.lastData().orNull();
+ final NormalizedNode<?, ?> actual = readResult.orNull();
+ final boolean equal = Objects.equals(expected, actual);
+ if (!equal) {
+ LOG.debug("Expected result {} read resulted in {}", expected, actual);
+ }
+
+ final RpcResultBuilder<UnsubscribeDtclOutput> b = RpcResultBuilder.success(new UnsubscribeDtclOutputBuilder()
+ .setCopyMatches(equal).build());
+
+// for (DataListenerViolation violation : state.violations()) {
+// final Optional<NormalizedNodeDiff> diff = violation.toDiff();
+// if (diff.isPresent()) {
+// b.withWarning(ErrorType.APPLICATION, "Sequence mismatch", diff.get().toString());
+// }
+// }
+
+ return b.build();
+ }
+
+ @Override
+ public Future<RpcResult<Void>> createPrefixShard(final CreatePrefixShardInput input) {
+ LOG.debug("create-prefix-shard, input: {}", input);
+
+ return prefixShardHandler.onCreatePrefixShard(input);
}
@Override
public Future<RpcResult<Void>> deconfigureIdIntsShard() {
+ // FIXME: implement this
return null;
}
@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())) {
+ return failure(newError(ErrorType.APPLICATION, "missing-registration",
+ "No ynl listener with this id registered."));
+ }
+
+ final ListenerRegistration<YnlListener> registration = ynlRegistrations.remove(input.getId());
+ final UnsubscribeYnlOutput output = registration.getInstance().getOutput();
+
+ registration.close();
+
+ return success(output);
}
@Override
- public Future<RpcResult<Void>> produceTransactions(ProduceTransactionsInput input) {
- return null;
+ public Future<RpcResult<CheckPublishNotificationsOutput>> checkPublishNotifications(
+ final CheckPublishNotificationsInput input) {
+
+ final PublishNotificationsTask task = publishNotificationsTasks.get(input.getId());
+ if (task == null) {
+ return success(new CheckPublishNotificationsOutputBuilder().setActive(Boolean.FALSE).build());
+ }
+
+ final CheckPublishNotificationsOutputBuilder checkPublishNotificationsOutputBuilder =
+ new CheckPublishNotificationsOutputBuilder().setActive(!task.isFinished());
+
+ if (task.getLastError() != null) {
+ final StringWriter sw = new StringWriter();
+ final PrintWriter pw = new PrintWriter(sw);
+ task.getLastError().printStackTrace(pw);
+ checkPublishNotificationsOutputBuilder.setLastError(task.getLastError().toString() + sw.toString());
+ }
+
+ return success(checkPublishNotificationsOutputBuilder.setPublishCount(task.getCurrentNotif()).build());
+ }
+
+ @Override
+ public Future<RpcResult<ProduceTransactionsOutput>> produceTransactions(final ProduceTransactionsInput input) {
+ LOG.debug("producer-transactions, input: {}", input);
+ return ProduceTransactionsHandler.start(domDataTreeService, input);
+ }
+
+ @Override
+ public Future<RpcResult<Void>> shutdownShardReplica(final ShutdownShardReplicaInput input) {
+ LOG.debug("Received shutdown-shard-replica rpc, input: {}", input);
+
+ final String shardName = input.getShardName();
+ if (Strings.isNullOrEmpty(shardName)) {
+ return failure(newError(ErrorType.APPLICATION, "bad-element", "A valid shard name must be specified"));
+ }
+
+ return shutdownShardGracefully(shardName);
+ }
+
+ @Override
+ public Future<RpcResult<Void>> shutdownPrefixShardReplica(final ShutdownPrefixShardReplicaInput input) {
+ LOG.debug("Received shutdown-prefix-shard-replica rpc, input: {}", input);
+
+ final InstanceIdentifier<?> shardPrefix = input.getPrefix();
+ if (shardPrefix == null) {
+ return failure(newError(ErrorType.APPLICATION, "bad-element", "A valid shard prefix must be specified"));
+ }
+
+ final YangInstanceIdentifier shardPath = bindingNormalizedNodeSerializer.toYangInstanceIdentifier(shardPrefix);
+ final String cleanPrefixShardName = ClusterUtils.getCleanShardName(shardPath);
+
+ return shutdownShardGracefully(cleanPrefixShardName);
+ }
+
+ private SettableFuture<RpcResult<Void>> shutdownShardGracefully(final String shardName) {
+ final SettableFuture<RpcResult<Void>> rpcResult = SettableFuture.create();
+ final ActorContext context = configDataStore.getActorContext();
+
+ long timeoutInMS = Math.max(context.getDatastoreContext().getShardRaftConfig()
+ .getElectionTimeOutInterval().$times(3).toMillis(), 10000);
+ final FiniteDuration duration = FiniteDuration.apply(timeoutInMS, TimeUnit.MILLISECONDS);
+ final scala.concurrent.Promise<Boolean> shutdownShardAsk = akka.dispatch.Futures.promise();
+
+ context.findLocalShardAsync(shardName).onComplete(new OnComplete<ActorRef>() {
+ @Override
+ public void onComplete(final Throwable throwable, final ActorRef actorRef) throws Throwable {
+ if (throwable != null) {
+ shutdownShardAsk.failure(throwable);
+ } else {
+ shutdownShardAsk.completeWith(Patterns.gracefulStop(actorRef, duration, Shutdown.INSTANCE));
+ }
+ }
+ }, context.getClientDispatcher());
+
+ shutdownShardAsk.future().onComplete(new OnComplete<Boolean>() {
+ @Override
+ public void onComplete(final Throwable throwable, final Boolean gracefulStopResult) {
+ if (throwable != null) {
+ rpcResult.set(RpcResultBuilder.<Void>failed().withError(ErrorType.APPLICATION,
+ "Failed to gracefully shutdown shard", throwable).build());
+ } else {
+ // according to Patterns.gracefulStop API, we don't have to
+ // check value of gracefulStopResult
+ rpcResult.set(RpcResultBuilder.<Void>success().build());
+ }
+ }
+ }, context.getClientDispatcher());
+ return rpcResult;
}
@Override
LOG.debug("Received register-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());
+ return failure(newError(ErrorType.RPC, "Invalid input.", "Constant value is null"));
}
if (globalGetConstantRegistration != null) {
- final RpcError error = RpcResultBuilder.newError(ErrorType.RPC, "Registration present.",
- "There is already a get-constant rpc registered.");
- return Futures.immediateFuture(RpcResultBuilder.<Void>failed().withRpcError(error).build());
+ return failure(newError(ErrorType.RPC, "Registration present.",
+ "There is already a get-constant rpc registered."));
}
globalGetConstantRegistration = GetConstantService.registerNew(domRpcService, input.getConstant());
- return Futures.immediateFuture(RpcResultBuilder.<Void>success().build());
+ return VOID_SUCCESS;
}
@Override
public Future<RpcResult<Void>> unregisterDefaultConstant() {
+ // FIXME: implement this
return null;
}
@Override
public Future<RpcResult<UnsubscribeDdtlOutput>> unsubscribeDdtl() {
- return null;
+ LOG.debug("Received unsubscribe-ddtl.");
+
+ if (idIntsDdtl == null || ddtlReg == null) {
+ return failure(newError(ErrorType.RPC, "Ddtl missing.", "No DOMDataTreeListener registered."));
+ }
+
+ final ListenableFuture<DataListenerState> future = idIntsDdtl.tryFinishProcessing(ddtlReg);
+ ddtlReg = null;
+
+ return Futures.withFallback(Futures.transform(future, this::ddtlOutput),
+ t -> failure(newError(ErrorType.RPC, "resource-denied-transport", "Failed to finish processing",
+ "clustering-it", "clustering-it", t)));
+ }
+
+ private RpcResult<UnsubscribeDdtlOutput> ddtlOutput(final DataListenerState state) {
+ if (state.changeCount() == 0) {
+ return failed(newError(ErrorType.APPLICATION, "No notification received.",
+ "id-ints listener has not received any notifications."));
+ }
+
+ final String shardName = ClusterUtils.getCleanShardName(ProduceTransactionsHandler.ID_INTS_YID);
+ LOG.debug("Creating distributed datastore client for shard {}", shardName);
+
+ final ActorContext actorContext = configDataStore.getActorContext();
+ final Props distributedDataStoreClientProps = SimpleDataStoreClientActor.props(
+ actorContext.getCurrentMemberName(), "Shard-" + shardName, actorContext, shardName);
+
+ final ActorRef clientActor = actorSystem.actorOf(distributedDataStoreClientProps);
+ final DataStoreClient distributedDataStoreClient;
+ try {
+ distributedDataStoreClient = SimpleDataStoreClientActor.getDistributedDataStoreClient(clientActor, 30,
+ TimeUnit.SECONDS);
+ } catch (final Exception e) {
+ LOG.error("Failed to get actor for {}", distributedDataStoreClientProps, e);
+ clientActor.tell(PoisonPill.getInstance(), noSender());
+ return failed(newError(ErrorType.APPLICATION, "Unable to create ds client for read.",
+ "Unable to create ds client for read."));
+ }
+
+ final ClientLocalHistory localHistory = distributedDataStoreClient.createLocalHistory();
+ final ClientTransaction tx = localHistory.createTransaction();
+ final CheckedFuture<Optional<NormalizedNode<?, ?>>,
+ org.opendaylight.mdsal.common.api.ReadFailedException> read =
+ tx.read(YangInstanceIdentifier.of(ProduceTransactionsHandler.ID_INT));
+
+ tx.abort();
+ localHistory.close();
+
+ final Optional<NormalizedNode<?, ?>> readResult;
+ try {
+ readResult = read.checkedGet();
+ } catch (org.opendaylight.mdsal.common.api.ReadFailedException e) {
+ LOG.error("Unable to read data to verify ddtl data.", e);
+ return failed(newError( ErrorType.APPLICATION, "Read failed.", "Final read from id-ints failed."));
+ } finally {
+ distributedDataStoreClient.close();
+ clientActor.tell(PoisonPill.getInstance(), noSender());
+ }
+
+ // FIXME run a diff
+ final NormalizedNode<?, ?> expected = state.lastData().orNull();
+ final NormalizedNode<?, ?> actual = readResult.orNull();
+ final boolean equal = Objects.equals(expected, actual);
+ if (!equal) {
+ LOG.debug("Expected result {} read resulted in {}", expected, actual);
+ }
+ final RpcResultBuilder<UnsubscribeDdtlOutput> b = RpcResultBuilder.success(new UnsubscribeDdtlOutputBuilder()
+ .setCopyMatches(equal).build());
+
+// for (DataListenerViolation violation : state.violations()) {
+// final Optional<NormalizedNodeDiff> diff = violation.toDiff();
+// if (diff.isPresent()) {
+// b.withWarning(ErrorType.APPLICATION, "Sequence mismatch", diff.get().toString());
+// }
+// }
+
+ return b.build();
+ }
+
+ private static <T> RpcResult<T> failed(final RpcError error) {
+ return RpcResultBuilder.<T>failed().withRpcError(error).build();
+ }
+
+ private static <T> ListenableFuture<RpcResult<T>> failure(final RpcError error) {
+ return Futures.immediateFuture(failed(error));
+ }
+
+ private static <T> ListenableFuture<RpcResult<T>> success(final T result) {
+ return Futures.immediateFuture(RpcResultBuilder.success(result).build());
}
}
Code Review / controller.git / blobdiff