Examples with MessageFilters org.apache.cassandra.distributed.shared.MessageFilters used on opensource projects

Example 1 with MessageFilters

use of org.apache.cassandra.distributed.shared.MessageFilters in project cassandra by apache.

the class MessageFiltersTest method simpleFiltersTest.

private static void simpleFiltersTest(boolean inbound) {
    int VERB1 = Verb.READ_REQ.id;
    int VERB2 = Verb.READ_RSP.id;
    int VERB3 = Verb.READ_REPAIR_REQ.id;
    int i1 = 1;
    int i2 = 2;
    int i3 = 3;
    String MSG1 = "msg1";
    String MSG2 = "msg2";
    MessageFilters filters = new MessageFilters();
    Permit permit = inbound ? filters::permitInbound : filters::permitOutbound;
    IMessageFilters.Filter filter = filters.allVerbs().inbound(inbound).from(1).drop();
    Assert.assertFalse(permit.test(i1, i2, msg(VERB1, MSG1)));
    Assert.assertFalse(permit.test(i1, i2, msg(VERB2, MSG1)));
    Assert.assertFalse(permit.test(i1, i2, msg(VERB3, MSG1)));
    Assert.assertTrue(permit.test(i2, i1, msg(VERB1, MSG1)));
    filter.off();
    Assert.assertTrue(permit.test(i1, i2, msg(VERB1, MSG1)));
    filters.reset();
    filters.verbs(VERB1).inbound(inbound).from(1).to(2).drop();
    Assert.assertFalse(permit.test(i1, i2, msg(VERB1, MSG1)));
    Assert.assertTrue(permit.test(i1, i2, msg(VERB2, MSG1)));
    Assert.assertTrue(permit.test(i2, i1, msg(VERB1, MSG1)));
    Assert.assertTrue(permit.test(i2, i3, msg(VERB2, MSG1)));
    filters.reset();
    AtomicInteger counter = new AtomicInteger();
    filters.verbs(VERB1).inbound(inbound).from(1).to(2).messagesMatching((from, to, msg) -> {
        counter.incrementAndGet();
        return Arrays.equals(msg.bytes(), MSG1.getBytes());
    }).drop();
    Assert.assertFalse(permit.test(i1, i2, msg(VERB1, MSG1)));
    Assert.assertEquals(counter.get(), 1);
    Assert.assertTrue(permit.test(i1, i2, msg(VERB1, MSG2)));
    Assert.assertEquals(counter.get(), 2);
    // filter chain gets interrupted because a higher level filter returns no match
    Assert.assertTrue(permit.test(i2, i1, msg(VERB1, MSG1)));
    Assert.assertEquals(counter.get(), 2);
    Assert.assertTrue(permit.test(i2, i1, msg(VERB2, MSG1)));
    Assert.assertEquals(counter.get(), 2);
    filters.reset();
    filters.allVerbs().inbound(inbound).from(3, 2).to(2, 1).drop();
    Assert.assertFalse(permit.test(i3, i1, msg(VERB1, MSG1)));
    Assert.assertFalse(permit.test(i3, i2, msg(VERB1, MSG1)));
    Assert.assertFalse(permit.test(i2, i1, msg(VERB1, MSG1)));
    Assert.assertTrue(permit.test(i2, i3, msg(VERB1, MSG1)));
    Assert.assertTrue(permit.test(i1, i2, msg(VERB1, MSG1)));
    Assert.assertTrue(permit.test(i1, i3, msg(VERB1, MSG1)));
    filters.reset();
    counter.set(0);
    filters.allVerbs().inbound(inbound).from(1).to(2).messagesMatching((from, to, msg) -> {
        counter.incrementAndGet();
        return false;
    }).drop();
    Assert.assertTrue(permit.test(i1, i2, msg(VERB1, MSG1)));
    Assert.assertTrue(permit.test(i1, i3, msg(VERB1, MSG1)));
    Assert.assertTrue(permit.test(i1, i2, msg(VERB1, MSG1)));
    Assert.assertEquals(2, counter.get());
}