org.testifyproject.bytebuddy.implementation.bind.annotation.This Maven / Gradle / Ivy
package org.testifyproject.bytebuddy.implementation.bind.annotation;
import org.testifyproject.bytebuddy.description.annotation.AnnotationDescription;
import org.testifyproject.bytebuddy.description.method.MethodDescription;
import org.testifyproject.bytebuddy.description.method.ParameterDescription;
import org.testifyproject.bytebuddy.implementation.Implementation;
import org.testifyproject.bytebuddy.implementation.bind.MethodDelegationBinder;
import org.testifyproject.bytebuddy.implementation.bytecode.StackManipulation;
import org.testifyproject.bytebuddy.implementation.bytecode.assign.Assigner;
import org.testifyproject.bytebuddy.implementation.bytecode.constant.NullConstant;
import org.testifyproject.bytebuddy.implementation.bytecode.member.MethodVariableAccess;
import java.lang.annotation.*;
/**
* Parameters that are annotated with this annotation will be assigned a reference to the instrumented object, if
* the instrumented method is not static. Otherwise, the method with this parameter annotation will be excluded from
* the list of possible binding candidates of the static source method.
*
* @see org.testifyproject.bytebuddy.implementation.MethodDelegation
* @see org.testifyproject.bytebuddy.implementation.bind.annotation.TargetMethodAnnotationDrivenBinder
*/
@Documented
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.PARAMETER)
public @interface This {
/**
* Determines if the annotated parameter should be bound to {@code null} when intercepting a {@code static} method.
*
* @return {@code true} if the annotated parameter should be bound to {@code null} as a fallback.
*/
boolean optional() default false;
/**
* A binder for handling the
* {@link org.testifyproject.bytebuddy.implementation.bind.annotation.This}
* annotation.
*
* @see TargetMethodAnnotationDrivenBinder
*/
enum Binder implements TargetMethodAnnotationDrivenBinder.ParameterBinder {
/**
* The singleton instance.
*/
INSTANCE;
@Override
public Class getHandledType() {
return This.class;
}
@Override
public MethodDelegationBinder.ParameterBinding bind(AnnotationDescription.Loadable annotation,
MethodDescription source,
ParameterDescription target,
Implementation.Target implementationTarget,
Assigner assigner,
Assigner.Typing typing) {
if (target.getType().isPrimitive()) {
throw new IllegalStateException(target + " uses a primitive type with a @This annotation");
} else if (target.getType().isArray()) {
throw new IllegalStateException(target + " uses an array type with a @This annotation");
} else if (source.isStatic() && !annotation.loadSilent().optional()) {
return MethodDelegationBinder.ParameterBinding.Illegal.INSTANCE;
}
return new MethodDelegationBinder.ParameterBinding.Anonymous(source.isStatic()
? NullConstant.INSTANCE
: new StackManipulation.Compound(MethodVariableAccess.loadThis(),
assigner.assign(implementationTarget.getInstrumentedType().asGenericType(), target.getType(), typing)));
}
}
}