Examples with AsyncContext com.navercorp.pinpoint.bootstrap.context.AsyncContext used on opensource projects

Example 16 with AsyncContext

use of com.navercorp.pinpoint.bootstrap.context.AsyncContext in project pinpoint by naver.

the class WorkerScheduleInterceptor method doInBeforeTrace.

@Override
protected void doInBeforeTrace(SpanEventRecorder recorder, Object target, Object[] args) {
    logBeforeInterceptor0(target, args);
    if (args != null && args.length > 0) {
        if (args[0] instanceof AsyncContextAccessor) {
            AsyncContext asyncContext = recorder.recordNextAsyncContext();
            ((AsyncContextAccessor) args[0])._$PINPOINT$_setAsyncContext(asyncContext);
        }
    }
}

Example 17 with AsyncContext

use of com.navercorp.pinpoint.bootstrap.context.AsyncContext in project pinpoint by naver.

the class DirectivesInterceptor method before.

@Override
public void before(Object target, Object[] args) {
    if (isDebug) {
        logger.beforeInterceptor(target, args);
    }
    int requestContextIndex = getArgsIndexOfRequestContext(args);
    if (requestContextIndex == -1) {
        return;
    }
    if (traceContext.currentTraceObject() != null) {
        return;
    }
    RequestContextImpl requestContext = (RequestContextImpl) args[requestContextIndex];
    if (!(requestContext instanceof AsyncContextAccessor)) {
        if (isDebug) {
            logger.debug("Invalid requestContext. Need metadata accessor({}).", AsyncContextAccessor.class.getName());
        }
        return;
    }
    akka.http.scaladsl.model.HttpRequest request = requestContext.request();
    final Trace trace = createTrace(request);
    if (trace == null || !trace.canSampled()) {
        return;
    }
    final SpanEventRecorder recorder = trace.traceBlockBegin();
    recorder.recordServiceType(AkkaHttpConstants.AKKA_HTTP_SERVER_INTERNAL);
    final AsyncContext asyncContext = recorder.recordNextAsyncContext(true);
    ((AsyncContextAccessor) requestContext)._$PINPOINT$_setAsyncContext(asyncContext);
    if (isDebug) {
        logger.debug("Set closeable-AsyncContext {}", asyncContext);
    }
}

Example 18 with AsyncContext

use of com.navercorp.pinpoint.bootstrap.context.AsyncContext in project pinpoint by naver.

the class ListenerConstructorInterceptor method doInAfterTrace.

@Override
protected void doInAfterTrace(SpanEventRecorder recorder, Object target, Object[] args, Object result, Throwable throwable) {
    if (target instanceof AsyncContextAccessor) {
        // set asynchronous trace
        final AsyncContext asyncContext = recorder.recordNextAsyncContext();
        ((AsyncContextAccessor) target)._$PINPOINT$_setAsyncContext(asyncContext);
        if (isDebug) {
            logger.debug("Set AsyncContext {}", asyncContext);
        }
    }
    recorder.recordApi(LISTENER_RESULT_DESCRIPTOR);
    recorder.recordException(throwable);
}

Example 19 with AsyncContext

use of com.navercorp.pinpoint.bootstrap.context.AsyncContext in project pinpoint by naver.

the class GrpcAsyncContextSpanEventEndPointInterceptor method after.

@Override
public void after(Object target, Object[] args, Object result, Throwable throwable) {
    if (isDebug) {
        logger.afterInterceptor(target, args, result, throwable);
    }
    final AsyncContext asyncContext = getAsyncContext(target, args, result, throwable);
    if (asyncContext == null) {
        logger.debug("Not found asynchronous invocation metadata");
        return;
    }
    if (isDebug) {
        logger.debug("Asynchronous invocation. asyncContext={}", asyncContext);
    }
    final Trace trace = asyncContext.currentAsyncTraceObject();
    if (trace == null) {
        return;
    }
    if (isDebug) {
        logger.debug("Asynchronous invocation. asyncTraceId={}, trace={}", asyncContext, trace);
    }
    // leave scope.
    if (!leaveAsyncTraceScope(trace)) {
        if (logger.isWarnEnabled()) {
            logger.warn("Failed to leave scope of async trace {}.", trace);
        }
        // delete unstable trace.
        deleteAsyncTrace(trace);
        return;
    }
    try {
        final SpanEventRecorder recorder = trace.currentSpanEventRecorder();
        doInAfterTrace(recorder, target, args, result, throwable);
    } catch (Throwable th) {
        if (logger.isWarnEnabled()) {
            logger.warn("AFTER error. Caused:{}", th.getMessage(), th);
        }
    } finally {
        trace.traceBlockEnd();
        if (isAsyncTraceDestination(trace)) {
            if (isDebug) {
                logger.debug("Arrived at async trace destination. asyncTraceId={}", asyncContext);
            }
            deleteAsyncTrace(trace);
        }
        finishAsyncState(asyncContext);
    }
}

Example 20 with AsyncContext

use of com.navercorp.pinpoint.bootstrap.context.AsyncContext in project pinpoint by naver.

the class CopyAsyncContextInterceptor method after.

@Override
public void after(Object target, Object[] args, Object result, Throwable throwable) {
    if (isDebug) {
        logger.afterInterceptor(target, args);
    }
    if (ArrayUtils.getLength(args) == 2) {
        AsyncContext asyncContext = getAsyncContext(args[0]);
        if (result instanceof AsyncContextAccessor) {
            logger.info("set AsyncContext:{}", asyncContext);
            ((AsyncContextAccessor) result)._$PINPOINT$_setAsyncContext(asyncContext);
        }
    }
}