Examples with CqAttributesMutator org.apache.geode.cache.query.CqAttributesMutator used on opensource projects

Example 1 with CqAttributesMutator

use of org.apache.geode.cache.query.CqAttributesMutator in project geode by apache.

the class CqQueryDUnitTest method mutateCQAttributes.

// Exercise CQ attributes mutator functions
private void mutateCQAttributes(VM vm, final String cqName, final int mutator_function) throws Exception {
    vm.invoke(new CacheSerializableRunnable("Stop CQ :" + cqName) {

        public void run2() throws CacheException {
            CqQuery cq1 = null;
            LogWriterUtils.getLogWriter().info("### CQ attributes mutator for ###" + cqName);
            // Get CQ Service.
            QueryService cqService = null;
            try {
                cqService = getCache().getQueryService();
            } catch (Exception cqe) {
                cqe.printStackTrace();
                fail("Failed to getCQService.");
            }
            // Get CQ.
            try {
                cq1 = cqService.getCq(cqName);
            } catch (Exception ex) {
                ex.printStackTrace();
                fail("Failed to get CQ " + cqName + " . " + ex.getMessage());
            }
            CqAttributesMutator cqAttrMutator = cq1.getCqAttributesMutator();
            CqAttributes cqAttr = cq1.getCqAttributes();
            CqListener[] cqListeners;
            switch(mutator_function) {
                case CREATE:
                    // Reinitialize with 2 CQ Listeners
                    CqListener[] cqListenersArray = { new CqQueryTestListener(getCache().getLogger()), new CqQueryTestListener(getCache().getLogger()) };
                    cqAttrMutator.initCqListeners(cqListenersArray);
                    cqListeners = cqAttr.getCqListeners();
                    assertEquals("CqListener count mismatch", cqListeners.length, 2);
                    break;
                case UPDATE:
                    // Add 2 new CQ Listeners
                    CqListener newListener1 = new CqQueryTestListener(getCache().getLogger());
                    CqListener newListener2 = new CqQueryTestListener(getCache().getLogger());
                    cqAttrMutator.addCqListener(newListener1);
                    cqAttrMutator.addCqListener(newListener2);
                    cqListeners = cqAttr.getCqListeners();
                    assertEquals("CqListener count mismatch", cqListeners.length, 3);
                    break;
                case DESTROY:
                    cqListeners = cqAttr.getCqListeners();
                    cqAttrMutator.removeCqListener(cqListeners[0]);
                    cqListeners = cqAttr.getCqListeners();
                    assertEquals("CqListener count mismatch", cqListeners.length, 2);
                    // Remove a listener and validate
                    cqAttrMutator.removeCqListener(cqListeners[0]);
                    cqListeners = cqAttr.getCqListeners();
                    assertEquals("CqListener count mismatch", cqListeners.length, 1);
                    break;
            }
        }
    });
}