Examples with ActorContext org.opendaylight.controller.cluster.datastore.utils.ActorContext used on opensource projects

Example 36 with ActorContext

use of org.opendaylight.controller.cluster.datastore.utils.ActorContext in project controller by opendaylight.

the class MdsalLowLevelTestProvider method shutdownShardGracefully.

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) throws Throwable {
            if (throwable != null) {
                final RpcResult<Void> failedResult = RpcResultBuilder.<Void>failed().withError(ErrorType.APPLICATION, "Failed to gracefully shutdown shard", throwable).build();
                rpcResult.set(failedResult);
            } 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;
}