Examples with Interceptor org.apache.activemq.artemis.api.core.Interceptor used on opensource projects

Example 6 with Interceptor

use of org.apache.activemq.artemis.api.core.Interceptor in project activemq-artemis by apache.

the class PingTest method testClientFailureNoServerPing.

/*
   * Test the client triggering failure due to no ping from server received in time
   */
@Test
public void testClientFailureNoServerPing() throws Exception {
    // server must received at least one ping from the client to pass
    // so that the server connection TTL is configured with the client value
    final CountDownLatch pingOnServerLatch = new CountDownLatch(2);
    server.getRemotingService().addIncomingInterceptor(new Interceptor() {

        @Override
        public boolean intercept(final Packet packet, final RemotingConnection connection) throws ActiveMQException {
            if (packet.getType() == PacketImpl.PING) {
                pingOnServerLatch.countDown();
            }
            return true;
        }
    });
    TransportConfiguration transportConfig = new TransportConfiguration(INVM_CONNECTOR_FACTORY);
    ServerLocator locator = addServerLocator(ActiveMQClient.createServerLocatorWithoutHA(transportConfig));
    locator.setClientFailureCheckPeriod(PingTest.CLIENT_FAILURE_CHECK_PERIOD);
    locator.setConnectionTTL(PingTest.CLIENT_FAILURE_CHECK_PERIOD * 2);
    ClientSessionFactory csf = createSessionFactory(locator);
    ClientSession session = csf.createSession(false, true, true);
    Assert.assertEquals(1, ((ClientSessionFactoryInternal) csf).numConnections());
    final CountDownLatch clientLatch = new CountDownLatch(1);
    SessionFailureListener clientListener = new SessionFailureListener() {

        @Override
        public void connectionFailed(final ActiveMQException me, boolean failedOver) {
            clientLatch.countDown();
        }

        @Override
        public void connectionFailed(final ActiveMQException me, boolean failedOver, String scaleDownTargetNodeID) {
            connectionFailed(me, failedOver);
        }

        @Override
        public void beforeReconnect(final ActiveMQException exception) {
        }
    };
    final CountDownLatch serverLatch = new CountDownLatch(1);
    CloseListener serverListener = new CloseListener() {

        @Override
        public void connectionClosed() {
            serverLatch.countDown();
        }
    };
    session.addFailureListener(clientListener);
    CoreRemotingConnection serverConn = null;
    while (serverConn == null) {
        Set<RemotingConnection> conns = server.getRemotingService().getConnections();
        if (!conns.isEmpty()) {
            serverConn = (CoreRemotingConnection) server.getRemotingService().getConnections().iterator().next();
        } else {
            // It's async so need to wait a while
            Thread.sleep(10);
        }
    }
    serverConn.addCloseListener(serverListener);
    Assert.assertTrue("server has not received any ping from the client", pingOnServerLatch.await(4000, TimeUnit.MILLISECONDS));
    // we let the server receives at least 1 ping (so that it uses the client ConnectionTTL value)
    // Setting the handler to null will prevent server sending pings back to client
    serverConn.getChannel(0, -1).setHandler(null);
    Assert.assertTrue(clientLatch.await(8 * PingTest.CLIENT_FAILURE_CHECK_PERIOD, TimeUnit.MILLISECONDS));
    // Server connection will be closed too, when client closes client side connection after failure is detected
    Assert.assertTrue(serverLatch.await(2 * server.getConfiguration().getConnectionTtlCheckInterval(), TimeUnit.MILLISECONDS));
    long start = System.currentTimeMillis();
    while (true) {
        if (!server.getRemotingService().getConnections().isEmpty() && System.currentTimeMillis() - start < 10000) {
            Thread.sleep(500);
        } else {
            break;
        }
    }
    Assert.assertTrue(server.getRemotingService().getConnections().isEmpty());
    session.close();
    csf.close();
    locator.close();
}

Example 7 with Interceptor

use of org.apache.activemq.artemis.api.core.Interceptor in project activemq-artemis by apache.

the class PingTest method testNoPingingOnInVMConnection.

/*
    * Test that pinging is disabled for in-vm connection when using the default settings
    */
@Test
public void testNoPingingOnInVMConnection() throws Exception {
    // server should receive one and only one ping from the client so that
    // the server connection TTL is configured with the client value
    final CountDownLatch requiredPings = new CountDownLatch(1);
    final CountDownLatch unwantedPings = new CountDownLatch(2);
    server.getRemotingService().addIncomingInterceptor(new Interceptor() {

        @Override
        public boolean intercept(final Packet packet, final RemotingConnection connection) throws ActiveMQException {
            if (packet.getType() == PacketImpl.PING) {
                Assert.assertEquals(ActiveMQClient.DEFAULT_CONNECTION_TTL_INVM, ((Ping) packet).getConnectionTTL());
                unwantedPings.countDown();
                requiredPings.countDown();
            }
            return true;
        }
    });
    TransportConfiguration transportConfig = new TransportConfiguration("org.apache.activemq.artemis.core.remoting.impl.invm.InVMConnectorFactory");
    ServerLocator locator = addServerLocator(ActiveMQClient.createServerLocatorWithoutHA(transportConfig));
    ClientSessionFactory csf = createSessionFactory(locator);
    ClientSession session = csf.createSession(false, true, true);
    Assert.assertEquals(1, ((ClientSessionFactoryInternal) csf).numConnections());
    Assert.assertTrue("server didn't received an expected ping from the client", requiredPings.await(5000, TimeUnit.MILLISECONDS));
    Assert.assertFalse("server received an unexpected ping from the client", unwantedPings.await(ActiveMQClient.DEFAULT_CONNECTION_TTL * 2, TimeUnit.MILLISECONDS));
    session.close();
    csf.close();
    locator.close();
}

Example 8 with Interceptor

use of org.apache.activemq.artemis.api.core.Interceptor in project wildfly by wildfly.

the class ServerAdd method processOutgoingInterceptors.

private void processOutgoingInterceptors(ModelNode model, ActiveMQServerService serverService) throws OperationFailedException {
    if (!model.isDefined()) {
        return;
    }
    List<ModelNode> interceptors = model.asList();
    for (Class clazz : unwrapClasses(interceptors)) {
        try {
            Interceptor interceptor = Interceptor.class.cast(clazz.newInstance());
            serverService.getOutgoingInterceptors().add(interceptor);
        } catch (Exception e) {
            throw new OperationFailedException(e);
        }
    }
}

Example 9 with Interceptor

use of org.apache.activemq.artemis.api.core.Interceptor in project wildfly by wildfly.

the class ServerAdd method processIncomingInterceptors.

private void processIncomingInterceptors(ModelNode model, ActiveMQServerService serverService) throws OperationFailedException {
    if (!model.isDefined()) {
        return;
    }
    List<ModelNode> interceptors = model.asList();
    for (Class clazz : unwrapClasses(interceptors)) {
        try {
            Interceptor interceptor = Interceptor.class.cast(clazz.newInstance());
            serverService.getIncomingInterceptors().add(interceptor);
        } catch (Exception e) {
            throw new OperationFailedException(e);
        }
    }
}

Example 10 with Interceptor

use of org.apache.activemq.artemis.api.core.Interceptor in project activemq-artemis by apache.

the class ChannelImpl method invokeInterceptors.

/**
 * @param packet the packet to intercept
 * @return the name of the interceptor that returned <code>false</code> or <code>null</code> if no interceptors
 * returned <code>false</code>.
 */
public static String invokeInterceptors(final Packet packet, final List<Interceptor> interceptors, final RemotingConnection connection) {
    if (interceptors != null) {
        for (final Interceptor interceptor : interceptors) {
            try {
                boolean callNext = interceptor.intercept(packet, connection);
                if (logger.isDebugEnabled()) {
                    // use a StringBuilder for speed since this may be executed a lot
                    StringBuilder msg = new StringBuilder();
                    msg.append("Invocation of interceptor ").append(interceptor.getClass().getName()).append(" on ").append(packet).append(" returned ").append(callNext);
                    logger.debug(msg.toString());
                }
                if (!callNext) {
                    return interceptor.getClass().getName();
                }
            } catch (final Throwable e) {
                ActiveMQClientLogger.LOGGER.errorCallingInterceptor(e, interceptor);
            }
        }
    }
    return null;
}