Examples with CacheSupplier co.cask.cdap.data2.transaction.coprocessor.CacheSupplier used on opensource projects

Example 1 with CacheSupplier

use of co.cask.cdap.data2.transaction.coprocessor.CacheSupplier in project cdap by caskdata.

the class ReferenceCountedSupplierTests method testGetSupplier.

/**
   * Starts a few threads to perform get operation on the given {@link CacheSupplier}s. It verifies that subsequent
   * 'get' operation returns the same instance as it did during the first invocation. And it also verifies that the
   * service is running.
   *
   * @param supplierList list of {@link CacheSupplier}
   * @throws Exception if an error occurs during testing
   */
private void testGetSupplier(final List<CacheSupplier> supplierList) throws Exception {
    // Get one instance now, for later comparisons
    final List<Service> serviceList = new ArrayList<>();
    for (CacheSupplier supplier : supplierList) {
        serviceList.add(supplier.get());
    }
    final AtomicInteger numOps = new AtomicInteger(NUM_OPS);
    final Random random = new Random(System.currentTimeMillis());
    // Start threads that will 'get' DummyService
    ExecutorService executor = Executors.newFixedThreadPool(NUM_THREADS);
    List<Future> futureList = new ArrayList<>();
    for (int i = 0; i < NUM_THREADS; i++) {
        futureList.add(executor.submit(new Runnable() {

            @Override
            public void run() {
                // Perform NUM_OPS 'gets' of DummyService
                while (numOps.decrementAndGet() > 0) {
                    for (int i = 0; i < supplierList.size(); i++) {
                        CacheSupplier supplier = supplierList.get(i);
                        Service newService = supplier.get();
                        Assert.assertTrue(newService == serviceList.get(i));
                        Assert.assertTrue(newService.isRunning());
                    }
                    int waitTime = random.nextInt(10);
                    try {
                        TimeUnit.MICROSECONDS.sleep(waitTime);
                    } catch (InterruptedException e) {
                        LOG.warn("Received an exception.", e);
                    }
                }
            }
        }));
    }
    for (Future future : futureList) {
        future.get(5, TimeUnit.SECONDS);
    }
    executor.shutdown();
    executor.awaitTermination(2, TimeUnit.SECONDS);
}

Example 2 with CacheSupplier

use of co.cask.cdap.data2.transaction.coprocessor.CacheSupplier in project cdap by caskdata.

the class ReferenceCountedSupplierTests method testReleaseSupplier.

/**
   * Starts a few threads to perform release operation on the given {@link CacheSupplier}s. It verifies that the
   * services are still running until the last release operation is performed after which they are checked to see
   * if the services have stopped.
   *
   * @param supplierList list of {@link CacheSupplier}
   * @throws Exception if an error occurs during testing
   */
private void testReleaseSupplier(final List<CacheSupplier> supplierList) throws Exception {
    final AtomicInteger numOps = new AtomicInteger(NUM_OPS);
    final Random random = new Random(System.currentTimeMillis());
    final List<Service> serviceList = new ArrayList<>();
    for (CacheSupplier supplier : supplierList) {
        serviceList.add(supplier.get());
        supplier.release();
    }
    // Start threads that will 'release' DummyService
    ExecutorService executor = Executors.newFixedThreadPool(NUM_THREADS);
    List<Future> futureList = new ArrayList<>();
    numOps.set(NUM_OPS);
    for (int i = 0; i < NUM_THREADS; i++) {
        futureList.add(executor.submit(new Runnable() {

            @Override
            public void run() {
                // DummyService
                while (numOps.decrementAndGet() > 0) {
                    for (int i = 0; i < supplierList.size(); i++) {
                        CacheSupplier supplier = supplierList.get(i);
                        supplier.release();
                        Assert.assertTrue(serviceList.get(i).isRunning());
                    }
                    try {
                        TimeUnit.MICROSECONDS.sleep(random.nextInt(10));
                    } catch (InterruptedException e) {
                        LOG.warn("Received an exception.", e);
                    }
                }
            }
        }));
    }
    for (Future future : futureList) {
        future.get(1, TimeUnit.SECONDS);
    }
    executor.shutdown();
    executor.awaitTermination(2, TimeUnit.SECONDS);
    // Verify that the DummyService is still running.
    for (Service service : serviceList) {
        Assert.assertTrue(service.isRunning());
    }
    // Since we got one instance in the beginning, we need to release it
    for (CacheSupplier supplier : supplierList) {
        supplier.release();
    }
    // Verify that the DummyService is shutdown.
    for (Service service : serviceList) {
        Assert.assertFalse(service.isRunning());
    }
    // Release it again but it should not cause any problem and the service should still be stopped.
    for (CacheSupplier supplier : supplierList) {
        supplier.release();
    }
    // Verify that the DummyService is shutdown.
    for (Service service : serviceList) {
        Assert.assertFalse(service.isRunning());
    }
}

Example 3 with CacheSupplier

use of co.cask.cdap.data2.transaction.coprocessor.CacheSupplier in project cdap by caskdata.

the class ReferenceCountedSupplierTests method testSupplier.

@Test
public void testSupplier() throws Exception {
    // ReferenceCountSupplier is created only once per CacheSupplier class. Thus it is fine to create separate
    // CacheSupplier objects
    CacheSupplier cConfSupplier = new CConfigurationCacheSupplier(null, null, DUMMY_PROPERTY, Integer.MAX_VALUE);
    testGetSupplier(Lists.newArrayList(cConfSupplier));
    testReleaseSupplier(Lists.newArrayList(cConfSupplier));
    // Test to make sure we handle more releases than gets.
    for (int i = 0; i < 10; i++) {
        cConfSupplier.release();
    }
    // Test increase and decrease of counts for cConf supplier
    testGetSupplier(Lists.<CacheSupplier>newArrayList(new CConfigurationCacheSupplier(null, null, DUMMY_PROPERTY, Integer.MAX_VALUE)));
    testReleaseSupplier(Lists.<CacheSupplier>newArrayList(new CConfigurationCacheSupplier(null, null, DUMMY_PROPERTY, Integer.MAX_VALUE)));
    // Repeat the tests for TopicMetadata CacheSupplier
    testGetSupplier(Lists.<CacheSupplier>newArrayList(new TopicMetadataCacheSupplier(null, null, null, null, null)));
    testReleaseSupplier(Lists.<CacheSupplier>newArrayList(new TopicMetadataCacheSupplier(null, null, null, null, null)));
    testGetSupplier(Lists.<CacheSupplier>newArrayList(new TopicMetadataCacheSupplier(null, null, null, null, null)));
    testReleaseSupplier(Lists.<CacheSupplier>newArrayList(new TopicMetadataCacheSupplier(null, null, null, null, null)));
    // Tests multiple suppliers at the same time
    List<CacheSupplier> cacheSupplierList = new ArrayList<>();
    cacheSupplierList.add(new CConfigurationCacheSupplier(null, null, DUMMY_PROPERTY, Integer.MAX_VALUE));
    cacheSupplierList.add(new TopicMetadataCacheSupplier(null, null, null, null, null));
    testGetSupplier(cacheSupplierList);
    testReleaseSupplier(cacheSupplierList);
}