Search in sources :

Example 1 with PinpointClientReconnectEventListener

use of com.navercorp.pinpoint.rpc.client.PinpointClientReconnectEventListener in project pinpoint by naver.

the class AgentInfoSenderTest method agentInfoShouldRetryUntilAttemptsAreExhaustedWhenRefreshing.

@Test
public void agentInfoShouldRetryUntilAttemptsAreExhaustedWhenRefreshing() throws InterruptedException {
    final AtomicInteger successServerRequestCount = new AtomicInteger();
    final AtomicInteger failServerRequestCount = new AtomicInteger();
    final AtomicInteger successCount = new AtomicInteger();
    final long agentInfoSendRetryIntervalMs = 1000L;
    final long agentInfoSendRefreshIntervalMs = 5000L;
    final int expectedSuccessServerTries = 1;
    final int expectedFailServerTries = AgentInfoSender.DEFAULT_MAX_TRY_COUNT_PER_ATTEMPT;
    final CountDownLatch agentReconnectLatch = new CountDownLatch(1);
    ResponseServerMessageListener successServerListener = new ResponseServerMessageListener(successServerRequestCount, successCount);
    ResponseServerMessageListener failServerListener = new ResponseServerMessageListener(failServerRequestCount, successCount, Integer.MAX_VALUE);
    PinpointServerAcceptor successServerAcceptor = createServerAcceptor(successServerListener);
    PinpointServerAcceptor failServerAcceptor = null;
    PinpointClientFactory socketFactory = createPinpointClientFactory();
    PinpointClient pinpointClient = ClientFactoryUtils.createPinpointClient(HOST, PORT, socketFactory);
    TcpDataSender dataSender = new TcpDataSender(pinpointClient);
    dataSender.addReconnectEventListener(new PinpointClientReconnectEventListener() {

        @Override
        public void reconnectPerformed(PinpointClient client) {
            agentReconnectLatch.countDown();
        }
    });
    AgentInfoSender agentInfoSender = new AgentInfoSender.Builder(dataSender, getAgentInfo()).refreshInterval(agentInfoSendRefreshIntervalMs).sendInterval(agentInfoSendRetryIntervalMs).build();
    try {
        agentInfoSender.start();
        waitExpectedRequestCount(successServerRequestCount, expectedSuccessServerTries);
        successServerAcceptor.close();
        Thread.sleep(agentInfoSendRetryIntervalMs * AgentInfoSender.DEFAULT_MAX_TRY_COUNT_PER_ATTEMPT);
        failServerAcceptor = createServerAcceptor(failServerListener);
        // wait till agent reconnects
        agentReconnectLatch.await();
        waitExpectedRequestCount(failServerRequestCount, expectedFailServerTries);
        failServerAcceptor.close();
    } finally {
        closeAll(null, agentInfoSender, pinpointClient, socketFactory);
    }
    assertEquals(1, successCount.get());
    assertEquals(expectedSuccessServerTries, successServerRequestCount.get());
    assertEquals(expectedFailServerTries, failServerRequestCount.get());
}
Also used : PinpointClient(com.navercorp.pinpoint.rpc.client.PinpointClient) CountDownLatch(java.util.concurrent.CountDownLatch) PinpointClientFactory(com.navercorp.pinpoint.rpc.client.PinpointClientFactory) DefaultPinpointClientFactory(com.navercorp.pinpoint.rpc.client.DefaultPinpointClientFactory) AtomicInteger(java.util.concurrent.atomic.AtomicInteger) PinpointServerAcceptor(com.navercorp.pinpoint.rpc.server.PinpointServerAcceptor) TcpDataSender(com.navercorp.pinpoint.profiler.sender.TcpDataSender) PinpointClientReconnectEventListener(com.navercorp.pinpoint.rpc.client.PinpointClientReconnectEventListener) Test(org.junit.Test)

Example 2 with PinpointClientReconnectEventListener

use of com.navercorp.pinpoint.rpc.client.PinpointClientReconnectEventListener in project pinpoint by naver.

the class AgentInfoSenderTest method agentInfoShouldBeSentOnlyOnceEvenAfterReconnect.

@Test
public void agentInfoShouldBeSentOnlyOnceEvenAfterReconnect() throws Exception {
    final AtomicInteger requestCount = new AtomicInteger();
    final AtomicInteger successCount = new AtomicInteger();
    final AtomicInteger reconnectCount = new AtomicInteger();
    final int expectedReconnectCount = 3;
    final long agentInfoSendRetryIntervalMs = 100L;
    final int maxTryCountPerAttempt = Integer.MAX_VALUE;
    final CyclicBarrier reconnectEventBarrier = new CyclicBarrier(2);
    ResponseServerMessageListener serverListener = new ResponseServerMessageListener(requestCount, successCount);
    PinpointServerAcceptor serverAcceptor = createServerAcceptor(serverListener);
    PinpointClientFactory clientFactory = createPinpointClientFactory();
    PinpointClient pinpointClient = ClientFactoryUtils.createPinpointClient(HOST, PORT, clientFactory);
    TcpDataSender dataSender = new TcpDataSender(pinpointClient);
    dataSender.addReconnectEventListener(new PinpointClientReconnectEventListener() {

        @Override
        public void reconnectPerformed(PinpointClient client) {
            reconnectCount.incrementAndGet();
            try {
                reconnectEventBarrier.await();
            } catch (Exception e) {
                // just fail
                throw new RuntimeException(e);
            }
        }
    });
    AgentInfoSender agentInfoSender = new AgentInfoSender.Builder(dataSender, getAgentInfo()).sendInterval(agentInfoSendRetryIntervalMs).maxTryPerAttempt(maxTryCountPerAttempt).build();
    try {
        // initial connect
        agentInfoSender.start();
        waitExpectedRequestCount(requestCount, 1);
        serverAcceptor.close();
        // reconnect
        for (int i = 0; i < expectedReconnectCount; ++i) {
            PinpointServerAcceptor reconnectServerAcceptor = createServerAcceptor(serverListener);
            // wait for agent to reconnect
            reconnectEventBarrier.await();
            // wait to see if AgentInfo is sent again (it shouldn't)
            Thread.sleep(1000L);
            reconnectServerAcceptor.close();
            reconnectEventBarrier.reset();
        }
    } finally {
        closeAll(null, agentInfoSender, pinpointClient, clientFactory);
    }
    assertEquals(1, successCount.get());
    assertEquals(expectedReconnectCount, reconnectCount.get());
}
Also used : PinpointClient(com.navercorp.pinpoint.rpc.client.PinpointClient) TException(org.apache.thrift.TException) CyclicBarrier(java.util.concurrent.CyclicBarrier) PinpointClientFactory(com.navercorp.pinpoint.rpc.client.PinpointClientFactory) DefaultPinpointClientFactory(com.navercorp.pinpoint.rpc.client.DefaultPinpointClientFactory) AtomicInteger(java.util.concurrent.atomic.AtomicInteger) PinpointServerAcceptor(com.navercorp.pinpoint.rpc.server.PinpointServerAcceptor) TcpDataSender(com.navercorp.pinpoint.profiler.sender.TcpDataSender) PinpointClientReconnectEventListener(com.navercorp.pinpoint.rpc.client.PinpointClientReconnectEventListener) Test(org.junit.Test)

Aggregations

TcpDataSender (com.navercorp.pinpoint.profiler.sender.TcpDataSender)2 DefaultPinpointClientFactory (com.navercorp.pinpoint.rpc.client.DefaultPinpointClientFactory)2 PinpointClient (com.navercorp.pinpoint.rpc.client.PinpointClient)2 PinpointClientFactory (com.navercorp.pinpoint.rpc.client.PinpointClientFactory)2 PinpointClientReconnectEventListener (com.navercorp.pinpoint.rpc.client.PinpointClientReconnectEventListener)2 PinpointServerAcceptor (com.navercorp.pinpoint.rpc.server.PinpointServerAcceptor)2 AtomicInteger (java.util.concurrent.atomic.AtomicInteger)2 Test (org.junit.Test)2 CountDownLatch (java.util.concurrent.CountDownLatch)1 CyclicBarrier (java.util.concurrent.CyclicBarrier)1 TException (org.apache.thrift.TException)1