Search in sources :

Example 6 with RegisterDataTreeNotificationListenerReply

use of org.opendaylight.controller.cluster.datastore.messages.RegisterDataTreeNotificationListenerReply 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();
        }
    };
}
Also used : Configuration(org.opendaylight.controller.cluster.datastore.config.Configuration) Dispatchers(org.opendaylight.controller.cluster.common.actor.Dispatchers) MoreExecutors(com.google.common.util.concurrent.MoreExecutors) Terminated(akka.actor.Terminated) NotInitializedException(org.opendaylight.controller.cluster.datastore.exceptions.NotInitializedException) TestModel(org.opendaylight.controller.md.cluster.datastore.model.TestModel) AsyncDataChangeListener(org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeListener) Timeout(akka.util.Timeout) LocalShardFound(org.opendaylight.controller.cluster.datastore.messages.LocalShardFound) Answer(org.mockito.stubbing.Answer) LocalShardNotFound(org.opendaylight.controller.cluster.datastore.messages.LocalShardNotFound) DoNothingActor(org.opendaylight.controller.cluster.raft.utils.DoNothingActor) ActorRef(akka.actor.ActorRef) Matchers.eq(org.mockito.Matchers.eq) Mockito.doAnswer(org.mockito.Mockito.doAnswer) ActorContext(org.opendaylight.controller.cluster.datastore.utils.ActorContext) YangInstanceIdentifier(org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier) CloseDataTreeNotificationListenerRegistration(org.opendaylight.controller.cluster.datastore.messages.CloseDataTreeNotificationListenerRegistration) Mockito.doReturn(org.mockito.Mockito.doReturn) AsyncDataBroker(org.opendaylight.controller.md.sal.common.api.data.AsyncDataBroker) Uninterruptibles(com.google.common.util.concurrent.Uninterruptibles) Future(scala.concurrent.Future) FiniteDuration(scala.concurrent.duration.FiniteDuration) Test(org.junit.Test) ClusteredDOMDataChangeListener(org.opendaylight.controller.md.sal.dom.api.ClusteredDOMDataChangeListener) TestKit(akka.testkit.javadsl.TestKit) ExecutionContexts(akka.dispatch.ExecutionContexts) Matchers.any(org.mockito.Matchers.any) Futures(akka.dispatch.Futures) TimeUnit(java.util.concurrent.TimeUnit) Mockito(org.mockito.Mockito) RegisterDataTreeNotificationListenerReply(org.opendaylight.controller.cluster.datastore.messages.RegisterDataTreeNotificationListenerReply) RegisterChangeListener(org.opendaylight.controller.cluster.datastore.messages.RegisterChangeListener) DataChangeScope(org.opendaylight.controller.md.sal.common.api.data.AsyncDataBroker.DataChangeScope) ActorSystem(akka.actor.ActorSystem) ExecutionContextExecutor(scala.concurrent.ExecutionContextExecutor) FindLocalShard(org.opendaylight.controller.cluster.datastore.messages.FindLocalShard) Assert(org.junit.Assert) Props(akka.actor.Props) NormalizedNode(org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode) Mockito.mock(org.mockito.Mockito.mock) RegisterDataTreeNotificationListenerReply(org.opendaylight.controller.cluster.datastore.messages.RegisterDataTreeNotificationListenerReply) ActorRef(akka.actor.ActorRef) Timeout(akka.util.Timeout) Future(scala.concurrent.Future) TestKit(akka.testkit.javadsl.TestKit) ActorContext(org.opendaylight.controller.cluster.datastore.utils.ActorContext) Test(org.junit.Test)

Example 7 with RegisterDataTreeNotificationListenerReply

use of org.opendaylight.controller.cluster.datastore.messages.RegisterDataTreeNotificationListenerReply in project controller by opendaylight.

the class DataTreeChangeListenerSupportTest method registerChangeListener.

@SuppressWarnings("checkstyle:IllegalCatch")
private Entry<MockDataTreeChangeListener, ActorSelection> registerChangeListener(final YangInstanceIdentifier path, final int expectedEvents) {
    MockDataTreeChangeListener listener = new MockDataTreeChangeListener(expectedEvents);
    ActorRef dclActor = actorFactory.createActor(DataTreeChangeListenerActor.props(listener, TestModel.TEST_PATH));
    RegisterDataTreeNotificationListenerReply reply;
    try {
        reply = (RegisterDataTreeNotificationListenerReply) Await.result(Patterns.ask(shardActor, new RegisterDataTreeChangeListener(path, dclActor, false), new Timeout(5, TimeUnit.SECONDS)), Duration.create(5, TimeUnit.SECONDS));
    } catch (RuntimeException e) {
        throw e;
    } catch (Exception e) {
        throw new RuntimeException(e);
    }
    return new SimpleEntry<>(listener, getSystem().actorSelection(reply.getListenerRegistrationPath()));
}
Also used : RegisterDataTreeNotificationListenerReply(org.opendaylight.controller.cluster.datastore.messages.RegisterDataTreeNotificationListenerReply) MockDataTreeChangeListener(org.opendaylight.controller.cluster.datastore.utils.MockDataTreeChangeListener) ActorRef(akka.actor.ActorRef) TestActorRef(akka.testkit.TestActorRef) Timeout(akka.util.Timeout) SimpleEntry(java.util.AbstractMap.SimpleEntry) RegisterDataTreeChangeListener(org.opendaylight.controller.cluster.datastore.messages.RegisterDataTreeChangeListener) DataValidationFailedException(org.opendaylight.yangtools.yang.data.api.schema.tree.DataValidationFailedException)

Example 8 with RegisterDataTreeNotificationListenerReply

use of org.opendaylight.controller.cluster.datastore.messages.RegisterDataTreeNotificationListenerReply in project controller by opendaylight.

the class AbstractDataListenerSupport method onMessage.

@Override
void onMessage(M message, boolean isLeader, boolean hasLeader) {
    log.debug("{}: {} for {}, isLeader: {}, hasLeader: {}", persistenceId(), logName(), message, isLeader, hasLeader);
    ActorRef registrationActor = createActor(DataTreeNotificationListenerRegistrationActor.props());
    if (hasLeader && message.isRegisterOnAllInstances() || isLeader) {
        doRegistration(message, registrationActor);
    } else {
        log.debug("{}: Shard is not the leader - delaying registration", persistenceId());
        DelayedListenerRegistration<L, M> delayedReg = new DelayedListenerRegistration<>(message, registrationActor);
        Collection<DelayedListenerRegistration<L, M>> delayedRegList;
        if (message.isRegisterOnAllInstances()) {
            delayedRegList = delayedListenerOnAllRegistrations;
        } else {
            delayedRegList = delayedListenerRegistrations;
        }
        delayedRegList.add(delayedReg);
        registrationActor.tell(new DataTreeNotificationListenerRegistrationActor.SetRegistration(delayedReg, () -> delayedRegList.remove(delayedReg)), ActorRef.noSender());
    }
    log.debug("{}: {} sending reply, listenerRegistrationPath = {} ", persistenceId(), logName(), registrationActor.path());
    tellSender(new RegisterDataTreeNotificationListenerReply(registrationActor));
}
Also used : RegisterDataTreeNotificationListenerReply(org.opendaylight.controller.cluster.datastore.messages.RegisterDataTreeNotificationListenerReply) ActorRef(akka.actor.ActorRef) DataTreeNotificationListenerRegistrationActor(org.opendaylight.controller.cluster.datastore.actors.DataTreeNotificationListenerRegistrationActor)

Example 9 with RegisterDataTreeNotificationListenerReply

use of org.opendaylight.controller.cluster.datastore.messages.RegisterDataTreeNotificationListenerReply in project controller by opendaylight.

the class DataTreeChangeListenerProxy method doRegistration.

private void doRegistration(final ActorRef shard) {
    Future<Object> future = actorContext.executeOperationAsync(shard, new RegisterDataTreeChangeListener(registeredPath, dataChangeListenerActor, getInstance() instanceof ClusteredDOMDataTreeChangeListener), actorContext.getDatastoreContext().getShardInitializationTimeout());
    future.onComplete(new OnComplete<Object>() {

        @Override
        public void onComplete(final Throwable failure, final Object result) {
            if (failure != null) {
                LOG.error("{}: Failed to register DataTreeChangeListener {} at path {}", logContext(), getInstance(), registeredPath, failure);
            } else {
                RegisterDataTreeNotificationListenerReply reply = (RegisterDataTreeNotificationListenerReply) result;
                setListenerRegistrationActor(actorContext.actorSelection(reply.getListenerRegistrationPath()));
            }
        }
    }, actorContext.getClientDispatcher());
}
Also used : RegisterDataTreeNotificationListenerReply(org.opendaylight.controller.cluster.datastore.messages.RegisterDataTreeNotificationListenerReply) ClusteredDOMDataTreeChangeListener(org.opendaylight.mdsal.dom.api.ClusteredDOMDataTreeChangeListener) RegisterDataTreeChangeListener(org.opendaylight.controller.cluster.datastore.messages.RegisterDataTreeChangeListener)

Example 10 with RegisterDataTreeNotificationListenerReply

use of org.opendaylight.controller.cluster.datastore.messages.RegisterDataTreeNotificationListenerReply in project controller by opendaylight.

the class DataChangeListenerRegistrationProxy method doRegistration.

private void doRegistration(final ActorRef shard, final YangInstanceIdentifier path, final DataChangeScope scope) {
    Future<Object> future = actorContext.executeOperationAsync(shard, new RegisterChangeListener(path, dataChangeListenerActor, scope, listener instanceof ClusteredDOMDataChangeListener), actorContext.getDatastoreContext().getShardInitializationTimeout());
    future.onComplete(new OnComplete<Object>() {

        @Override
        public void onComplete(final Throwable failure, final Object result) {
            if (failure != null) {
                LOG.error("Failed to register DataChangeListener {} at path {}", listener, path.toString(), failure);
            } else {
                RegisterDataTreeNotificationListenerReply reply = (RegisterDataTreeNotificationListenerReply) result;
                setListenerRegistrationActor(actorContext.actorSelection(reply.getListenerRegistrationPath()));
            }
        }
    }, actorContext.getClientDispatcher());
}
Also used : RegisterDataTreeNotificationListenerReply(org.opendaylight.controller.cluster.datastore.messages.RegisterDataTreeNotificationListenerReply) RegisterChangeListener(org.opendaylight.controller.cluster.datastore.messages.RegisterChangeListener) ClusteredDOMDataChangeListener(org.opendaylight.controller.md.sal.dom.api.ClusteredDOMDataChangeListener)

Aggregations

RegisterDataTreeNotificationListenerReply (org.opendaylight.controller.cluster.datastore.messages.RegisterDataTreeNotificationListenerReply)18 Test (org.junit.Test)14 ActorRef (akka.actor.ActorRef)13 YangInstanceIdentifier (org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier)12 TestActorRef (akka.testkit.TestActorRef)10 RegisterDataTreeChangeListener (org.opendaylight.controller.cluster.datastore.messages.RegisterDataTreeChangeListener)9 RegisterChangeListener (org.opendaylight.controller.cluster.datastore.messages.RegisterChangeListener)8 MockDataTreeChangeListener (org.opendaylight.controller.cluster.datastore.utils.MockDataTreeChangeListener)6 TestKit (akka.testkit.javadsl.TestKit)5 Configuration (org.opendaylight.controller.cluster.datastore.config.Configuration)5 FindLocalShard (org.opendaylight.controller.cluster.datastore.messages.FindLocalShard)5 LocalShardFound (org.opendaylight.controller.cluster.datastore.messages.LocalShardFound)5 ActorContext (org.opendaylight.controller.cluster.datastore.utils.ActorContext)5 FiniteDuration (scala.concurrent.duration.FiniteDuration)5 MockDataChangeListener (org.opendaylight.controller.cluster.datastore.utils.MockDataChangeListener)4 Timeout (akka.util.Timeout)3 ActorSystem (akka.actor.ActorSystem)2 Props (akka.actor.Props)2 Terminated (akka.actor.Terminated)2 ExecutionContexts (akka.dispatch.ExecutionContexts)2