Examples with AsyncContextAccessor com.navercorp.pinpoint.bootstrap.async.AsyncContextAccessor used on opensource projects

Example 26 with AsyncContextAccessor

use of com.navercorp.pinpoint.bootstrap.async.AsyncContextAccessor in project pinpoint by naver.

the class ActiveMQMessageConsumerDispatchInterceptor method before.

@Override
public void before(Object target, Object[] args) {
    if (!validate(target, args)) {
        return;
    }
    if (isDebug) {
        logger.beforeInterceptor(target, args);
    }
    try {
        final Trace trace = createTrace(target, args);
        if (trace == null) {
            return;
        }
        if (!trace.canSampled()) {
            return;
        }
        // ------------------------------------------------------
        SpanEventRecorder recorder = trace.traceBlockBegin();
        recorder.recordServiceType(ActiveMQClientConstants.ACTIVEMQ_CLIENT_INTERNAL);
        AsyncContextAccessor accessor = ArrayArgumentUtils.getArgument(args, 0, AsyncContextAccessor.class);
        if (accessor != null) {
            AsyncContext asyncContext = recorder.recordNextAsyncContext();
            accessor._$PINPOINT$_setAsyncContext(asyncContext);
        }
    } catch (Throwable th) {
        if (logger.isWarnEnabled()) {
            logger.warn("BEFORE. Caused:{}", th.getMessage(), th);
        }
    }
}

Example 27 with AsyncContextAccessor

use of com.navercorp.pinpoint.bootstrap.async.AsyncContextAccessor in project pinpoint by naver.

the class TAsyncClientManagerCallInterceptor method injectAsyncContext.

private void injectAsyncContext(final Object asyncMethodCallObj, final SpanEventRecorder recorder) {
    final AsyncContext asyncContext = recorder.recordNextAsyncContext();
    ((AsyncContextAccessor) asyncMethodCallObj)._$PINPOINT$_setAsyncContext(asyncContext);
    if (isDebug) {
        logger.debug("Set AsyncContext {}", asyncContext);
    }
}

Example 28 with AsyncContextAccessor

use of com.navercorp.pinpoint.bootstrap.async.AsyncContextAccessor in project pinpoint by naver.

the class AsyncTaskExecutorSubmitInterceptor method doInBeforeTrace.

@Override
protected void doInBeforeTrace(SpanEventRecorder recorder, Object target, Object[] args) {
    if (validate(args)) {
        // make asynchronous trace-id
        final AsyncContext asyncContext = recorder.recordNextAsyncContext();
        ((AsyncContextAccessor) args[0])._$PINPOINT$_setAsyncContext(asyncContext);
        if (isDebug) {
            logger.debug("Set asyncContext {}", asyncContext);
        }
    }
}

Example 29 with AsyncContextAccessor

use of com.navercorp.pinpoint.bootstrap.async.AsyncContextAccessor in project pinpoint by naver.

the class ReactiveMethodInterceptor method doInAfterTrace.

@Override
public void doInAfterTrace(SpanEventRecorder recorder, Object target, Object[] args, Object result, Throwable throwable) {
    if (result instanceof AsyncContextAccessor) {
        if (AsyncContextAccessorUtils.getAsyncContext(result) == null) {
            // Avoid duplicate async context
            final AsyncContext asyncContext = recorder.recordNextAsyncContext();
            ((AsyncContextAccessor) result)._$PINPOINT$_setAsyncContext(asyncContext);
        }
    }
    if (this.keyTrace) {
        Method method = ArrayArgumentUtils.getArgument(args, 0, Method.class);
        if (method != null && StringUtils.hasLength(method.getName())) {
            recorder.recordAttribute(AnnotationKey.ARGS0, method.getName());
        }
    }
    recorder.recordApi(getMethodDescriptor());
    recorder.recordServiceType(RedissonConstants.REDISSON_REACTIVE);
    recorder.recordException(throwable);
}

Example 30 with AsyncContextAccessor

use of com.navercorp.pinpoint.bootstrap.async.AsyncContextAccessor in project pinpoint by naver.

the class ExchangeFunctionMethodInterceptor method doInAfterTrace.

@Override
protected void doInAfterTrace(SpanEventRecorder recorder, Object target, Object[] args, Object result, Throwable throwable) {
    recorder.recordApi(methodDescriptor);
    recorder.recordServiceType(SpringWebFluxConstants.SPRING_WEBFLUX);
    recorder.recordException(throwable);
    if (isAsync(args) && isAsync(result)) {
        final AsyncContext asyncContext = AsyncContextAccessorUtils.getAsyncContext(args, 0);
        if (asyncContext != null) {
            ((AsyncContextAccessor) result)._$PINPOINT$_setAsyncContext(asyncContext);
            if (isDebug) {
                logger.debug("Set closeable-AsyncContext {}", asyncContext);
            }
        }
    }
}