如何在应用初始化后创建自定义Spring Data JPA Repository实例？

核心思路 #

Spring Data JPA本身通过动态代理实现Repository接口，我们可以直接利用它的JpaRepositoryFactory手动生成不在扫描范围内的Repository实例，完全不需要依赖ApplicationContext或@Autowired。

完整实现代码 #

先保留你的现有Repository结构：

public interface FoodRepository<T extends FoodEntity> extends JpaRepository<T, UUID>, FoodRepositoryCustom {
}

public interface FoodRepositoryCustom {
    List<FoodEntity> executeQuery(Map<String, String> query);
}

public class FoodRepositoryCustomImpl implements FoodRepositoryCustom {

    private final EntityManager em;

    public FoodRepositoryCustomImpl(EntityManager em) {
        this.em = em;
    }

    @Override
    public List<FoodEntity> executeQuery(Map<String, String> query) {
        // 这里写你的实际查询逻辑
        return List.of();
    }
}

再编写动态生成Repository的工具类：

import org.springframework.data.jpa.repository.support.JpaRepositoryFactory;
import org.springframework.data.repository.core.support.RepositoryFactorySupport;

import jakarta.persistence.EntityManager;

public class RepositoryGenerator {

    private final EntityManager entityManager;

    public RepositoryGenerator(EntityManager entityManager) {
        this.entityManager = entityManager;
    }

    @SuppressWarnings("unchecked")
    public <F extends FoodEntity, R extends FoodRepository<F>> R getFoodRepository(Class<R> repositoryClazz) {
        // 创建JPA Repository工厂实例
        RepositoryFactorySupport factory = new JpaRepositoryFactory(entityManager);
        
        // 指定自定义实现类，Spring Data会自动将Custom接口的方法委托给该实现
        factory.setRepositoryBaseClass(FoodRepositoryCustomImpl.class);
        
        // 生成并返回Repository代理实例
        return factory.getRepository(repositoryClazz);
    }
}

使用示例 #

// 假设你已获取EntityManager实例（可从EntityManagerFactory创建）
EntityManager em = entityManagerFactory.createEntityManager();
RepositoryGenerator generator = new RepositoryGenerator(em);

// 生成PineappleRepository实例
PineappleRepository pineappleRepo = generator.getFoodRepository(PineappleRepository.class);

// 调用自定义查询方法
List<FoodEntity> results = pineappleRepo.executeQuery(Map.of("type", "pineapple"));

关键注意事项 #

• Repository接口要求：目标Repository（比如PineappleRepository）必须继承自FoodRepository，否则工厂无法正确识别并生成代理。
• EntityManager管理：如果是手动创建的EntityManager，使用完毕后记得调用em.close()释放资源，避免数据库连接泄漏。
• 自定义实现匹配规则：Spring Data JPA默认会匹配「Custom接口名+Impl」的实现类，你的FoodRepositoryCustomImpl刚好符合这个规则，无需额外配置。

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