如何动态调用指定JPA Repository实例的findById方法

问题背景 #

现有多个继承JpaRepository的仓库接口，需要根据表名动态调用对应仓库的findById(id)方法。已通过枚举关联表名与实体类，但直接反射调用findById时抛出NoSuchMethodException，原因是该方法是JpaRepository的默认方法，并非自定义仓库接口的原生方法。

可行解决方案 #

方案1：利用Spring上下文获取Repository实例（推荐） #

Spring会为每个JpaRepository接口生成代理实现类，直接通过上下文获取对应实例即可调用方法，无需反射。

首先完善枚举，添加对应仓库接口的类型：

public enum DynamicEntityRepository {
    Table1(Entity1.class, Entity1Repository.class),
    Table2(Entity2.class, Entity2Repository.class),
    Table3(Entity3.class, Entity3Repository.class),
    Table4(Entity4.class, Entity4Repository.class);

    private final Class<?> entityClass;
    private final Class<? extends JpaRepository<?, ?>> repoClass;

    DynamicEntityRepository(Class<?> entityClass, Class<? extends JpaRepository<?, ?>> repoClass) {
        this.entityClass = entityClass;
        this.repoClass = repoClass;
    }

    public Class<?> getEntityClass() {
        return entityClass;
    }

    public Class<? extends JpaRepository<?, ?>> getRepoClass() {
        return repoClass;
    }
}

然后在业务类中注入ApplicationContext，获取仓库实例并调用方法：

@Service
public class DynamicQueryService {

    @Autowired
    private ApplicationContext applicationContext;

    public Optional<?> findByTableNameAndId(String tableName, Serializable id) {
        DynamicEntityRepository entityEnum = DynamicEntityRepository.valueOf(tableName);
        // 获取对应仓库的代理实例
        JpaRepository<?, ?> repository = applicationContext.getBean(entityEnum.getRepoClass());
        // 直接调用findById方法
        return repository.findById(id);
    }
}

方案2：通过EntityManager直接查询 #

绕开Repository层，利用EntityManager的find方法实现相同逻辑，适合不想依赖Repository实例的场景。可以结合Spring Data JPA提供的JpaContext获取对应实体的EntityManager：

@Service
public class DynamicQueryService {

    @Autowired
    private JpaContext jpaContext;

    public Optional<?> findByTableNameAndId(String tableName, Serializable id) {
        DynamicEntityRepository entityEnum = DynamicEntityRepository.valueOf(tableName);
        // 获取对应实体的EntityManager
        EntityManager em = jpaContext.getEntityManagerByManagedType(entityEnum.getEntityClass());
        // 执行查询
        Object result = em.find(entityEnum.getEntityClass(), id);
        return Optional.ofNullable(result);
    }
}

方案3：反射调用默认方法（不推荐） #

如果必须用反射处理默认方法，可通过MethodHandle调用接口的默认方法：

@Service
public class DynamicQueryService {

    @Autowired
    private ApplicationContext applicationContext;

    public Optional<?> findByTableNameAndId(String tableName, Serializable id) throws Throwable {
        DynamicEntityRepository entityEnum = DynamicEntityRepository.valueOf(tableName);
        JpaRepository<?, ?> repository = applicationContext.getBean(entityEnum.getRepoClass());
        
        // 获取JpaRepository接口的findById方法
        Method findByIdMethod = JpaRepository.class.getMethod("findById", Object.class);
        // 创建MethodHandle并绑定到实例
        MethodHandles.Lookup lookup = MethodHandles.lookup();
        MethodHandle methodHandle = lookup.unreflectSpecial(findByIdMethod, repository.getClass());
        
        // 调用方法并返回结果
        return (Optional<?>) methodHandle.invoke(repository, id);
    }
}

该方法实现复杂，且容易因代理类结构变化出现问题，仅作特殊场景备选。

内容的提问来源于stack exchange，提问作者NkTech