springboot mongodb 动态数据源

要实现在Spring Boot中使用MongoDB动态数据源，可以按照以下步骤进行操作：

1. 导入MongoDB的Java驱动和Spring Boot的MongoDB依赖：

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-data-mongodb</artifactId>
</dependency>
<dependency>
    <groupId>org.mongodb</groupId>
    <artifactId>mongo-java-driver</artifactId>
</dependency>

2. 创建动态数据源配置类，继承AbstractRoutingDataSource类，并实现determineCurrentLookupKey()方法，用于动态获取数据源的key：

public class DynamicDataSource extends AbstractRoutingDataSource {

    @Override
    protected Object determineCurrentLookupKey() {
        return DataSourceContextHolder.getDataSourceKey();
    }
}

3. 创建数据源上下文类DataSourceContextHolder，用于存储当前数据源的key：

public class DataSourceContextHolder {

    private static final ThreadLocal<String> CONTEXT_HOLDER = new ThreadLocal<>();

    public static void setDataSourceKey(String dataSourceKey) {
        CONTEXT_HOLDER.set(dataSourceKey);
    }

    public static String getDataSourceKey() {
        return CONTEXT_HOLDER.get();
    }

    public static void clearDataSourceKey() {
        CONTEXT_HOLDER.remove();
    }
}

4. 创建数据源配置类，用于配置多个MongoDB数据源：

@Configuration
public class DataSourceConfig {

    @Bean(name = "dataSource1")
    @ConfigurationProperties(prefix = "spring.data.mongodb.datasource1")
    public MongoClient mongoClient1() {
        return MongoClients.create();
    }

    @Bean(name = "dataSource2")
    @ConfigurationProperties(prefix = "spring.data.mongodb.datasource2")
    public MongoClient mongoClient2() {
        return MongoClients.create();
    }

    @Bean
    public DynamicDataSource dynamicDataSource(@Qualifier("dataSource1") MongoClient dataSource1,
                                               @Qualifier("dataSource2") MongoClient dataSource2) {
        Map<Object, Object> targetDataSources = new HashMap<>();
        targetDataSources.put("dataSource1", dataSource1);
        targetDataSources.put("dataSource2", dataSource2);
        DynamicDataSource dynamicDataSource = new DynamicDataSource();
        dynamicDataSource.setTargetDataSources(targetDataSources);
        dynamicDataSource.setDefaultTargetDataSource(dataSource1);
        return dynamicDataSource;
    }
}

5. 在需要使用动态数据源的地方，通过调用DataSourceContextHolder.setDataSourceKey()方法来设置当前数据源的key：

@Service
public class UserServiceImpl implements UserService {

    @Autowired
    private MongoTemplate mongoTemplate;

    @Override
    public void addUser(User user) {
        DataSourceContextHolder.setDataSourceKey("dataSource2");
        mongoTemplate.save(user);
        DataSourceContextHolder.clearDataSourceKey();
    }
}

以上就是在Spring Boot中使用MongoDB动态数据源的步骤，希望能对你有所帮助。