io.opentelemetry.javaagent.instrumentation.ratpack.DefaultExecutionInstrumentation Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of opentelemetry-javaagent-ratpack-1.4 Show documentation
Show all versions of opentelemetry-javaagent-ratpack-1.4 Show documentation
Instrumentation of Java libraries using OpenTelemetry.
/*
* Copyright The OpenTelemetry Authors
* SPDX-License-Identifier: Apache-2.0
*/
package io.opentelemetry.javaagent.instrumentation.ratpack;
import static net.bytebuddy.matcher.ElementMatchers.nameStartsWith;
import static net.bytebuddy.matcher.ElementMatchers.named;
import static net.bytebuddy.matcher.ElementMatchers.takesArgument;
import io.opentelemetry.javaagent.extension.instrumentation.TypeInstrumentation;
import io.opentelemetry.javaagent.extension.instrumentation.TypeTransformer;
import net.bytebuddy.asm.Advice;
import net.bytebuddy.description.type.TypeDescription;
import net.bytebuddy.matcher.ElementMatcher;
import ratpack.exec.internal.Continuation;
import ratpack.func.Action;
public class DefaultExecutionInstrumentation implements TypeInstrumentation {
@Override
public ElementMatcher typeMatcher() {
return named("ratpack.exec.internal.DefaultExecution");
}
@Override
public void transform(TypeTransformer transformer) {
transformer.applyAdviceToMethod(
nameStartsWith("delimit") // include delimitStream
.and(takesArgument(0, named("ratpack.func.Action")))
.and(takesArgument(1, named("ratpack.func.Action"))),
DefaultExecutionInstrumentation.class.getName() + "$DelimitAdvice");
}
@SuppressWarnings("unused")
public static class DelimitAdvice {
@Advice.OnMethodEnter(suppress = Throwable.class)
public static void wrap(
@Advice.Argument(value = 0, readOnly = false) Action onError,
@Advice.Argument(value = 1, readOnly = false) Action segment) {
onError = ActionWrapper.wrapIfNeeded(onError);
segment = ActionWrapper.wrapIfNeeded(segment);
}
}
}