如何获取实现JpaRepository的自定义Repository实例？

核心问题原因 #

Spring Data JPA的Repository实例是动态代理生成的，底层实际类型是SimpleJpaRepository，但自定义查询方法定义在你自己的Repository接口（比如RepoFoo）上，直接强转成JpaRepository会丢失自定义方法的访问权限。

解决步骤 #

1. （可选但推荐）定义标记接口

创建一个空标记接口，让所有带@SpiderExtract注解的Repository都实现它，方便统一管理：

public interface SpiderMarkRepository {}

修改你的Repository，实现这个标记接口：

@SpiderExtract(tableName = "TABLE_NAME")
@Repository
public interface RepoFoo extends JpaRepository<Foo, Long>, SpiderMarkRepository {
    @Query(value = "select * from foo")
    List<Foo> getFoos();
}

2. 调整仓库实例收集逻辑

修改收集方法，不再强制转换为JpaRepository，而是保留原接口类型（或用标记接口/Object），同时优化注解获取逻辑（避免依赖接口顺序）：

// 标记接口版本
private Map<String, SpiderMarkRepository> getSpiderRepos() {
    return applicationContext.getBeansOfType(SpiderMarkRepository.class).entrySet().stream()
            .collect(Collectors.toMap(
                    this::extractTableName,
                    Map.Entry::getValue
            ));
}

// 无标记接口版本，直接用Object存储
private Map<String, Object> getSpiderReposWithoutMarker() {
    return applicationContext.getBeansWithAnnotation(SpiderExtract.class).values().stream()
            .collect(Collectors.toMap(
                    this::extractTableName,
                    Function.identity()
            ));
}

// 通用方法：从Repository实例中提取表名
private String extractTableName(Object repository) {
    for (Class<?> iface : repository.getClass().getInterfaces()) {
        SpiderExtract annotation = iface.getAnnotation(SpiderExtract.class);
        if (annotation != null) {
            return annotation.tableName();
        }
    }
    throw new IllegalArgumentException("目标Repository未标注@SpiderExtract注解");
}

3. 使用自定义Repository方法

获取实例后，通过类型判断强转为对应的Repository接口，即可调用自定义方法：

// 以标记接口版本为例
SpiderMarkRepository repo = getSpiderRepos().get("TABLE_NAME");
if (repo instanceof RepoFoo) {
    List<Foo> foos = ((RepoFoo) repo).getFoos();
    // 处理业务逻辑
}

注意事项 #

• 不要依赖getClass().getInterfaces()[0]获取注解，接口顺序可能因代理生成逻辑变化导致错误，遍历所有接口找注解更可靠。
• 如果不想定义标记接口，用Object存储实例完全可行，只是使用时需要额外的类型判断。

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