SpringBoot结合配置与设计模式动态选择@Repository方案咨询

1. 定义统一仓库接口 #

先让所有仓库实现同一个接口，确保都具备save方法：

public interface MyPojoRepository {
    MyPojo save(MyPojo pojo);
}

然后让具体仓库实现该接口，同时指定Bean名称（对应配置文件中的映射值）：

@Repository("mysqlRepository")
public class MysqlMyPojoRepository implements MyPojoRepository {
    @Override
    public MyPojo save(MyPojo pojo) {
        // MySQL存储逻辑实现
        return pojo;
    }
}

@Repository("redisRepository")
public class RedisMyPojoRepository implements MyPojoRepository {
    @Override
    public MyPojo save(MyPojo pojo) {
        // Redis存储逻辑实现
        return pojo;
    }
}

2. 配置文件中定义映射 #

在application.yml中添加标识与仓库Bean名称的映射（新增/移除仓库只需修改此处）：

repo-mappings:
  mysql: mysqlRepository
  redis: redisRepository

如果使用application.properties格式：

repo-mappings.mysql=mysqlRepository
repo-mappings.redis=redisRepository

3. 实现动态仓库选择器 #

创建工具类，借助Spring的ApplicationContext动态获取Bean，避免硬编码映射关系：

@Component
public class RepositorySelector {

    private final Map<String, String> repoMapping;
    private final ApplicationContext applicationContext;
    // 缓存转换后的save方法，提升性能
    private final Map<String, Function<MyPojo, MyPojo>> repoFunctionMap = new ConcurrentHashMap<>();

    // 构造注入配置映射与ApplicationContext
    public RepositorySelector(@Value("#{${repo-mappings}}") Map<String, String> repoMapping,
                              ApplicationContext applicationContext) {
        this.repoMapping = repoMapping;
        this.applicationContext = applicationContext;
        // 初始化时加载所有映射的save方法
        initRepoFunctionMap();
    }

    private void initRepoFunctionMap() {
        repoMapping.forEach((key, beanName) -> {
            MyPojoRepository repository = applicationContext.getBean(beanName, MyPojoRepository.class);
            repoFunctionMap.put(key, repository::save);
        });
    }

    // 根据标识获取对应的save方法
    public Function<MyPojo, MyPojo> getSaveFunction(String repoKey) {
        Function<MyPojo, MyPojo> function = repoFunctionMap.get(repoKey);
        if (function == null) {
            throw new IllegalArgumentException("不支持的仓库标识: " + repoKey);
        }
        return function;
    }
}

4. 业务代码中使用 #

在业务类中注入RepositorySelector，直接根据请求标识获取对应方法执行存储：

@Service
public class MyPojoService {

    private final RepositorySelector repositorySelector;

    public MyPojoService(RepositorySelector repositorySelector) {
        this.repositorySelector = repositorySelector;
    }

    public MyPojo savePojo(String repoKey, MyPojo pojo) {
        Function<MyPojo, MyPojo> saveFunction = repositorySelector.getSaveFunction(repoKey);
        return saveFunction.apply(pojo);
    }
}

核心优势 #

• 无硬编码绑定：新增仓库只需实现接口、添加@Repository注解并在配置文件加映射，完全无需修改业务代码
• 精准性：用Bean名称而非类名配置，避免Spring中同一类多实例的冲突问题
• 性能优化：初始化时缓存所有save方法，避免重复从容器中获取Bean

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