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

The following examples show how to use net.bytebuddy.asm.Advice#Origin . You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.
Example 1
Source File: FixedDelayAgentRule.java    From java-specialagent with Apache License 2.0 6 votes vote down vote up 
@Advice.OnMethodEnter
public static void exit(final @ClassName String className, final @Advice.Origin String origin, @Advice.Argument(value = 0, readOnly = false, typing = Typing.DYNAMIC) Runnable arg) throws Exception {
  if (!isAllowed(className, origin))
    return;

  final Tracer tracer = GlobalTracer.get();
  if (isVerbose(className)) {
    final Span span = tracer
      .buildSpan("scheduleWithFixedDelay")
      .withTag(Tags.COMPONENT, "java-concurrent")
      .start();
    arg = WrapperProxy.wrap(arg, new TracedRunnable(arg, span, true));
    span.finish();
  }
  else if (tracer.activeSpan() != null) {
    arg = WrapperProxy.wrap(arg, new TracedRunnable(arg, tracer.activeSpan(), false));
  }
}
Example 2
Source File: PrivilegedEventRule.java    From java-specialagent with Apache License 2.0 6 votes vote down vote up 
@Advice.OnMethodExit(onThrowable = Throwable.class)
public static void exit(final @ClassName String className, final @Advice.Origin String origin, final @Advice.Argument(value = 0) Object event) {
  if (!isAllowed(className, origin))
    return;

  if (event == null) {
    MDC.remove(TRACE_ID_MDC_KEY);
    MDC.remove(SPAN_ID_MDC_KEY);
    return;
  }

  final String correlationId = ((Event)event).getCorrelationId();
  if (correlationId == null)
    return;

  final Span span = SpanAssociations.get().retrieveSpan(correlationId);
  if (span == null)
    return;

  MDC.put(TRACE_ID_MDC_KEY, span.context().toTraceId());
  MDC.put(SPAN_ID_MDC_KEY, span.context().toSpanId());
}
Example 3
Source File: HttpUrlConnectionInstrumentation.java    From apm-agent-java with Apache License 2.0 6 votes vote down vote up 
@Advice.OnMethodEnter(suppress = Throwable.class)
public static void enter(@Advice.This HttpURLConnection thiz,
                         @Advice.FieldValue("connected") boolean connected,
                         @Advice.Local("span") Span span,
                         @Advice.Origin String signature) {
    if (tracer == null || tracer.getActive() == null) {
        return;
    }
    span = inFlightSpans.get(thiz);
    if (span == null && !connected) {
        final URL url = thiz.getURL();
        span = HttpClientHelper.startHttpClientSpan(tracer.getActive(), thiz.getRequestMethod(), url.toString(), url.getProtocol(), url.getHost(), url.getPort());
        if (span != null) {
            if (!TraceContext.containsTraceContextTextHeaders(thiz, UrlConnectionPropertyAccessor.instance())) {
                span.propagateTraceContext(thiz, UrlConnectionPropertyAccessor.instance());
            }
        }
    }
    if (span != null) {
        span.activate();
    }
}
Example 4
Source File: ApmSpanBuilderInstrumentation.java    From apm-agent-java with Apache License 2.0 5 votes vote down vote up 
@Advice.OnMethodExit(suppress = Throwable.class)
public static void createSpan(@Advice.Argument(value = 0, typing = Assigner.Typing.DYNAMIC) @Nullable AbstractSpan<?> parentContext,
                              @Advice.Origin Class<?> spanBuilderClass,
                              @Advice.FieldValue(value = "tags") Map<String, Object> tags,
                              @Advice.FieldValue(value = "operationName") String operationName,
                              @Advice.FieldValue(value = "microseconds") long microseconds,
                              @Advice.Argument(1) @Nullable Iterable<Map.Entry<String, String>> baggage,
                              @Advice.Return(readOnly = false) Object span) {
    span = doCreateTransactionOrSpan(parentContext, tags, operationName, microseconds, baggage, spanBuilderClass.getClassLoader());
}
Example 5
Source File: OkHttpClientAsyncInstrumentation.java    From apm-agent-java with Apache License 2.0 5 votes vote down vote up 
@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 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();

    Request request = originalRequest;
    URL url = request.url();
    span = HttpClientHelper.startHttpClientSpan(parent, request.method(), url.toString(), url.getProtocol(),
        OkHttpClientHelper.computeHostName(url.getHost()), url.getPort());
    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);
    }
}
Example 6
Source File: SpringRabbitMQAgentRule.java    From java-specialagent with Apache License 2.0 4 votes vote down vote up 
@Advice.OnMethodExit(onThrowable = Throwable.class)
public static void exit(final @ClassName String className, final @Advice.Origin String origin, final @Advice.Thrown Throwable thrown) {
  if (isAllowed(className, origin))
    SpringRabbitMQAgentIntercept.handleDeliveryEnd(thrown);
}
Example 7
Source File: SpringJmsMQAgentRule.java    From java-specialagent with Apache License 2.0 4 votes vote down vote up 
@Advice.OnMethodExit(onThrowable = Throwable.class)
public static void exit(final @ClassName String className, final @Advice.Origin String origin, final @Advice.Thrown Throwable thrown) {
  if (isAllowed(className, origin))
    SpringJmsAgentIntercept.onMessageExit(thrown);
}
Example 8
Source File: AkkaAgentRule.java    From java-specialagent with Apache License 2.0 4 votes vote down vote up 
@Advice.OnMethodExit(onThrowable = Throwable.class)
public static void exit(final @ClassName String className, final @Advice.Origin String origin, final @Advice.Argument(value = 0) Object actorRef, final @Advice.Argument(value = 1) Object message, final @Advice.Thrown Throwable thrown) {
  if (isAllowed(className, origin))
    AkkaAgentIntercept.askEnd(actorRef, message, thrown, null);
}
Example 9
Source File: Neo4jDriverAgentRule.java    From java-specialagent with Apache License 2.0 4 votes vote down vote up 
@Advice.OnMethodExit
public static void enter(final @ClassName String className, final @Advice.Origin String origin, @Advice.Return(readOnly = false, typing = Typing.DYNAMIC) Object returned) {
  if (isAllowed(className, origin))
    returned = Neo4jDriverAgentIntercept.exit(returned);
}
Example 10
Source File: ThriftAgentRule.java    From java-specialagent with Apache License 2.0 4 votes vote down vote up 
@Advice.OnMethodExit
public static void exit(final @ClassName String className, final @Advice.Origin String origin, @Advice.Return(readOnly = false, typing = Typing.DYNAMIC) Object returned) {
  if (isAllowed(className, origin))
    returned = ThriftProtocolFactoryAgentIntercept.exit(returned);
}
Example 11
Source File: JdbcAgentRule.java    From java-specialagent with Apache License 2.0 4 votes vote down vote up 
@Advice.OnMethodEnter
public static void enter(final @ClassName String className, final @Advice.Origin String origin, final @Advice.Argument(value = 1) Class<?> caller) throws Exception {
  if (isAllowed(className, origin))
    JdbcAgentIntercept.isDriverAllowed(caller);
}
Example 12
Source File: AsyncHttpClientAgentRule.java    From java-specialagent with Apache License 2.0 4 votes vote down vote up 
@Advice.OnMethodEnter
public static void enter(final @ClassName String className, final @Advice.Origin String origin, final @Advice.Argument(value = 0) Object request, @Advice.Argument(value = 1, readOnly = false, typing = Typing.DYNAMIC) Object handler) {
  if (isAllowed(className, origin))
    handler = AsyncHttpClientAgentIntercept.enter(request, handler);
}
Example 13
Source File: LettuceAgentRule.java    From java-specialagent with Apache License 2.0 4 votes vote down vote up 
@Advice.OnMethodExit(onThrowable = Throwable.class)
public static void exit(final @ClassName String className, final @Advice.Origin String origin, final @Advice.Argument(value = 0) Object arg, final @Advice.Return Object returned, final @Advice.Thrown Throwable thrown) {
  if (isAllowed(className, origin))
   LettuceAgentIntercept.dispatchEnd(arg, returned, thrown);
}
Example 14
Source File: JmsAgentRule.java    From java-specialagent with Apache License 2.0 4 votes vote down vote up 
@Advice.OnMethodExit
public static void enter(final @ClassName String className, final @Advice.Origin String origin, @Advice.Return(readOnly = false, typing = Typing.DYNAMIC) Object returned) {
  if (isAllowed(className, origin) && !WrapperProxy.isWrapper(returned))
    returned = WrapperProxy.wrap(returned, JmsAgentIntercept.createConsumer(returned));
}
Example 15
Source File: SpringKafkaAgentRule.java    From java-specialagent with Apache License 2.0 4 votes vote down vote up 
@Advice.OnMethodExit(onThrowable = Throwable.class)
public static void exit(final @ClassName String className, final @Advice.Origin String origin, final @Advice.Thrown Throwable thrown) {
  if (isAllowed(className, origin))
    SpringKafkaAgentIntercept.onMessageExit(thrown);
}
Example 16
Source File: SpringRabbitMQAgentRule.java    From java-specialagent with Apache License 2.0 4 votes vote down vote up 
@Advice.OnMethodExit(onThrowable = Throwable.class)
public static void exit(final @ClassName String className, final @Advice.Origin String origin, final @Advice.Thrown Throwable thrown) {
  if (isAllowed(className, origin))
    SpringRabbitMQAgentIntercept.onMessageExit(thrown);
}
Example 17
Source File: CxfWsClientAgentRule.java    From java-specialagent with Apache License 2.0 4 votes vote down vote up 
@Advice.OnMethodEnter
public static void enter(final @ClassName String className, final @Advice.Origin String origin, final @Advice.This Object thiz) {
  if (isAllowed(className, origin))
    CxfAgentIntercept.addClientTracingFeature(thiz);
}
Example 18
Source File: JedisAgentRule.java    From java-specialagent with Apache License 2.0 4 votes vote down vote up 
@Advice.OnMethodEnter
public static void enter(final @ClassName String className, final @Advice.Origin String origin, final @Advice.Argument(value = 0) Object command, final @Advice.Argument(value = 1, readOnly = false, typing = Typing.DYNAMIC) byte[][] args) {
  if (isAllowed(className, origin))
    JedisAgentIntercept.sendCommand(command, args);
}
Example 19
Source File: FeignAgentRule.java    From java-specialagent with Apache License 2.0 4 votes vote down vote up 
@Advice.OnMethodExit(onThrowable = Exception.class)
public static void exit(final @ClassName String className, final @Advice.Origin String origin, final @Advice.Thrown Exception thrown, @Advice.Return Object response, final @Advice.Argument(value = 0) Object request, final @Advice.Argument(value = 1) Object options) {
  if (isAllowed(className, origin))
    FeignAgentIntercept.onResponse(response, request, options, thrown);
}
Example 20
Source File: SpymemcachedAgentRule.java    From java-specialagent with Apache License 2.0 4 votes vote down vote up 
@Advice.OnMethodEnter
public static void enter(final @ClassName String className, final @Advice.Origin String origin, @Advice.Argument(value = 1, typing = Typing.DYNAMIC, readOnly = false) Object callback) {
  if (isAllowed(className, origin))
    callback = SpymemcachedAgentIntercept.tracingCallback("flush", null, callback);
}