Search in sources :

Example 11 with TCommandTransfer

use of com.navercorp.pinpoint.thrift.dto.command.TCommandTransfer in project pinpoint by pinpoint-apm.

the class AgentServiceImpl method openStream.

@Override
public ClientStreamChannel openStream(AgentInfo agentInfo, byte[] payload, ClientStreamChannelEventHandler streamChannelEventHandler) throws TException, StreamException {
    assertClusterEnabled();
    TCommandTransfer transferObject = createCommandTransferObject(agentInfo, payload);
    List<PinpointSocket> socketList = clusterManager.getSocket(agentInfo);
    if (CollectionUtils.nullSafeSize(socketList) == 1) {
        PinpointSocket socket = socketList.get(0);
        return socket.openStream(serializeRequest(transferObject), streamChannelEventHandler);
    } else if (CollectionUtils.isEmpty(socketList)) {
        throw new StreamException(StreamCode.CONNECTION_NOT_FOUND);
    } else {
        throw new StreamException(StreamCode.CONNECTION_DUPLICATED);
    }
}
Also used : TCommandTransfer(com.navercorp.pinpoint.thrift.dto.command.TCommandTransfer) PinpointSocket(com.navercorp.pinpoint.rpc.PinpointSocket) StreamException(com.navercorp.pinpoint.rpc.stream.StreamException)

Example 12 with TCommandTransfer

use of com.navercorp.pinpoint.thrift.dto.command.TCommandTransfer in project pinpoint by pinpoint-apm.

the class AgentServiceImpl method invoke.

@Override
public PinpointRouteResponse invoke(AgentInfo agentInfo, byte[] payload, long timeout) throws TException {
    TCommandTransfer transferObject = createCommandTransferObject(agentInfo, payload);
    List<PinpointSocket> socketList = clusterManager.getSocket(agentInfo);
    Future<ResponseMessage> future = null;
    if (CollectionUtils.nullSafeSize(socketList) == 1) {
        PinpointSocket socket = socketList.get(0);
        future = socket.request(serializeRequest(transferObject));
    }
    PinpointRouteResponse response = getResponse(future, timeout);
    return response;
}
Also used : TCommandTransfer(com.navercorp.pinpoint.thrift.dto.command.TCommandTransfer) PinpointSocket(com.navercorp.pinpoint.rpc.PinpointSocket) PinpointRouteResponse(com.navercorp.pinpoint.web.cluster.PinpointRouteResponse) FailedPinpointRouteResponse(com.navercorp.pinpoint.web.cluster.FailedPinpointRouteResponse) DefaultPinpointRouteResponse(com.navercorp.pinpoint.web.cluster.DefaultPinpointRouteResponse) ResponseMessage(com.navercorp.pinpoint.rpc.ResponseMessage)

Example 13 with TCommandTransfer

use of com.navercorp.pinpoint.thrift.dto.command.TCommandTransfer in project pinpoint by pinpoint-apm.

the class AgentServiceImpl method invoke.

@Override
public Map<AgentInfo, PinpointRouteResponse> invoke(List<AgentInfo> agentInfoList, byte[] payload, long timeout) throws TException {
    Map<AgentInfo, Future<ResponseMessage>> futureMap = new HashMap<>();
    for (AgentInfo agentInfo : agentInfoList) {
        TCommandTransfer transferObject = createCommandTransferObject(agentInfo, payload);
        List<PinpointSocket> socketList = clusterManager.getSocket(agentInfo);
        if (CollectionUtils.nullSafeSize(socketList) == 1) {
            PinpointSocket socket = socketList.get(0);
            Future<ResponseMessage> future = socket.request(serializeRequest(transferObject));
            futureMap.put(agentInfo, future);
        } else {
            futureMap.put(agentInfo, null);
        }
    }
    long startTime = System.currentTimeMillis();
    Map<AgentInfo, PinpointRouteResponse> result = new HashMap<>();
    for (Map.Entry<AgentInfo, Future<ResponseMessage>> futureEntry : futureMap.entrySet()) {
        AgentInfo agentInfo = futureEntry.getKey();
        Future<ResponseMessage> future = futureEntry.getValue();
        PinpointRouteResponse response = getResponse(future, getTimeoutMillis(startTime, timeout));
        result.put(agentInfo, response);
    }
    return result;
}
Also used : TCommandTransfer(com.navercorp.pinpoint.thrift.dto.command.TCommandTransfer) HashMap(java.util.HashMap) ResponseMessage(com.navercorp.pinpoint.rpc.ResponseMessage) PinpointSocket(com.navercorp.pinpoint.rpc.PinpointSocket) Future(com.navercorp.pinpoint.rpc.Future) AgentInfo(com.navercorp.pinpoint.web.vo.AgentInfo) PinpointRouteResponse(com.navercorp.pinpoint.web.cluster.PinpointRouteResponse) FailedPinpointRouteResponse(com.navercorp.pinpoint.web.cluster.FailedPinpointRouteResponse) DefaultPinpointRouteResponse(com.navercorp.pinpoint.web.cluster.DefaultPinpointRouteResponse) HashMap(java.util.HashMap) Map(java.util.Map)

Example 14 with TCommandTransfer

use of com.navercorp.pinpoint.thrift.dto.command.TCommandTransfer in project pinpoint by naver.

the class ClusterPointRouter method handleStreamRouteCreate.

private StreamCode handleStreamRouteCreate(TCommandTransfer request, StreamCreatePacket packet, ServerStreamChannelContext streamChannelContext) {
    byte[] payload = ((TCommandTransfer) request).getPayload();
    TBase<?, ?> command = deserialize(payload);
    if (command == null) {
        return StreamCode.TYPE_UNKNOWN;
    }
    TCommandTransferResponse response = streamRouteHandler.onRoute(new StreamEvent((TCommandTransfer) request, streamChannelContext, command));
    TRouteResult routeResult = response.getRouteResult();
    if (routeResult != TRouteResult.OK) {
        logger.warn("handleStreamRouteCreate failed. command:{}, routeResult:{}", command, routeResult);
        return convertToStreamCode(routeResult);
    }
    return StreamCode.OK;
}
Also used : TCommandTransfer(com.navercorp.pinpoint.thrift.dto.command.TCommandTransfer) TCommandTransferResponse(com.navercorp.pinpoint.thrift.dto.command.TCommandTransferResponse) StreamEvent(com.navercorp.pinpoint.collector.cluster.route.StreamEvent) TRouteResult(com.navercorp.pinpoint.thrift.dto.command.TRouteResult)

Example 15 with TCommandTransfer

use of com.navercorp.pinpoint.thrift.dto.command.TCommandTransfer in project pinpoint by naver.

the class AgentEventHandlerTest method handler_should_handle_serialization_of_request_events.

@Test
@SuppressWarnings({ "rawtypes", "unchecked" })
public void handler_should_handle_serialization_of_request_events() throws Exception {
    // given
    final AgentEventType expectedEventType = AgentEventType.USER_THREAD_DUMP;
    final TCommandThreadDumpResponse expectedThreadDumpResponse = new TCommandThreadDumpResponse();
    final byte[] expectedThreadDumpResponseBody = new byte[0];
    final TCommandTransfer tCommandTransfer = new TCommandTransfer();
    tCommandTransfer.setAgentId(TEST_AGENT_ID);
    tCommandTransfer.setStartTime(TEST_START_TIMESTAMP);
    final TCommandTransferResponse tCommandTransferResponse = new TCommandTransferResponse();
    tCommandTransferResponse.setRouteResult(TRouteResult.OK);
    tCommandTransferResponse.setPayload(expectedThreadDumpResponseBody);
    final ResponseEvent responseEvent = new ResponseEvent(tCommandTransfer, null, 0, tCommandTransferResponse);
    ArgumentCaptor<AgentEventBo> argCaptor = ArgumentCaptor.forClass(AgentEventBo.class);
    HeaderTBaseDeserializer deserializer = mock(HeaderTBaseDeserializer.class);
    when(this.deserializerFactory.createDeserializer()).thenReturn(deserializer);
    when(deserializer.deserialize(expectedThreadDumpResponseBody)).thenReturn((TBase) expectedThreadDumpResponse);
    // when
    this.agentEventHandler.handleResponseEvent(responseEvent, TEST_EVENT_TIMESTAMP);
    // then
    verify(this.agentEventDao, atLeast(1)).insert(argCaptor.capture());
    AgentEventBo actualAgentEventBo = argCaptor.getValue();
    assertEquals(TEST_AGENT_ID, actualAgentEventBo.getAgentId());
    assertEquals(TEST_START_TIMESTAMP, actualAgentEventBo.getStartTimestamp());
    assertEquals(TEST_EVENT_TIMESTAMP, actualAgentEventBo.getEventTimestamp());
    assertEquals(expectedEventType, actualAgentEventBo.getEventType());
    assertArrayEquals(expectedThreadDumpResponseBody, actualAgentEventBo.getEventBody());
}
Also used : TCommandThreadDumpResponse(com.navercorp.pinpoint.thrift.dto.command.TCommandThreadDumpResponse) AgentEventType(com.navercorp.pinpoint.common.server.util.AgentEventType) TCommandTransfer(com.navercorp.pinpoint.thrift.dto.command.TCommandTransfer) TCommandTransferResponse(com.navercorp.pinpoint.thrift.dto.command.TCommandTransferResponse) AgentEventBo(com.navercorp.pinpoint.common.server.bo.AgentEventBo) ResponseEvent(com.navercorp.pinpoint.collector.cluster.route.ResponseEvent) HeaderTBaseDeserializer(com.navercorp.pinpoint.thrift.io.HeaderTBaseDeserializer) Test(org.junit.Test)

Aggregations

TCommandTransfer (com.navercorp.pinpoint.thrift.dto.command.TCommandTransfer)23 PinpointSocket (com.navercorp.pinpoint.rpc.PinpointSocket)9 TCommandTransferResponse (com.navercorp.pinpoint.thrift.dto.command.TCommandTransferResponse)9 Test (org.junit.Test)7 ResponseEvent (com.navercorp.pinpoint.collector.cluster.route.ResponseEvent)6 AgentEventBo (com.navercorp.pinpoint.common.server.bo.event.AgentEventBo)6 HeaderTBaseDeserializer (com.navercorp.pinpoint.thrift.io.HeaderTBaseDeserializer)6 AgentEventType (com.navercorp.pinpoint.common.server.util.AgentEventType)5 HeaderV1 (com.navercorp.pinpoint.io.header.v1.HeaderV1)4 DefaultMessage (com.navercorp.pinpoint.io.request.DefaultMessage)4 ResponseMessage (com.navercorp.pinpoint.rpc.ResponseMessage)4 StreamException (com.navercorp.pinpoint.rpc.stream.StreamException)4 DefaultPinpointRouteResponse (com.navercorp.pinpoint.web.cluster.DefaultPinpointRouteResponse)4 FailedPinpointRouteResponse (com.navercorp.pinpoint.web.cluster.FailedPinpointRouteResponse)4 PinpointRouteResponse (com.navercorp.pinpoint.web.cluster.PinpointRouteResponse)4 TBase (org.apache.thrift.TBase)4 TCommandEcho (com.navercorp.pinpoint.thrift.dto.command.TCommandEcho)3 TCommandThreadDumpResponse (com.navercorp.pinpoint.thrift.dto.command.TCommandThreadDumpResponse)3 AgentEventBo (com.navercorp.pinpoint.common.server.bo.AgentEventBo)2 Future (com.navercorp.pinpoint.rpc.Future)2