Search in sources :

Example 1 with UDPReceiverBean

use of com.navercorp.pinpoint.collector.receiver.thrift.UDPReceiverBean in project pinpoint by naver.

the class DataReceiverGroupTest method receiverGroupTest1.

@Test
public void receiverGroupTest1() throws Exception {
    DataReceiverGroupConfiguration mockConfig = createMockConfig(true, true);
    TestDispatchHandler dispatchHandler = new TestDispatchHandler(2, 1);
    UDPReceiverBean udpReceiverBean = createUdpReceiverBean(mockConfig, dispatchHandler);
    TCPReceiverBean tcpReceiverBean = createTcpReceiverBean(mockConfig, dispatchHandler);
    DataSender<TBase<?, ?>> udpDataSender = null;
    TcpDataSender<TBase<?, ?>> tcpDataSender = null;
    PinpointClientFactory pinpointClientFactory = null;
    try {
        udpReceiverBean.afterPropertiesSet();
        tcpReceiverBean.afterPropertiesSet();
        udpDataSender = newUdpDataSender(mockConfig);
        pinpointClientFactory = createPinpointClientFactory();
        tcpDataSender = new TcpDataSender<>(this.getClass().getName(), "127.0.0.1", mockConfig.getTcpBindPort(), pinpointClientFactory);
        udpDataSender.send(new TResult());
        tcpDataSender.send(new TResult());
        tcpDataSender.request(new TResult());
        Assert.assertTrue(tcpDataSender.isConnected());
        Assert.assertTrue(dispatchHandler.getSendLatch().await(1000, TimeUnit.MILLISECONDS));
        Assert.assertTrue(dispatchHandler.getRequestLatch().await(1000, TimeUnit.MILLISECONDS));
    } finally {
        closeDataSender(udpDataSender);
        closeDataSender(tcpDataSender);
        closeClientFactory(pinpointClientFactory);
        closeBean(udpReceiverBean);
        closeBean(tcpReceiverBean);
    }
}
Also used : PinpointClientFactory(com.navercorp.pinpoint.rpc.client.PinpointClientFactory) DefaultPinpointClientFactory(com.navercorp.pinpoint.rpc.client.DefaultPinpointClientFactory) DataReceiverGroupConfiguration(com.navercorp.pinpoint.collector.thrift.config.DataReceiverGroupConfiguration) UDPReceiverBean(com.navercorp.pinpoint.collector.receiver.thrift.UDPReceiverBean) TBase(org.apache.thrift.TBase) TCPReceiverBean(com.navercorp.pinpoint.collector.receiver.thrift.TCPReceiverBean) TResult(com.navercorp.pinpoint.thrift.dto.TResult) Test(org.junit.Test)

Example 2 with UDPReceiverBean

use of com.navercorp.pinpoint.collector.receiver.thrift.UDPReceiverBean in project pinpoint by naver.

the class DataReceiverGroupTest method receiverGroupTest3.

@Test
public void receiverGroupTest3() throws Exception {
    DataReceiverGroupConfiguration mockConfig = createMockConfig(false, true);
    TestDispatchHandler testDispatchHandler = new TestDispatchHandler(1, 1);
    UDPReceiverBean receiver = createUdpReceiverBean(mockConfig, testDispatchHandler);
    DataSender<TBase<?, ?>> udpDataSender = null;
    TcpDataSender<TBase<?, ?>> tcpDataSender = null;
    PinpointClientFactory pinpointClientFactory = null;
    try {
        receiver.afterPropertiesSet();
        udpDataSender = newUdpDataSender(mockConfig);
        udpDataSender.send(new TResult());
        Assert.assertTrue(testDispatchHandler.getSendLatch().await(1000, TimeUnit.MILLISECONDS));
        pinpointClientFactory = createPinpointClientFactory();
        tcpDataSender = new TcpDataSender<>(this.getClass().getName(), "127.0.0.1", mockConfig.getTcpBindPort(), pinpointClientFactory);
        Assert.assertFalse(tcpDataSender.isConnected());
    } finally {
        closeDataSender(udpDataSender);
        closeDataSender(tcpDataSender);
        closeClientFactory(pinpointClientFactory);
        closeBean(receiver);
    }
}
Also used : PinpointClientFactory(com.navercorp.pinpoint.rpc.client.PinpointClientFactory) DefaultPinpointClientFactory(com.navercorp.pinpoint.rpc.client.DefaultPinpointClientFactory) DataReceiverGroupConfiguration(com.navercorp.pinpoint.collector.thrift.config.DataReceiverGroupConfiguration) UDPReceiverBean(com.navercorp.pinpoint.collector.receiver.thrift.UDPReceiverBean) TBase(org.apache.thrift.TBase) TResult(com.navercorp.pinpoint.thrift.dto.TResult) Test(org.junit.Test)

Example 3 with UDPReceiverBean

use of com.navercorp.pinpoint.collector.receiver.thrift.UDPReceiverBean in project pinpoint by naver.

the class DataReceiverGroupTest method createUdpReceiverBean.

private UDPReceiverBean createUdpReceiverBean(DataReceiverGroupConfiguration mockConfig, DispatchHandler<TBase<?, ?>, TBase<?, ?>> dispatchHandler) {
    UDPReceiverBean udpReceiverBean = new UDPReceiverBean();
    udpReceiverBean.setBeanName("udpReceiver");
    udpReceiverBean.setBindIp(mockConfig.getUdpBindIp());
    udpReceiverBean.setBindPort(mockConfig.getUdpBindPort());
    udpReceiverBean.setAddressFilter(AddressFilter.ALL);
    udpReceiverBean.setDispatchHandler(dispatchHandler);
    udpReceiverBean.setExecutor(MoreExecutors.directExecutor());
    udpReceiverBean.setUdpBufferSize(mockConfig.getUdpReceiveBufferSize());
    udpReceiverBean.setEnable(true);
    return udpReceiverBean;
}
Also used : UDPReceiverBean(com.navercorp.pinpoint.collector.receiver.thrift.UDPReceiverBean)

Aggregations

UDPReceiverBean (com.navercorp.pinpoint.collector.receiver.thrift.UDPReceiverBean)3 DataReceiverGroupConfiguration (com.navercorp.pinpoint.collector.thrift.config.DataReceiverGroupConfiguration)2 DefaultPinpointClientFactory (com.navercorp.pinpoint.rpc.client.DefaultPinpointClientFactory)2 PinpointClientFactory (com.navercorp.pinpoint.rpc.client.PinpointClientFactory)2 TResult (com.navercorp.pinpoint.thrift.dto.TResult)2 TBase (org.apache.thrift.TBase)2 Test (org.junit.Test)2 TCPReceiverBean (com.navercorp.pinpoint.collector.receiver.thrift.TCPReceiverBean)1