Search in sources :

Example 16 with AcceptApplication

use of com.navercorp.pinpoint.web.service.map.AcceptApplication in project pinpoint by naver.

the class RpcCallProcessorTest method multipleAcceptApplications_with_AliasAndOriginal.

@Test
public void multipleAcceptApplications_with_AliasAndOriginal() {
    ServiceType AliasServiceType = of(1008, "TEST_ALIAS_CLIENT", ALIAS);
    ServiceType ServerServiceType = of(1009, "TEST_ALIAS_SERVER", RECORD_STATISTICS, TERMINAL);
    // Given
    ServiceType rpcClientServiceType = mock(ServiceType.class);
    when(rpcClientServiceType.isRpcClient()).thenReturn(true);
    String rpcUri = "accept.host/foo";
    Application fromApplication = new Application("WAS", ServiceType.TEST_STAND_ALONE);
    Application toApplication = new Application(rpcUri, rpcClientServiceType);
    LinkDataMap linkDataMap = new LinkDataMap();
    linkDataMap.addLinkData(new LinkData(fromApplication, toApplication));
    Application expectedToApplication1 = new Application("AliasClient", AliasServiceType);
    Application expectedToApplication2 = new Application("AliasServer", ServerServiceType);
    when(hostApplicationMapDao.findAcceptApplicationName(fromApplication, testRange)).thenReturn(Sets.newHashSet(new AcceptApplication(rpcUri, expectedToApplication1), new AcceptApplication(rpcUri, expectedToApplication2)));
    // When
    VirtualLinkMarker virtualLinkMarker = new VirtualLinkMarker();
    RpcCallProcessor rpcCallProcessor = new RpcCallProcessor(hostApplicationMapDao, virtualLinkMarker);
    LinkDataMap replacedLinkDataMap = rpcCallProcessor.processLinkDataMap(linkDataMap, testRange);
    // Then
    LinkKey originalLinkKey = new LinkKey(fromApplication, toApplication);
    Assert.assertNull(replacedLinkDataMap.getLinkData(originalLinkKey));
    LinkKey replacedLinkKey2 = new LinkKey(fromApplication, expectedToApplication2);
    LinkData replacedLinkData2 = replacedLinkDataMap.getLinkData(replacedLinkKey2);
    Assert.assertNotNull(replacedLinkData2);
    Assert.assertEquals(fromApplication, replacedLinkData2.getFromApplication());
    Assert.assertEquals(expectedToApplication2, replacedLinkData2.getToApplication());
}
Also used : LinkKey(com.navercorp.pinpoint.web.vo.LinkKey) LinkData(com.navercorp.pinpoint.web.applicationmap.rawdata.LinkData) ServiceType(com.navercorp.pinpoint.common.trace.ServiceType) AcceptApplication(com.navercorp.pinpoint.web.service.map.AcceptApplication) Application(com.navercorp.pinpoint.web.vo.Application) AcceptApplication(com.navercorp.pinpoint.web.service.map.AcceptApplication) VirtualLinkMarker(com.navercorp.pinpoint.web.service.map.VirtualLinkMarker) LinkDataMap(com.navercorp.pinpoint.web.applicationmap.rawdata.LinkDataMap) Test(org.junit.Test)

Example 17 with AcceptApplication

use of com.navercorp.pinpoint.web.service.map.AcceptApplication in project pinpoint by naver.

the class AcceptApplicationLocalCacheTest method testFind.

@Test
public void testFind() {
    AcceptApplicationLocalCache cache = new AcceptApplicationLocalCache();
    Application tomcat = new Application("Tomcat", ServiceType.STAND_ALONE);
    RpcApplication rpc = new RpcApplication("localhost:8080", tomcat);
    // find the application that accept the rpc request of calling to localhost:8080 at tomcat itself
    Set<AcceptApplication> findSet = createAcceptApplication();
    cache.put(rpc, findSet);
    // found
    Set<AcceptApplication> acceptApplications = cache.get(rpc);
    Assert.assertEquals(acceptApplications.size(), 1);
    Assert.assertEquals(acceptApplications.iterator().next(), localhost);
    // not found
    Set<AcceptApplication> unknown = cache.get(new RpcApplication("unknown:8080", tomcat));
    Assert.assertTrue(unknown.isEmpty());
    Assert.assertFalse(unknown.iterator().hasNext());
}
Also used : AcceptApplication(com.navercorp.pinpoint.web.service.map.AcceptApplication) AcceptApplication(com.navercorp.pinpoint.web.service.map.AcceptApplication) Application(com.navercorp.pinpoint.web.vo.Application) Test(org.junit.Test)

Example 18 with AcceptApplication

use of com.navercorp.pinpoint.web.service.map.AcceptApplication in project pinpoint by naver.

the class HbaseHostApplicationMapDao method findAcceptApplicationName.

@Override
public Set<AcceptApplication> findAcceptApplicationName(Application fromApplication, Range range) {
    Objects.requireNonNull(fromApplication, "fromApplication");
    final Scan scan = createScan(fromApplication, range);
    TableName hostApplicationMapTableName = tableNameProvider.getTableName(HbaseTable.HOST_APPLICATION_MAP_VER2);
    final List<List<AcceptApplication>> result = hbaseOperations2.findParallel(hostApplicationMapTableName, scan, acceptApplicationRowKeyDistributor, hostApplicationMapperVer2, HOST_APPLICATION_MAP_VER2_NUM_PARTITIONS);
    if (CollectionUtils.isNotEmpty(result)) {
        final Set<AcceptApplication> resultSet = new HashSet<>();
        for (List<AcceptApplication> resultList : result) {
            resultSet.addAll(resultList);
        }
        logger.debug("findAcceptApplicationName result:{}", resultSet);
        return resultSet;
    } else {
        return Collections.emptySet();
    }
}
Also used : TableName(org.apache.hadoop.hbase.TableName) Scan(org.apache.hadoop.hbase.client.Scan) List(java.util.List) AcceptApplication(com.navercorp.pinpoint.web.service.map.AcceptApplication) HashSet(java.util.HashSet)

Aggregations

AcceptApplication (com.navercorp.pinpoint.web.service.map.AcceptApplication)18 Application (com.navercorp.pinpoint.web.vo.Application)11 Test (org.junit.Test)6 LinkKey (com.navercorp.pinpoint.web.vo.LinkKey)5 LinkData (com.navercorp.pinpoint.web.applicationmap.rawdata.LinkData)4 RpcApplication (com.navercorp.pinpoint.web.service.map.RpcApplication)4 ServiceType (com.navercorp.pinpoint.common.trace.ServiceType)3 LinkDataMap (com.navercorp.pinpoint.web.applicationmap.rawdata.LinkDataMap)3 VirtualLinkMarker (com.navercorp.pinpoint.web.service.map.VirtualLinkMarker)3 Range (com.navercorp.pinpoint.web.vo.Range)2 HashSet (java.util.HashSet)2 Buffer (com.navercorp.pinpoint.common.buffer.Buffer)1 OffsetFixedBuffer (com.navercorp.pinpoint.common.buffer.OffsetFixedBuffer)1 ArrayList (java.util.ArrayList)1 List (java.util.List)1 Cell (org.apache.hadoop.hbase.Cell)1 TableName (org.apache.hadoop.hbase.TableName)1 Scan (org.apache.hadoop.hbase.client.Scan)1