AbstractRoutingDataSource 不会改变连接

Question

我使用AbstractRoutingDataSource 动态更改数据源并使用ThreadLocal 设置currentLookupKey。当我每个 http 请求只使用一个数据源时，它工作得很好。我用JpaRepository

@Component
@Primary
public class RoutingDataSource extends AbstractRoutingDataSource {

    @Autowired
    private DatabaseMap databaseMap;

    @Override
    public void afterPropertiesSet() {
        setTargetDataSources(databaseMap.getSourcesMap());
        setDefaultTargetDataSource(databaseMap.getSourcesMap().get("DEFAULT"));
        super.afterPropertiesSet();
    }

    @Override
    protected Object determineCurrentLookupKey() {
        return DatabaseContextHolder.getDatabaseType();
    }

}

public class DatabaseContextHolder {

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

    public static void setDatabaseType(String string) {
        contextHolder.set(string);
    }

    public static String getDatabaseType() {
        return (String) contextHolder.get();
    }

    public static void clearDatabaseType() {
        contextHolder.remove();
    }
}

当我尝试在我的 REST 控制器中获取数据时，我只从一个数据库中获取数据。

我的 REST 控制器中的一些代码

DatabaseContextHolder.setDatabaseType("db1");
//here I get data from db1 as expected
//I use JpaRepository
DatabaseContextHolder.clearDatabaseType();
DatabaseContextHolder.setDatabaseType("db2");
//here I should get data from db2 but get from db1

我尝试调试，看起来Spring在http请求中只获取了一次数据源。

这个方法只被调用一次。

@Override
public Connection getConnection() throws SQLException {
    return determineTargetDataSource().getConnection();
}

有什么办法可以强制Spring改变数据源。

Answer 1

• 我的怀疑是你有一个用@Transactional 注释注释的方法。在调用该事务方法之前，您首先指定一个数据源键并调用事务方法。在事务方法中，您首先调用存储库，它使用您设置的数据源查找键按预期工作。但是，您在事务方法中设置不同的密钥，并调用另一个存储库，它仍然使用您第一次设置的密钥。

• DataSource 将在transaction 启动时由框架选择，因此如果您使用@Transactional 注释，那么您在方法内进行的任何切换都是无用的。 因为数据源将具有由为@Transactional 注释创建的代理选择。最好的选择是在非事务性服务中使用分支逻辑或使用TransactionTemplate 而不是@Transactional

• 例如，确保YourRestController 在此yourRestControllerMethod 中没有类级别@Transactional 以及@Transactional 注释，您将保留它们以供您使用。

     @RestController
     public class YourRestController {

      @Autowired
      TransactionalService transactional

      public void yourRestControllerMethod(){
        //set the datasource look up key to A1 and then
        transactional.methodA1();
        //change datasource look up key to A2 and then
        transactional.methodA2();
      }

     }

    @Service
    public class TransactionalService {

       @Transactional
       public void methodA1(){

       }
       
       @Transactional
       public void methodA2() {

       }

    }

Answer 2

您的问题可能与事务定界有关。

当您在代码中定义 @Transactional 注释时，Spring 将代表您创建开始和结束所需的所有内容，并在需要时提交或回滚事务。

正如您在DataSourceTransactionManager 类的source code 中的doBegin 方法中看到的那样——这同样适用于其他事务管理器——，Spring 在事务初始化时获得一个Connection——这就是为什么方法 getConnection 仅被调用一次 - ，它将在该事务中针对数据库的所有底层操作重用该连接（这对于 ACID 保存有意义）。

因此，如果您需要在同一个请求处理中连接到多个数据源，您可以在服务代码中定义不同的方法，每个方法都使用@Transactional 注释进行注释，并根据您之前的需要更改底层数据源调用它们：

DatabaseContextHolder.setDatabaseType("db1");
// Invoke a service method annotated with @Transactional
// It can use the underlying JpaRepositories that you need
DatabaseContextHolder.clearDatabaseType();
DatabaseContextHolder.setDatabaseType("db2");
// Invoke again another (or the same, what you need) service method
// annotated with @Transactional, You should get data from db2 this time

Answer 3

我遇到了同样的问题，上述解决方案都无法解决它。但是使我的服务方法 final（在我的 REST 控制器中）

public final Response

Answer 4

将spring.jpa.open-in-view 设置为false。