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());
}
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());
}
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();
}
}
Aggregations