Java Spring中双接口对应不同数据库的DAO Bean配置方案咨询

你的现有方案（调用Dao::setDatabase切换数据库）存在线程安全隐患——因为Spring默认Bean是单例的，多线程请求同时调用时会互相干扰，导致数据库操作混乱。下面是更合理的实现方式，完全不用修改Service层，还能保证线程安全：

1. 定义通用DAO接口 #

先抽离出DAO层的通用业务方法，让不同数据库的DAO实现同一接口：

public interface RemoveDao {
    void removeExpiredData(); // 你的核心业务方法
}

2. 实现两个DAO实例（对应不同数据库） #

分别创建两个DAO实现类，各自绑定对应的数据源，用@Repository+@Qualifier标识区分：

@Repository("oldDbDao")
public class OldDbRemoveDao implements RemoveDao {
    // 注入对应旧数据库的DataSource/JdbcTemplate/MybatisMapper等
    @Autowired
    @Qualifier("oldDataSource")
    private DataSource dataSource;

    @Override
    public void removeExpiredData() {
        // 旧数据库的删除逻辑
    }
}

@Repository("newDbDao")
public class NewDbRemoveDao implements RemoveDao {
    @Autowired
    @Qualifier("newDataSource")
    private DataSource dataSource;

    @Override
    public void removeExpiredData() {
        // 新数据库的删除逻辑
    }
}

（注：需要提前在配置类中定义两个不同的DataSource Bean，用@Qualifier区分）

3. 通用Service层（无需修改） #

Service层只依赖RemoveDao接口，不绑定具体实现：

@Service
public class RemoveService {
    private final RemoveDao removeDao;

    // 通过构造方法注入，这里可以用@Qualifier指定默认实现，或者留空让上层选择
    public RemoveService(RemoveDao removeDao) {
        this.removeDao = removeDao;
    }

    public void executeRemove() {
        // 你的业务逻辑，和原来完全一样
        removeDao.removeExpiredData();
    }
}

4. 配置两个Service Bean，分别注入不同DAO #

在配置类中创建两个RemoveService实例，分别绑定对应的DAO：

@Configuration
public class ServiceConfig {
    @Bean("oldDbService")
    public RemoveService oldDbService(@Qualifier("oldDbDao") RemoveDao dao) {
        return new RemoveService(dao);
    }

    @Bean("newDbService")
    public RemoveService newDbService(@Qualifier("newDbDao") RemoveDao dao) {
        return new RemoveService(dao);
    }
}

5. 控制器层注入两个Service，对应不同端点 #

控制器中注入两个Service实例，分别处理两个端点：

@RestController
public class RemoveController {
    @Autowired
    @Qualifier("oldDbService")
    private RemoveService oldDbService;

    @Autowired
    @Qualifier("newDbService")
    private RemoveService newDbService;

    @PostMapping("/remove_old")
    public ResponseEntity<Void> removeOld() {
        oldDbService.executeRemove();
        return ResponseEntity.ok().build();
    }

    @PostMapping("/remove_new")
    public ResponseEntity<Void> removeNew() {
        newDbService.executeRemove();
        return ResponseEntity.ok().build();
    }
}

另一种更轻量的方式（动态选择DAO） #

如果不想创建多个Service Bean，也可以在Service层注入所有DAO实现，通过请求标识动态选择：

@Service
public class RemoveService {
    private final Map<String, RemoveDao> daoMap;

    // Spring会自动把所有RemoveDao实现注入到这个Map中，key是Bean的名称
    public RemoveService(Map<String, RemoveDao> daoMap) {
        this.daoMap = daoMap;
    }

    public void executeRemove(String dbType) {
        RemoveDao dao = daoMap.get(dbType + "DbDao");
        dao.removeExpiredData();
    }
}

然后控制器层调用时传入标识：

@PostMapping("/remove_old")
public ResponseEntity<Void> removeOld() {
    removeService.executeRemove("old");
    return ResponseEntity.ok().build();
}

@PostMapping("/remove_new")
public ResponseEntity<Void> removeNew() {
    removeService.executeRemove("new");
    return ResponseEntity.ok().build();
}

这种方式不用创建多个Service Bean，同样不需要修改原有业务逻辑，也能保证线程安全。

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