Search in sources :

Example 6 with PropertyStringSerializer

use of com.linkedin.d2.discovery.stores.PropertyStringSerializer in project rest.li by linkedin.

the class ZooKeeperChildrenDataPublisherTest method testPublishInitialize.

@Test
public void testPublishInitialize() throws InterruptedException, IOException, PropertyStoreException, ExecutionException {
    ZKConnection client = new ZKConnection("localhost:" + _port, 5000);
    client.start();
    final ZooKeeperChildrenDataPublisher<Map<String, String>, String> publisher = new ZooKeeperChildrenDataPublisher<Map<String, String>, String>(client, new PropertyStringSerializer(), "/");
    final CountDownLatch initLatch = new CountDownLatch(1);
    final CountDownLatch startLatch = new CountDownLatch(1);
    final PropertyEventSubscriber<Map<String, String>> subscriber = new PropertyEventSubscriber<Map<String, String>>() {

        @Override
        public void onInitialize(String propertyName, Map<String, String> propertyValue) {
            _outputData = propertyValue;
            initLatch.countDown();
        }

        @Override
        public void onAdd(String propertyName, Map<String, String> propertyValue) {
        }

        @Override
        public void onRemove(String propertyName) {
        }
    };
    publisher.start(new Callback<None>() {

        @Override
        public void onError(Throwable e) {
            Assert.fail("publisher start onError called", e);
        }

        @Override
        public void onSuccess(None result) {
            _eventBus = new PropertyEventBusImpl<Map<String, String>>(_executor, publisher);
            _eventBus.register(Collections.singleton("bucket"), subscriber);
            startLatch.countDown();
        }
    });
    if (!startLatch.await(60, TimeUnit.SECONDS)) {
        Assert.fail("unable to start ZookeeperChildrenDataPublisher");
    }
    if (!initLatch.await(60, TimeUnit.SECONDS)) {
        Assert.fail("unable to publish initial property value");
    }
    assertEquals(_outputData, _testData);
    _eventBus.unregister(Collections.singleton("bucket"), subscriber);
    client.shutdown();
}
Also used : PropertyEventSubscriber(com.linkedin.d2.discovery.event.PropertyEventSubscriber) CountDownLatch(java.util.concurrent.CountDownLatch) PropertyEventBusImpl(com.linkedin.d2.discovery.event.PropertyEventBusImpl) HashMap(java.util.HashMap) Map(java.util.Map) None(com.linkedin.common.util.None) Test(org.testng.annotations.Test)

Example 7 with PropertyStringSerializer

use of com.linkedin.d2.discovery.stores.PropertyStringSerializer in project rest.li by linkedin.

the class ZooKeeperEphemeralStoreStrawMan method main.

public static void main(String[] args) throws IOException, InterruptedException, PropertyStoreException {
    ZKConnection zkClient = new ZKConnection("localhost:2181", 30000);
    PropertyStringMerger merger = new PropertyStringMerger();
    Set<String> listenTos = new HashSet<String>();
    ZooKeeperEphemeralStore<String> zk = new ZooKeeperEphemeralStore<String>(zkClient, new PropertyStringSerializer(), merger, "/test/lb/test-property-ephemeral", false, true);
    listenTos.add("foo12");
    ScheduledExecutorService executorService = Executors.newSingleThreadScheduledExecutor();
    PropertyEventBus<String> bus = new PropertyEventBusImpl<String>(executorService, zk);
    bus.register(listenTos, new PropertyEventSubscriber<String>() {

        @Override
        public void onAdd(String propertyName, String propertyValue) {
            System.err.println("onAdd: " + propertyName + "\t" + propertyValue);
        }

        @Override
        public void onInitialize(String propertyName, String propertyValue) {
            System.err.println("onInitialize: " + propertyName + "\t" + propertyValue);
        }

        @Override
        public void onRemove(String propertyName) {
            System.err.println("onRemove: " + propertyName);
        }
    });
    zk.put("foo12", "TEST1");
    zk.put("foo12", "TEST2");
    zk.put("foo12", "TEST3");
    zk.put("foo12", "TEST4");
    zk.put("foo12", "TEST5");
    try {
        Thread.sleep(5000);
    } catch (InterruptedException e) {
        e.printStackTrace();
    }
    zk.remove("foo12");
    try {
        Thread.sleep(5000);
    } catch (InterruptedException e) {
        e.printStackTrace();
    }
    zkClient.getZooKeeper().close();
    executorService.shutdown();
}
Also used : PropertyStringMerger(com.linkedin.d2.discovery.stores.zk.ZooKeeperEphemeralStoreTest.PropertyStringMerger) ScheduledExecutorService(java.util.concurrent.ScheduledExecutorService) PropertyStringSerializer(com.linkedin.d2.discovery.stores.PropertyStringSerializer) PropertyEventBusImpl(com.linkedin.d2.discovery.event.PropertyEventBusImpl) HashSet(java.util.HashSet)

Example 8 with PropertyStringSerializer

use of com.linkedin.d2.discovery.stores.PropertyStringSerializer in project rest.li by linkedin.

the class ZooKeeperEphemeralStoreTest method getStore.

public ZooKeeperEphemeralStore<String> getStore() throws IOException, PropertyStoreException, InterruptedException, ExecutionException {
    ZKConnection client = new ZKConnection("localhost:" + _port, 5000);
    client.start();
    ZooKeeperEphemeralStore<String> store = new ZooKeeperEphemeralStore<String>(client, new PropertyStringSerializer(), new PropertyStringMerger(), "/test-path", false, true);
    FutureCallback<None> callback = new FutureCallback<None>();
    store.start(callback);
    callback.get();
    return store;
}
Also used : None(com.linkedin.common.util.None) PropertyStringSerializer(com.linkedin.d2.discovery.stores.PropertyStringSerializer) FutureCallback(com.linkedin.common.callback.FutureCallback)

Example 9 with PropertyStringSerializer

use of com.linkedin.d2.discovery.stores.PropertyStringSerializer in project rest.li by linkedin.

the class ZooKeeperPermanentStoreStrawMan method main.

public static void main(String[] args) throws IOException, InterruptedException, PropertyStoreException {
    ZKConnection zkClient = new ZKConnection("localhost:2181", 1000);
    Set<String> listenTos = new HashSet<String>();
    ZooKeeperPermanentStore<String> zk = new ZooKeeperPermanentStore<String>(zkClient, new PropertyStringSerializer(), "/test/lb/test-property");
    listenTos.add("foo12");
    ScheduledExecutorService executorService = Executors.newSingleThreadScheduledExecutor();
    PropertyEventBus<String> bus = new PropertyEventBusImpl<String>(executorService, zk);
    bus.register(listenTos, new PropertyEventSubscriber<String>() {

        @Override
        public void onAdd(String propertyName, String propertyValue) {
            System.err.println("onAdd: " + propertyName + "\t" + propertyValue);
        }

        @Override
        public void onInitialize(String propertyName, String propertyValue) {
            System.err.println("onInitialize: " + propertyName + "\t" + propertyValue);
        }

        @Override
        public void onRemove(String propertyName) {
            System.err.println("onRemove: " + propertyName);
        }
    });
    zk.put("foo12", "TEST1");
    zk.put("foo12", "TEST2");
    zk.put("foo12", "TEST3");
    zk.put("foo12", "TEST4");
    zk.put("foo12", "TEST5");
    zk.remove("foo12");
    try {
        Thread.sleep(10000);
    } catch (InterruptedException e) {
        e.printStackTrace();
    }
    zkClient.getZooKeeper().close();
    executorService.shutdown();
}
Also used : ScheduledExecutorService(java.util.concurrent.ScheduledExecutorService) PropertyStringSerializer(com.linkedin.d2.discovery.stores.PropertyStringSerializer) PropertyEventBusImpl(com.linkedin.d2.discovery.event.PropertyEventBusImpl) HashSet(java.util.HashSet)

Example 10 with PropertyStringSerializer

use of com.linkedin.d2.discovery.stores.PropertyStringSerializer in project rest.li by linkedin.

the class ZooKeeperPermanentStorePublisherTest method getStore.

@Override
protected ZooKeeperStore<String> getStore() {
    ZooKeeperPermanentStore<String> store = new ZooKeeperPermanentStore<String>(getConnection(), new PropertyStringSerializer(), "/testing/testPath");
    try {
        FutureCallback<None> callback = new FutureCallback<None>();
        store.start(callback);
        callback.get(30, TimeUnit.SECONDS);
    } catch (Exception e) {
        throw new RuntimeException(e);
    }
    return store;
}
Also used : ZooKeeperPermanentStore(com.linkedin.d2.discovery.stores.zk.ZooKeeperPermanentStore) None(com.linkedin.common.util.None) PropertyStringSerializer(com.linkedin.d2.discovery.stores.PropertyStringSerializer) FutureCallback(com.linkedin.common.callback.FutureCallback)

Aggregations

None (com.linkedin.common.util.None)8 FutureCallback (com.linkedin.common.callback.FutureCallback)7 PropertyEventBusImpl (com.linkedin.d2.discovery.event.PropertyEventBusImpl)6 PropertyStringSerializer (com.linkedin.d2.discovery.stores.PropertyStringSerializer)6 PropertyEventSubscriber (com.linkedin.d2.discovery.event.PropertyEventSubscriber)4 HashMap (java.util.HashMap)4 Map (java.util.Map)4 CountDownLatch (java.util.concurrent.CountDownLatch)4 Test (org.testng.annotations.Test)4 PropertyStoreException (com.linkedin.d2.discovery.stores.PropertyStoreException)2 HashSet (java.util.HashSet)2 ScheduledExecutorService (java.util.concurrent.ScheduledExecutorService)2 ZooKeeperEphemeralStore (com.linkedin.d2.discovery.stores.zk.ZooKeeperEphemeralStore)1 PropertyStringMerger (com.linkedin.d2.discovery.stores.zk.ZooKeeperEphemeralStoreTest.PropertyStringMerger)1 ZooKeeperPermanentStore (com.linkedin.d2.discovery.stores.zk.ZooKeeperPermanentStore)1 IOException (java.io.IOException)1 TimeoutException (java.util.concurrent.TimeoutException)1