具有 2 个数据源的 Spring JDBC 存储库对其中一个存储库使用了不正确的数据源

如何解决具有 2 个数据源的 Spring JDBC 存储库对其中一个存储库使用了不正确的数据源

我正在尝试使用 Spring JDBC Repositories 将一些数据从一个数据库 (Apache Derby) 迁移到另一个 (Postgres)，即我只从 Derby 数据库读取数据，并且只写入 Postres DB。我已经配置了 2 个 DataSource 和 2 个 TransactionManager 对象：

@Configuration
@EnableJdbcRepositories(basePackages = "com.example.jdbctest.repository.derby",jdbcOperationsRef = DERBY_NAMED_PARAMETER_JDBC_OPERATIONS)
public class DerbyDataSourceConfig extends AbstractJdbcConfiguration
{
    @Bean(DERBY_CONFIG)
    @Primary
    @ConfigurationProperties(prefix = "spring.derby.datasource.hikari")
    public HikariConfig derbyDataSourceProperties()
    {

        return new HikariConfig();
    }

    @Bean(name = DERBY_DATA_SOURCE)
    @Primary
    public DataSource derbyDataSource(@Qualifier(DERBY_CONFIG) HikariConfig dataSourceProperties)
    {
        return new HikariDataSource(dataSourceProperties);
    }

    @Bean(name = DERBY_NAMED_PARAMETER_JDBC_OPERATIONS)
    @Primary
    public NamedParameterJdbcOperations namedParameterJdbcOperations(@Qualifier(DERBY_DATA_SOURCE) final DataSource dataSource)
    {
        return new NamedParameterJdbcTemplate(dataSource);
    }

    @Bean(name = DERBY_TRANSACTION_MANAGER)
    @Primary
    public TransactionManager transactionManager(@Qualifier(DERBY_DATA_SOURCE) final DataSource dataSource)
    {
        return new DataSourceTransactionManager(dataSource);
    }

}

@Configuration
@EnableJdbcRepositories(basePackages = "com.example.jdbctest.repository.postgres",jdbcOperationsRef = POSTGRES_NAMED_PARAMETER_JDBC_OPERATIONS)
public class PostgresDataSourceConfig extends AbstractJdbcConfiguration
{
    @Bean(POSTGRES_CONFIG)
    @ConfigurationProperties(prefix = "spring.postgres.datasource.hikari")
    public HikariConfig postgresDataSourceProperties()
    {
        return new HikariConfig();
    }

    @Bean(name = POSTGRES_DATA_SOURCE)
    public DataSource postgresDataSource(@Qualifier(POSTGRES_CONFIG) final HikariConfig dataSourceProperties)
    {
        return new HikariDataSource(dataSourceProperties);
    }

    @Bean(name = POSTGRES_NAMED_PARAMETER_JDBC_OPERATIONS)
    public NamedParameterJdbcOperations namedParameterJdbcOperations(@Qualifier(POSTGRES_DATA_SOURCE) final DataSource dataSource)
    {
        return new NamedParameterJdbcTemplate(dataSource);
    }

    @Bean(name = POSTGRES_TRANSACTION_MANAGER)
    public TransactionManager transactionManager(@Qualifier(POSTGRES_DATA_SOURCE) final DataSource dataSource)
    {
        return new DataSourceTransactionManager(dataSource);
    }
}

因为我使用的是 Apache derby，所以我必须创建自己的 Dialect 和 DialectResolver，虽然我不知道 Derby 方言是什么，所以我只是扩展了 AnsiDialect 以查看发生：

public class DerbyDialect extends AnsiDialect
{
    public static final DerbyDialect INSTANCE = new DerbyDialect();
}

public class DerbyDialectResolver implements DialectResolver.JdbcDialectProvider
{
    @Override
    public Optional<Dialect> getDialect(final JdbcOperations jdbcOperations)
    {
        return Optional.ofNullable(
                jdbcOperations.execute((ConnectionCallback<Dialect>) DerbyDialectResolver::getDialect));

    }

    private static Dialect getDialect(Connection connection) throws SQLException
    {
        DatabaseMetaData metaData = connection.getMetaData();
        String name = metaData.getDatabaseProductName().toLowerCase(Locale.ROOT);
        if (name.contains("derby")) {
            return DerbyDialect.INSTANCE;
        }
        return null;
    }
}

读取和写入的两个存储库是：

@DerbyTransactionManager
public interface DerbyStockRepository extends CrudRepository<Stock,Long>
{
    @Query("SELECT * FROM TT.STOCK")
    List<Stock> selectAll();
}

@PostgresTransactionManager
public interface PostgresStockRepository extends CrudRepository<PostgresStock,Long>
{
    @Override
    @PostgresTransactionManager
    <S extends PostgresStock> Iterable<S> saveAll(Iterable<S> iterable);
}

上面的两个自定义注解是：

@Target({ElementType.TYPE,ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Inherited
@Transactional(transactionManager = DERBY_TRANSACTION_MANAGER)
public @interface DerbyTransactionManager
{
}

@Target({ElementType.TYPE,ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Inherited
@Transactional(transactionManager = POSTGRES_TRANSACTION_MANAGER)
public @interface PostgresTransactionManager
{
}

使用 Derby 存储库读取数据很好，但是当我尝试使用 postgresRepository.saveAll() 写入 postgres 存储库时，出现以下错误：

Caused by: org.springframework.dao.DataAccessResourceFailureException: PreparedStatementCallback; SQL [INSERT INTO "stock" ("name","symbol") VALUES (?,?)]; Schema 'EXAMPLE' does not exist; nested exception is java.sql.SQLSyntaxErrorException: Schema 'EXAMPLE' does not exist

...Stack trace ommitted for brevity ...

Caused by: java.sql.SQLSyntaxErrorException: Schema 'EXAMPLE' does not exist
    at org.apache.derby.client.am.SQLExceptionFactory.getSQLException(Unknown Source) ~[derbyclient-10.14.2.0.jar:na]
    at org.apache.derby.client.am.SqlException.getSQLException(Unknown Source) ~[derbyclient-10.14.2.0.jar:na]

从堆栈跟踪中我可以看出 PostgreRepository 使用了不正确的数据源，因为“缺失”模式是 Derby 数据库中的一个模式，后来在堆栈跟踪中它指的是 Apache Derby 类。不确定这是如何发生的，因为我相信我已经仔细地注释了配置类和存储库。