Search in sources :

Example 1 with QueryChangeLogSource

use of org.apache.rya.streams.querymanager.QueryChangeLogSource in project incubator-rya by apache.

the class KafkaQueryChangeLogSourceIT method unsubscribedDoesNotReceiveNotifications.

@Test
public void unsubscribedDoesNotReceiveNotifications() throws Exception {
    // Create the source.
    final QueryChangeLogSource source = new KafkaQueryChangeLogSource(kafka.getKafkaHostname(), Integer.parseInt(kafka.getKafkaPort()), Scheduler.newFixedRateSchedule(0, 100, TimeUnit.MILLISECONDS));
    try {
        // Start the source.
        source.startAndWait();
        // Create a listener that flips a boolean to true when it is notified.
        final AtomicBoolean notified = new AtomicBoolean(false);
        final SourceListener listener = new SourceListener() {

            @Override
            public void notifyCreate(final String ryaInstanceName, final QueryChangeLog log) {
                notified.set(true);
            }

            @Override
            public void notifyDelete(final String ryaInstanceName) {
                notified.set(true);
            }
        };
        // Register and then unregister it.
        source.subscribe(listener);
        source.unsubscribe(listener);
        // Create a topic.
        final String ryaInstance = UUID.randomUUID().toString();
        final String topic = KafkaTopics.queryChangeLogTopic(ryaInstance);
        kafka.createTopic(topic);
        // Wait longer than the polling time for the listener to be notified.
        Thread.sleep(300);
        // Show the boolean was never flipped to true.
        assertFalse(notified.get());
    } finally {
        source.stopAndWait();
    }
}
Also used : AtomicBoolean(java.util.concurrent.atomic.AtomicBoolean) SourceListener(org.apache.rya.streams.querymanager.QueryChangeLogSource.SourceListener) QueryChangeLogSource(org.apache.rya.streams.querymanager.QueryChangeLogSource) QueryChangeLog(org.apache.rya.streams.api.queries.QueryChangeLog) Test(org.junit.Test)

Example 2 with QueryChangeLogSource

use of org.apache.rya.streams.querymanager.QueryChangeLogSource in project incubator-rya by apache.

the class KafkaQueryChangeLogSourceIT method discoverExistingLogs.

@Test
public void discoverExistingLogs() throws Exception {
    // Create a valid Query Change Log topic.
    final String ryaInstance = UUID.randomUUID().toString();
    final String topic = KafkaTopics.queryChangeLogTopic(ryaInstance);
    kafka.createTopic(topic);
    // Create the source.
    final QueryChangeLogSource source = new KafkaQueryChangeLogSource(kafka.getKafkaHostname(), Integer.parseInt(kafka.getKafkaPort()), Scheduler.newFixedRateSchedule(0, 1, TimeUnit.SECONDS));
    // Register a listener that counts down a latch if it sees the new topic.
    final CountDownLatch created = new CountDownLatch(1);
    source.subscribe(new SourceListener() {

        @Override
        public void notifyCreate(final String ryaInstanceName, final QueryChangeLog log) {
            assertEquals(ryaInstance, ryaInstanceName);
            created.countDown();
        }

        @Override
        public void notifyDelete(final String ryaInstanceName) {
        }
    });
    try {
        // Start the source.
        source.startAndWait();
        // If the latch isn't counted down, then fail the test.
        assertTrue(created.await(5, TimeUnit.SECONDS));
    } finally {
        source.stopAndWait();
    }
}
Also used : SourceListener(org.apache.rya.streams.querymanager.QueryChangeLogSource.SourceListener) QueryChangeLogSource(org.apache.rya.streams.querymanager.QueryChangeLogSource) CountDownLatch(java.util.concurrent.CountDownLatch) QueryChangeLog(org.apache.rya.streams.api.queries.QueryChangeLog) Test(org.junit.Test)

Example 3 with QueryChangeLogSource

use of org.apache.rya.streams.querymanager.QueryChangeLogSource in project incubator-rya by apache.

the class KafkaQueryChangeLogSourceIT method discoverLogDeletions.

@Test
public void discoverLogDeletions() throws Exception {
    // Create a valid Query Change Log topic.
    final String ryaInstance = UUID.randomUUID().toString();
    final String topic = KafkaTopics.queryChangeLogTopic(ryaInstance);
    kafka.createTopic(topic);
    // Create the source.
    final QueryChangeLogSource source = new KafkaQueryChangeLogSource(kafka.getKafkaHostname(), Integer.parseInt(kafka.getKafkaPort()), Scheduler.newFixedRateSchedule(0, 1, TimeUnit.SECONDS));
    // Register a listener that uses latches to indicate when the topic is created and deleted.
    final CountDownLatch created = new CountDownLatch(1);
    final CountDownLatch deleted = new CountDownLatch(1);
    source.subscribe(new SourceListener() {

        @Override
        public void notifyCreate(final String ryaInstanceName, final QueryChangeLog log) {
            assertEquals(ryaInstance, ryaInstanceName);
            created.countDown();
        }

        @Override
        public void notifyDelete(final String ryaInstanceName) {
            assertEquals(ryaInstance, ryaInstanceName);
            deleted.countDown();
        }
    });
    try {
        // Start the source
        source.startAndWait();
        // Wait for it to indicate the topic was created.
        assertTrue(created.await(5, TimeUnit.SECONDS));
        // Delete the topic.
        kafka.deleteTopic(topic);
        // If the latch isn't counted down, then fail the test.
        assertTrue(deleted.await(5, TimeUnit.SECONDS));
    } finally {
        source.stopAndWait();
    }
}
Also used : SourceListener(org.apache.rya.streams.querymanager.QueryChangeLogSource.SourceListener) QueryChangeLogSource(org.apache.rya.streams.querymanager.QueryChangeLogSource) CountDownLatch(java.util.concurrent.CountDownLatch) QueryChangeLog(org.apache.rya.streams.api.queries.QueryChangeLog) Test(org.junit.Test)

Example 4 with QueryChangeLogSource

use of org.apache.rya.streams.querymanager.QueryChangeLogSource in project incubator-rya by apache.

the class KafkaQueryChangeLogSourceIT method newListenerReceivesAllKnownLogs.

@Test
public void newListenerReceivesAllKnownLogs() throws Exception {
    // Create a valid Query Change Log topic.
    final String ryaInstance = UUID.randomUUID().toString();
    final String topic = KafkaTopics.queryChangeLogTopic(ryaInstance);
    kafka.createTopic(topic);
    // Create the source.
    final QueryChangeLogSource source = new KafkaQueryChangeLogSource(kafka.getKafkaHostname(), Integer.parseInt(kafka.getKafkaPort()), Scheduler.newFixedRateSchedule(0, 1, TimeUnit.SECONDS));
    // Register a listener that counts down a latch if it sees the new topic.
    final CountDownLatch created = new CountDownLatch(1);
    source.subscribe(new SourceListener() {

        @Override
        public void notifyCreate(final String ryaInstanceName, final QueryChangeLog log) {
            assertEquals(ryaInstance, ryaInstanceName);
            created.countDown();
        }

        @Override
        public void notifyDelete(final String ryaInstanceName) {
        }
    });
    try {
        // Start the source
        source.startAndWait();
        // Wait for that first listener to indicate the topic was created. This means that one has been cached.
        assertTrue(created.await(5, TimeUnit.SECONDS));
        // Register a second listener that counts down when that same topic is encountered. This means the
        // newly subscribed listener was notified with the already known change log.
        final CountDownLatch newListenerCreated = new CountDownLatch(1);
        source.subscribe(new SourceListener() {

            @Override
            public void notifyCreate(final String ryaInstanceName, final QueryChangeLog log) {
                assertEquals(ryaInstance, ryaInstanceName);
                newListenerCreated.countDown();
            }

            @Override
            public void notifyDelete(final String ryaInstanceName) {
            }
        });
        assertTrue(newListenerCreated.await(5, TimeUnit.SECONDS));
    } finally {
        source.stopAndWait();
    }
}
Also used : SourceListener(org.apache.rya.streams.querymanager.QueryChangeLogSource.SourceListener) QueryChangeLogSource(org.apache.rya.streams.querymanager.QueryChangeLogSource) CountDownLatch(java.util.concurrent.CountDownLatch) QueryChangeLog(org.apache.rya.streams.api.queries.QueryChangeLog) Test(org.junit.Test)

Example 5 with QueryChangeLogSource

use of org.apache.rya.streams.querymanager.QueryChangeLogSource in project incubator-rya by apache.

the class KafkaQueryChangeLogSourceIT method discoverNewLogs.

@Test
public void discoverNewLogs() throws Exception {
    // Create the source.
    final QueryChangeLogSource source = new KafkaQueryChangeLogSource(kafka.getKafkaHostname(), Integer.parseInt(kafka.getKafkaPort()), Scheduler.newFixedRateSchedule(0, 100, TimeUnit.MILLISECONDS));
    // Register a listener that counts down a latch if it sees the new topic.
    final String ryaInstance = UUID.randomUUID().toString();
    final CountDownLatch created = new CountDownLatch(1);
    source.subscribe(new SourceListener() {

        @Override
        public void notifyCreate(final String ryaInstanceName, final QueryChangeLog log) {
            assertEquals(ryaInstance, ryaInstanceName);
            created.countDown();
        }

        @Override
        public void notifyDelete(final String ryaInstanceName) {
        }
    });
    try {
        // Start the source.
        source.startAndWait();
        // Wait twice the polling duration to ensure it iterates at least once.
        Thread.sleep(200);
        // Create a valid Query Change Log topic.
        final String topic = KafkaTopics.queryChangeLogTopic(ryaInstance);
        kafka.createTopic(topic);
        // If the latch isn't counted down, then fail the test.
        assertTrue(created.await(5, TimeUnit.SECONDS));
    } finally {
        source.stopAndWait();
    }
}
Also used : SourceListener(org.apache.rya.streams.querymanager.QueryChangeLogSource.SourceListener) QueryChangeLogSource(org.apache.rya.streams.querymanager.QueryChangeLogSource) CountDownLatch(java.util.concurrent.CountDownLatch) QueryChangeLog(org.apache.rya.streams.api.queries.QueryChangeLog) Test(org.junit.Test)

Aggregations

QueryChangeLog (org.apache.rya.streams.api.queries.QueryChangeLog)5 QueryChangeLogSource (org.apache.rya.streams.querymanager.QueryChangeLogSource)5 SourceListener (org.apache.rya.streams.querymanager.QueryChangeLogSource.SourceListener)5 Test (org.junit.Test)5 CountDownLatch (java.util.concurrent.CountDownLatch)4 AtomicBoolean (java.util.concurrent.atomic.AtomicBoolean)1