如何在ByteBuddy中用returns方法处理Future<T>及匹配子类泛型？

1. 如何匹配返回 Future 的方法？ #

要匹配返回Future<T>的方法，ByteBuddy 支持通过泛型类型描述来实现精准匹配：

• 匹配任意泛型参数的 Future（即Future<?>）：

  import net.bytebuddy.matcher.ElementMatchers;
  import java.util.concurrent.Future;
  import net.bytebuddy.description.type.GenericType;
  
  // 匹配所有返回Future<?>的方法
  ElementMatchers.returns(new GenericType<Future<?>>() {})
  

• 匹配特定泛型参数的 Future（比如Future<String>）：

  // 匹配返回Future<String>的方法
  ElementMatchers.returns(GenericType.of(Future.class, String.class))
  

2. 匹配返回 Future 且 T 是 E 子类的方法 #

要实现类似returns(Future<isSubtypeOf(E.class)>)的效果，需要结合泛型解析与子类型校验，以下是两种可行方案：

方案一：使用GenericTypeMatcher（简洁写法） #

通过声明带通配符的泛型类型，配合GenericTypeMatcher完成子类型匹配：

import net.bytebuddy.matcher.ElementMatchers;
import net.bytebuddy.matcher.GenericTypeMatcher;
import java.util.concurrent.Future;
import net.bytebuddy.description.type.GenericType;

class E {}

// 匹配返回Future<T>且T是E子类的方法
ElementMatchers.returns(new GenericTypeMatcher(new GenericType<Future<? extends E>>() {}))

方案二：自定义TypeMatcher（灵活扩展） #

如果需要更复杂的判断逻辑，可自定义TypeMatcher解析泛型参数并验证子类型：

import net.bytebuddy.description.type.TypeDescription;
import net.bytebuddy.matcher.ElementMatchers;
import net.bytebuddy.matcher.TypeMatcher;
import java.util.concurrent.Future;

class E {}

ElementMatchers.returns(new TypeMatcher() {
    @Override
    public boolean matches(TypeDescription typeDescription) {
        // 先判断是否为Future类型
        if (!typeDescription.isAssignableTo(Future.class)) {
            return false;
        }
        // 获取Future的泛型参数
        TypeDescription.Generic genericType = typeDescription.asGenericType();
        if (genericType.getTypeArguments().isEmpty()) {
            return false;
        }
        // 校验泛型参数是否为E的子类
        return genericType.getTypeArguments().get(0).isSubtypeOf(E.class);
    }
})

完整拦截示例 #

将匹配逻辑整合到ByteBuddy拦截流程中：

import net.bytebuddy.ByteBuddy;
import net.bytebuddy.implementation.MethodDelegation;
import net.bytebuddy.matcher.ElementMatchers;

class E {}
class SubE extends E {}

// 拦截目标类中返回Future<E子类>的方法
new ByteBuddy()
    .subclass(TargetClass.class)
    .method(ElementMatchers.returns(new GenericTypeMatcher(new GenericType<Future<? extends E>>() {})))
    .intercept(MethodDelegation.to(Interceptor.class))
    .make();

内容的提问来源于stack exchange，提问作者Dr Drake Ramoray