net.bytebuddy.asm.Advice#Local

Source File: ApacheMonitorFilterAdvice.java From apm-agent-java with Apache License 2.0

6 votes

@Advice.OnMethodExit(suppress = Throwable.class, onThrowable = Throwable.class)
public static void onExitFilterInvoke(@Advice.Argument(1) Invocation invocation,
                                      @Advice.Return Result result,
                                      @Nullable @Advice.Local("span") final Span span,
                                      @Advice.Thrown Throwable t,
                                      @Nullable @Advice.Local("transaction") Transaction transaction) {

    RpcContext context = RpcContext.getContext();
    AbstractSpan<?> actualSpan = context.isConsumerSide() ? span : transaction;
    if (actualSpan == null) {
        return;
    }

    actualSpan.deactivate();
    if (result instanceof AsyncRpcResult) {
        AsyncCallbackCreator callbackCreator = asyncCallbackCreatorClassManager.getForClassLoaderOfClass(Result.class);
        if (callbackCreator == null) {
            actualSpan.end();
            return;
        }
        context.set(DubboTraceHelper.SPAN_KEY, actualSpan);
        result.whenCompleteWithContext(callbackCreator.create(actualSpan));
    } else {
        actualSpan.end();
    }
}

Source File: OkHttp3ClientInstrumentation.java From apm-agent-java with Apache License 2.0

6 votes

@Advice.OnMethodExit(suppress = Throwable.class, onThrowable = Throwable.class)
public static void onAfterExecute(@Advice.Return @Nullable okhttp3.Response response,
                                  @Advice.Local("span") @Nullable Span span,
                                  @Advice.Thrown @Nullable Throwable t) {
    if (span != null) {
        try {
            if (response != null) {
                int statusCode = response.code();
                span.getContext().getHttp().withStatusCode(statusCode);
            }
            span.captureException(t);
        } finally {
            span.deactivate().end();
        }
    }
}

Source File: ScheduledTransactionNameInstrumentation.java From apm-agent-java with Apache License 2.0

6 votes

@Advice.OnMethodEnter(suppress = Throwable.class)
private static void setTransactionName(@SimpleMethodSignature String signature, @Advice.Origin Class<?> clazz, @Advice.Local("transaction") Transaction transaction) {
    if (tracer != null) {
        AbstractSpan<?> active = tracer.getActive();
        if (active == null) {
            transaction = tracer.startRootTransaction(clazz.getClassLoader());
            if (transaction != null) {
                transaction.withName(signature)
                    .withType("scheduled")
                    .activate();
            }

        } else {
            logger.debug("Not creating transaction for method {} because there is already a transaction running ({})", signature, active);
        }
    }
}

Source File: SpringRestTemplateInstrumentation.java From apm-agent-java with Apache License 2.0

6 votes

@Advice.OnMethodExit(suppress = Throwable.class, onThrowable = Throwable.class)
private static void afterExecute(@Advice.Return @Nullable ClientHttpResponse clientHttpResponse,
                                 @Advice.Local("span") @Nullable Span span,
                                 @Advice.Thrown @Nullable Throwable t) throws IOException {
    if (span != null) {
        try {
            if (clientHttpResponse != null) {
                int statusCode = clientHttpResponse.getRawStatusCode();
                span.getContext().getHttp().withStatusCode(statusCode);
            }
            span.captureException(t);
        } finally {
            span.deactivate().end();
        }
    }
}

Source File: CaptureSpanInstrumentation.java From apm-agent-java with Apache License 2.0

6 votes

@Advice.OnMethodEnter(suppress = Throwable.class)
public static void onMethodEnter(
    @SimpleMethodSignatureOffsetMappingFactory.SimpleMethodSignature String signature,
    @AnnotationValueOffsetMappingFactory.AnnotationValueExtractor(annotationClassName = "co.elastic.apm.api.CaptureSpan", method = "value") String spanName,
    @AnnotationValueOffsetMappingFactory.AnnotationValueExtractor(annotationClassName = "co.elastic.apm.api.CaptureSpan", method = "type") String type,
    @Nullable @AnnotationValueOffsetMappingFactory.AnnotationValueExtractor(annotationClassName = "co.elastic.apm.api.CaptureSpan", method = "subtype") String subtype,
    @Nullable @AnnotationValueOffsetMappingFactory.AnnotationValueExtractor(annotationClassName = "co.elastic.apm.api.CaptureSpan", method = "action") String action,
    @Advice.Local("span") Span span) {
    if (tracer != null) {
        final AbstractSpan<?> parent = tracer.getActive();
        if (parent != null) {
            span = parent.createSpan();
            span.setType(type, subtype, action);
            span.withName(spanName.isEmpty() ? signature : spanName)
                .activate();
        } else {
            logger.debug("Not creating span for {} because there is no currently active span.", signature);
        }
    }

}

Source File: ServerCallListenerInstrumentation.java From apm-agent-java with Apache License 2.0

6 votes

@Advice.OnMethodExit(suppress = Throwable.class, onThrowable = Throwable.class)
private static void onExit(@Advice.Thrown @Nullable Throwable thrown,
                           @Advice.This ServerCall.Listener<?> listener,
                           @Advice.Local("transaction") @Nullable Transaction transaction) {

    if (null == tracer || grpcHelperManager == null || transaction == null) {
        return;
    }

    GrpcHelper helper = grpcHelperManager.getForClassLoaderOfClass(ServerCall.Listener.class);
    if (helper == null) {
        return;
    }

    helper.exitServerListenerMethod(thrown, listener, transaction, false);
}

Source File: ServerCallListenerInstrumentation.java From apm-agent-java with Apache License 2.0

5 votes

@Advice.OnMethodEnter(suppress = Throwable.class)
private static void onEnter(@Advice.This ServerCall.Listener<?> listener,
                            @Advice.Local("transaction") Transaction transaction) {

    if (null == tracer || grpcHelperManager == null) {
        return;
    }

    GrpcHelper helper = grpcHelperManager.getForClassLoaderOfClass(ServerCall.Listener.class);
    if (helper == null) {
        return;
    }

    transaction = helper.enterServerListenerMethod(listener);
}

Source File: Lettuce34StartSpanInstrumentation.java From apm-agent-java with Apache License 2.0

5 votes

@Advice.OnMethodEnter(suppress = Throwable.class)
private static void beforeDispatch(@Nullable @Advice.Argument(0) RedisCommand command, @Advice.Local("span") Span span) throws Exception {
    if (command != null) {
        span = RedisSpanUtils.createRedisSpan(command.getType().toString());
        if (span != null) {
            commandToSpan.put(command, span);
        }
    }
}

Source File: Lettuce5StartSpanInstrumentation.java From apm-agent-java with Apache License 2.0

5 votes

@Advice.OnMethodEnter(suppress = Throwable.class)
private static void beforeDispatch(@Nullable @Advice.Argument(0) RedisCommand<?, ?, ?> command, @Advice.Local("span") Span span) throws Exception {
    if (command != null) {
        span = RedisSpanUtils.createRedisSpan(command.getType().name());
        if (span != null) {
            commandToSpan.put(command, span);
        }
    }
}

Source File: ElasticsearchClientSyncInstrumentation.java From apm-agent-java with Apache License 2.0

5 votes

@Advice.OnMethodEnter(suppress = Throwable.class)
private static void onBeforeExecute(@Advice.Argument(0) Request request,
                                    @Advice.Local("span") Span span,
                                    @Advice.Local("helper") ElasticsearchRestClientInstrumentationHelper<HttpEntity, Response, ResponseListener> helper) {

    helper = esClientInstrHelperManager.getForClassLoaderOfClass(Request.class);
    if (helper != null) {
        span = helper.createClientSpan(request.getMethod(), request.getEndpoint(), request.getEntity());
    }
}

Source File: ChannelInstrumentation.java From apm-agent-java with Apache License 2.0

5 votes

@Advice.OnMethodEnter(suppress = Throwable.class)
private static void onEnter(@Advice.This Channel channel,
                            @Advice.Argument(0) MethodDescriptor<?, ?> method,
                            @Advice.Local("span") Span span) {

    if (tracer == null || grpcHelperManager == null) {
        return;
    }

    GrpcHelper helper = grpcHelperManager.getForClassLoaderOfClass(ClientCall.class);
    if (helper != null) {
        span = helper.startSpan(tracer.getActive(), method, channel.authority());
    }

}

Source File: LegacyApacheHttpClientInstrumentation.java From apm-agent-java with Apache License 2.0

5 votes

@Advice.OnMethodEnter(suppress = Throwable.class)
private static void onBeforeExecute(@Advice.Argument(0) HttpHost host,
                                    @Advice.Argument(1) HttpRequest request,
                                    @Advice.Local("span") Span span) {
    if (tracer == null || tracer.getActive() == null) {
        return;
    }
    final AbstractSpan<?> parent = tracer.getActive();
    String method;
    if (request instanceof HttpUriRequest) {
        HttpUriRequest uriRequest = (HttpUriRequest) request;
        span = HttpClientHelper.startHttpClientSpan(parent, uriRequest.getMethod(), uriRequest.getURI(), host.getHostName());
        TextHeaderSetter<HttpRequest> headerSetter = headerSetterHelperClassManager.getForClassLoaderOfClass(HttpRequest.class);
        TextHeaderGetter<HttpRequest> headerGetter = headerGetterHelperClassManager.getForClassLoaderOfClass(HttpRequest.class);
        if (span != null) {
            span.activate();
            if (headerSetter != null) {
                span.propagateTraceContext(request, headerSetter);
            }
        } else if (headerGetter != null && !TraceContext.containsTraceContextTextHeaders(request, headerGetter)
            && headerSetter != null && parent != null) {
            // re-adds the header on redirects
            parent.propagateTraceContext(request, headerSetter);
        }

    }
}

Source File: AlibabaMonitorFilterAdvice.java From apm-agent-java with Apache License 2.0

5 votes

@Advice.OnMethodEnter(suppress = Throwable.class)
public static void onEnterFilterInvoke(@Advice.Argument(1) Invocation invocation,
                                       @Advice.Local("span") Span span,
                                       @Advice.Local("apiClazz") Class<?> apiClazz,
                                       @Advice.Local("transaction") Transaction transaction) {
    RpcContext context = RpcContext.getContext();
    AlibabaDubboAttachmentHelper helper = helperManager.getForClassLoaderOfClass(Invocation.class);
    if (helper == null || tracer == null) {
        return;
    }
    // for consumer side, just create span, more information will be collected in provider side
    AbstractSpan<?> active = tracer.getActive();
    if (context.isConsumerSide() && active != null) {
        span = DubboTraceHelper.createConsumerSpan(tracer, invocation.getInvoker().getInterface(),
            invocation.getMethodName(), context.getRemoteAddress());
        if (span != null) {
            span.propagateTraceContext(invocation, helper);
        }
    } else if (active == null) {
        // for provider side
        transaction = tracer.startChildTransaction(invocation, helper, Invocation.class.getClassLoader());
        if (transaction != null) {
            transaction.activate();
            DubboTraceHelper.fillTransaction(transaction, invocation.getInvoker().getInterface(), invocation.getMethodName());
        }
    }

}

Source File: OkHttp3ClientAsyncInstrumentation.java From apm-agent-java with Apache License 2.0

5 votes

@Advice.OnMethodEnter(suppress = Throwable.class)
private static void onBeforeEnqueue(@Advice.Origin Class<? extends Call> clazz,
                                    @Advice.FieldValue(value = "originalRequest", typing = Assigner.Typing.DYNAMIC, readOnly = false) @Nullable okhttp3.Request originalRequest,
                                    @Advice.Argument(value = 0, readOnly = false) @Nullable Callback callback,
                                    @Advice.Local("span") Span span) {
    if (tracer == null || tracer.getActive() == null || callbackWrapperCreator == null) {
        return;
    }

    final WrapperCreator<Callback> wrapperCreator = callbackWrapperCreator.getForClassLoaderOfClass(clazz);
    if (originalRequest == null || callback == null || wrapperCreator == null) {
        return;
    }

    final AbstractSpan<?> parent = tracer.getActive();

    okhttp3.Request request = originalRequest;
    HttpUrl url = request.url();
    span = HttpClientHelper.startHttpClientSpan(parent, request.method(), url.toString(), url.scheme(),
        OkHttpClientHelper.computeHostName(url.host()), url.port());
    if (span != null) {
        span.activate();
        if (headerSetterHelperManager != null) {
            TextHeaderSetter<Request.Builder> headerSetter = headerSetterHelperManager.getForClassLoaderOfClass(Request.class);
            if (headerSetter != null) {
                Request.Builder builder = originalRequest.newBuilder();
                span.propagateTraceContext(builder, headerSetter);
                originalRequest = builder.build();
            }
        }
        callback = wrapperCreator.wrap(callback, span);
    }
}

Source File: ApacheHttpAsyncClientInstrumentation.java From apm-agent-java with Apache License 2.0

5 votes

@Advice.OnMethodExit(suppress = Throwable.class, onThrowable = Throwable.class)
public static void onAfterExecute(@Advice.Local("span") @Nullable Span span,
                                  @Advice.Local("wrapped") boolean wrapped,
                                  @Advice.Thrown @Nullable Throwable t) {
    if (span != null) {
        // Deactivate in this thread. Span will be ended and reported by the listener
        span.deactivate();

        if (!wrapped) {
            // Listener is not wrapped- we need to end the span so to avoid leak and report error if occurred during method invocation
            span.captureException(t);
            span.end();
        }
    }
}

Source File: ServerCallListenerInstrumentation.java From apm-agent-java with Apache License 2.0

5 votes

@Advice.OnMethodEnter(suppress = Throwable.class)
private static void onEnter(@Advice.This ServerCall.Listener<?> listener,
                            @Advice.Local("transaction") Transaction transaction) {

    if (null == tracer || grpcHelperManager == null) {
        return;
    }

    GrpcHelper helper = grpcHelperManager.getForClassLoaderOfClass(ServerCall.Listener.class);
    if (helper == null) {
        return;
    }

    transaction = helper.enterServerListenerMethod(listener);
}

Source File: JobTransactionNameAdvice.java From apm-agent-java with Apache License 2.0

5 votes

@Advice.OnMethodEnter(suppress = Throwable.class)
private static void setTransactionName(@Advice.Argument(value = 0) @Nullable JobExecutionContext context,
                                       @SimpleMethodSignature String signature, @Advice.Origin Class<?> clazz, @Advice.Local("transaction") Transaction transaction) {
    if (ElasticApmInstrumentation.tracer != null) {
        AbstractSpan<?> active = ElasticApmInstrumentation.tracer.getActive();
        if (context == null) {
            logger.warn("Cannot correctly name transaction for method {} because JobExecutionContext is null", signature);
            transaction = ElasticApmInstrumentation.tracer.startRootTransaction(clazz.getClassLoader());
            if (transaction != null) {
                transaction.withName(signature)
                    .withType(JobTransactionNameInstrumentation.TRANSACTION_TYPE)
                    .activate();
            }
        } else if (active == null) {
            transaction = ElasticApmInstrumentation.tracer.startRootTransaction(clazz.getClassLoader());
            if (transaction != null) {
                transaction.withName(context.getJobDetail().getKey().toString())
                    .withType(JobTransactionNameInstrumentation.TRANSACTION_TYPE)
                    .activate();
            }
        } else {
            logger.debug("Not creating transaction for method {} because there is already a transaction running ({})", signature, active);
        }
        if (transaction != null) {
            transaction.setFrameworkName("Quartz");
            transaction.setFrameworkVersion(VersionUtils.getVersion(JobExecutionContext.class, "org.quartz-scheduler", "quartz"));
        }
    }
}

Source File: AbstractAsyncHttpClientInstrumentation.java From apm-agent-java with Apache License 2.0

5 votes

@Advice.OnMethodExit(suppress = Throwable.class, onThrowable = Throwable.class)
private static void onAfterExecute(@Advice.Local("span") @Nullable Span span,
                                   @Advice.Argument(value = 1) AsyncHandler<?> asyncHandler,
                                   @Advice.Thrown @Nullable Throwable t) {
    if (span != null) {
        span.deactivate();
        if (t != null) {
            handlerSpanMap.remove(asyncHandler);
            span.captureException(t).end();
        }
    }
}

Source File: ClientCallImplInstrumentation.java From apm-agent-java with Apache License 2.0

5 votes

@Advice.OnMethodExit(suppress = Throwable.class, onThrowable = Throwable.class)
private static void onExit(@Advice.Argument(0) ClientCall.Listener<?> listener,
                           @Advice.Thrown @Nullable Throwable thrown,
                           @Advice.Local("span") @Nullable Span span) {

    if (span == null) {
        return;
    }
    GrpcHelper helper = grpcHelperManager.getForClassLoaderOfClass(ClientCall.class);
    if (helper != null) {
        helper.clientCallStartExit(listener, thrown);
    }
}

Source File: KafkaProducerHeadersInstrumentation.java From apm-agent-java with Apache License 2.0

4 votes

@SuppressWarnings({"unused", "DuplicatedCode", "ParameterCanBeLocal"})
@Advice.OnMethodEnter(suppress = Throwable.class)
@Nullable
public static Span beforeSend(@Advice.FieldValue("apiVersions") final ApiVersions apiVersions,
                              @Advice.Argument(0) final ProducerRecord record,
                              @Advice.Local("helper") @Nullable KafkaInstrumentationHelper<Callback, ProducerRecord, KafkaProducer> helper,
                              @Nullable @Advice.Argument(value = 1, readOnly = false) Callback callback) {
    if (tracer == null) {
        return null;
    }
    Span span = null;

    //noinspection ConstantConditions
    helper = kafkaInstrHelperManager.getForClassLoaderOfClass(KafkaProducer.class);

    if (helper != null) {
        span = helper.onSendStart(record);
    }
    if (span == null) {
        return null;
    }

    // Avoid adding headers to records sent to a version older than 0.11.0 - see specifications in
    // https://kafka.apache.org/0110/documentation.html#messageformat
    if (apiVersions.maxUsableProduceMagic() >= RecordBatch.MAGIC_VALUE_V2 && headersSupported) {
        try {
            //noinspection ConstantConditions
            KafkaInstrumentationHeadersHelper<ConsumerRecord, ProducerRecord> kafkaInstrumentationHelper =
                kafkaInstrHeadersHelperManager.getForClassLoaderOfClass(KafkaProducer.class);
            if (kafkaInstrumentationHelper != null) {
                kafkaInstrumentationHelper.setOutgoingTraceContextHeaders(span, record);
            }
        } catch (final IllegalStateException e) {
            // headers are in a read-only state
            logger.debug("Failed to add header to Kafka record {}, probably to headers' read-only state.", record);
        }
    }

    //noinspection UnusedAssignment
    callback = helper.wrapCallback(callback, span);
    return span;
}

Java Code Examples for net.bytebuddy.asm.Advice#Local