use of org.opendaylight.controller.cluster.common.actor.Dispatchers in project controller by opendaylight.
the class ShardManager method removeShardReplica.
private void removeShardReplica(final RemoveShardReplica contextMessage, final String shardName, final String primaryPath, final ActorRef sender) {
if (isShardReplicaOperationInProgress(shardName, sender)) {
return;
}
shardReplicaOperationsInProgress.add(shardName);
final ShardIdentifier shardId = getShardIdentifier(contextMessage.getMemberName(), shardName);
final DatastoreContext datastoreContext = newShardDatastoreContextBuilder(shardName).build();
// inform ShardLeader to remove this shard as a replica by sending an RemoveServer message
LOG.debug("{}: Sending RemoveServer message to peer {} for shard {}", persistenceId(), primaryPath, shardId);
Timeout removeServerTimeout = new Timeout(datastoreContext.getShardLeaderElectionTimeout().duration());
Future<Object> futureObj = ask(getContext().actorSelection(primaryPath), new RemoveServer(shardId.toString()), removeServerTimeout);
futureObj.onComplete(new OnComplete<Object>() {
@Override
public void onComplete(final Throwable failure, final Object response) {
if (failure != null) {
shardReplicaOperationsInProgress.remove(shardName);
String msg = String.format("RemoveServer request to leader %s for shard %s failed", primaryPath, shardName);
LOG.debug("{}: {}", persistenceId(), msg, failure);
// FAILURE
sender.tell(new Status.Failure(new RuntimeException(msg, failure)), self());
} else {
// SUCCESS
self().tell(new WrappedShardResponse(shardId, response, primaryPath), sender);
}
}
}, new Dispatchers(context().system().dispatchers()).getDispatcher(Dispatchers.DispatcherType.Client));
}
use of org.opendaylight.controller.cluster.common.actor.Dispatchers in project controller by opendaylight.
the class DataChangeListenerRegistrationProxyTest method testCloseBeforeRegistration.
@Test
public void testCloseBeforeRegistration() {
new TestKit(getSystem()) {
{
ActorContext actorContext = mock(ActorContext.class);
String shardName = "shard-1";
final DataChangeListenerRegistrationProxy proxy = new DataChangeListenerRegistrationProxy(shardName, actorContext, mockListener);
doReturn(DatastoreContext.newBuilder().build()).when(actorContext).getDatastoreContext();
doReturn(getSystem().dispatchers().defaultGlobalDispatcher()).when(actorContext).getClientDispatcher();
doReturn(getSystem()).when(actorContext).getActorSystem();
doReturn(Dispatchers.DEFAULT_DISPATCHER_PATH).when(actorContext).getNotificationDispatcherPath();
doReturn(getSystem().actorSelection(getRef().path())).when(actorContext).actorSelection(getRef().path());
doReturn(duration("5 seconds")).when(actorContext).getOperationDuration();
doReturn(Futures.successful(getRef())).when(actorContext).findLocalShardAsync(eq(shardName));
Answer<Future<Object>> answer = invocation -> {
proxy.close();
return Futures.successful((Object) new RegisterDataTreeNotificationListenerReply(getRef()));
};
doAnswer(answer).when(actorContext).executeOperationAsync(any(ActorRef.class), any(Object.class), any(Timeout.class));
proxy.init(YangInstanceIdentifier.of(TestModel.TEST_QNAME), AsyncDataBroker.DataChangeScope.ONE);
expectMsgClass(duration("5 seconds"), CloseDataTreeNotificationListenerRegistration.class);
Assert.assertEquals("getListenerRegistrationActor", null, proxy.getListenerRegistrationActor());
proxy.close();
}
};
}
use of org.opendaylight.controller.cluster.common.actor.Dispatchers in project controller by opendaylight.
the class DispatchersTest method testGetDispatcherPath.
@Test
public void testGetDispatcherPath() {
akka.dispatch.Dispatchers mockDispatchers = mock(akka.dispatch.Dispatchers.class);
doReturn(true).when(mockDispatchers).hasDispatcher(anyString());
Dispatchers dispatchers = new Dispatchers(mockDispatchers);
assertEquals(Dispatchers.CLIENT_DISPATCHER_PATH, dispatchers.getDispatcherPath(Dispatchers.DispatcherType.Client));
assertEquals(Dispatchers.TXN_DISPATCHER_PATH, dispatchers.getDispatcherPath(Dispatchers.DispatcherType.Transaction));
assertEquals(Dispatchers.SHARD_DISPATCHER_PATH, dispatchers.getDispatcherPath(Dispatchers.DispatcherType.Shard));
assertEquals(Dispatchers.NOTIFICATION_DISPATCHER_PATH, dispatchers.getDispatcherPath(Dispatchers.DispatcherType.Notification));
}
use of org.opendaylight.controller.cluster.common.actor.Dispatchers in project controller by opendaylight.
the class AbstractShardDataTreeNotificationPublisherActorProxy method publisherActor.
protected final ActorRef publisherActor() {
if (publisherActor == null) {
String dispatcher = new Dispatchers(actorContext.system().dispatchers()).getDispatcherPath(Dispatchers.DispatcherType.Notification);
publisherActor = actorContext.actorOf(props().withDispatcher(dispatcher), actorName);
log.debug("{}: Created publisher actor {} with name {}", logContext, publisherActor, actorName);
}
return publisherActor;
}
use of org.opendaylight.controller.cluster.common.actor.Dispatchers in project controller by opendaylight.
the class DispatchersTest method testGetDefaultDispatcherPath.
@Test
public void testGetDefaultDispatcherPath() {
akka.dispatch.Dispatchers mockDispatchers = mock(akka.dispatch.Dispatchers.class);
doReturn(false).when(mockDispatchers).hasDispatcher(anyString());
Dispatchers dispatchers = new Dispatchers(mockDispatchers);
for (Dispatchers.DispatcherType type : Dispatchers.DispatcherType.values()) {
assertEquals(Dispatchers.DEFAULT_DISPATCHER_PATH, dispatchers.getDispatcherPath(type));
}
}
Aggregations