Spring Boot 1数据库多个数据库用户

Question

我目前正在构建一个访问数据库的restful API（带有Spring Boot）。此应用程序最终将托管在服务器上。

我想做的事情：

1. 配置多个用户的数据库，并为他们分配不同的权限到不同的表
2. 根据调用的endpoint，使用特定用户在该函数中执行该查询

如何配置上述应用程序？

到目前为止，我找到的答案涉及到配置多个数据源，但是对于上面的应用程序，只有一个数据源，但是有多个用户。

我已经阅读了以下链接，但仍在努力理解它

https://kimrudolph.de/blog/spring-datasource-routing

我是否使用相同的URL（在本例中为jdbcURL）配置多个数据源，并为每个角色配置多组用户名和密码？

谢谢你！

Answer 1

您已经找到了完成该任务的最佳方法，我的意思是AbstractRoutingDataSource。如果用户数量固定，则可以使用以下方法：

public final class RoutingDataSource extends AbstractRoutingDataSource {
    @Override
    protected Object determineCurrentLookupKey() {
        return UserContextHolder.getUserName() + "DataSource";
    }
}

和配置：

@Bean
@Qualifier("user1DataSource")
public DataSource userOneDataSource() {
    return DataSourceBuilder.create()
            .username("user1")
            .password("pass")
             ...
             .build();
}

@Bean
@Qualifier("user2DataSource")
public DataSource userOneDataSource() {
    return DataSourceBuilder.create()
            .username("user2")
             ...
             .build();
}

@Bean
@Primary
public RoutingDataSource dataSource(Map<String, DataSource> datasources) {
    return new RoutingDataSource().dataSource(datasources);
 }

如果您需要在运行时添加用户和数据源，您可以使用如下内容：

public final class RoutingDataSource extends AbstractRoutingDataSource {
    private final ConcurrentHashMap<String, DataSource> dynamicDataSources = new ConcurrentHashMap<>();

    @Override
    protected Object determineCurrentLookupKey() {
        return UserContextHolder.getUserName() + "_datasource";
    }

    @Override
    protected DataSource determineTargetDataSource() {
        String currentLookupKey = this.determineCurrentLookupKey().toString();
        String userName = UserContextHolder.getUserName();
        String password = UserContextHolder.getPassword();

        return this.dynamicDataSources.computeIfAbsent(currentLookupKey, (key) -> DataSourceBuilder.create()
                .driverClassName("your.driver.class")
                .url("jdbc:same:url/here")
                .username(userName)
                .password(password)
                .build());
    }
}