Spring Boot использует несколько источников данных в одной транзакции - PullRequest
1 голос
/ 27 сентября 2019

Мои настройки следующие:

У меня есть приложение Spring Boot 2.1.1.Для пула подключений используется Hikari.

Я хотел бы написать одну службу, которая перенаправляет READ в базу данных READER, а WRITES - в базу данных Master в одну транзакцию / метод.

Итак, я хочу, чтобы следующее работало:

@Transactional
public test(Test test){
    Optional<Res> result = myRepo.findByAttr(test.getTest()); //do that on reader database
    validateIfResultExists(result); 

    myRepo.save(test);//do this on master database
}

Для этого я написал пользовательский DbContextHolder:

public class DbContextHolder {
private static final ThreadLocal<DbType> contextHolder = new ThreadLocal<DbType>();


public static void setDbType(DbType dbType) {
    if(dbType == null){
        throw new NullPointerException();
    }
    contextHolder.set(dbType);
}

public static DbType getDbType() {
    if(contextHolder.get() == null){
        DbContextHolder.setDbType(DbType.MASTER);
    }

    return (DbType) contextHolder.get();
}

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

У меня также есть собственный RoutingDataSource:

public class RoutingDataSource extends AbstractRoutingDataSource {
@Override
protected Object determineCurrentLookupKey() {
    return DbContextHolder.getDbType();
}
   }

И источник данных для настройки базы данных с помощью hikari:

@Configuration
public class DataSourceConfig {

private static final String PRIMARY_DATASOURCE_PREFIX = "spring.primary.datasource";
private static final String REPLICA_DATASOURCE_PREFIX = "spring.replica.datasource";

@Autowired
private Environment environment;

@Value("${spring.datasource.hikari.connectionTimeout}")
private Long CONNECTION_TIMEOUT;

@Value("${spring.datasource.hikari.idleTimeout}")
private Long IDLE_TIMEOUT;

@Value("${spring.datasource.hikari.maxLifetime}")
private Long MAX_LIFETIME_TIMEOUT;

@Value("${spring.datasource.hikari.minimumIdle}")
private int MINIMUM_IDLE;

@Value("${spring.datasource.hikari.maximumPoolSize}")
private int MAX_POOL_SIZE;

@Resource

@Bean
@Primary
public DataSource dataSource() {
    final RoutingDataSource routingDataSource = new RoutingDataSource();

    final DataSource primaryDataSource = buildDataSource("PrimaryHikariPool", PRIMARY_DATASOURCE_PREFIX);
    final DataSource replicaDataSource = buildDataSource("ReplicaHikariPool", REPLICA_DATASOURCE_PREFIX);

    final Map<Object, Object> targetDataSources = new HashMap<>();
    targetDataSources.put(DbType.MASTER, primaryDataSource);
    targetDataSources.put(DbType.REPLICA, replicaDataSource);

    routingDataSource.setTargetDataSources(targetDataSources);
    routingDataSource.setDefaultTargetDataSource(primaryDataSource);

    return routingDataSource;
}

private DataSource buildDataSource(String poolName, String dataSourcePrefix) {
    final HikariConfig hikariConfig = new HikariConfig();
    hikariConfig.setConnectionTimeout(CONNECTION_TIMEOUT);
    hikariConfig.setIdleTimeout(IDLE_TIMEOUT);
    hikariConfig.setMaxLifetime(MAX_LIFETIME_TIMEOUT);
    hikariConfig.setMinimumIdle(MINIMUM_IDLE);
    hikariConfig.setMaximumPoolSize(MAX_POOL_SIZE);

    hikariConfig.setPoolName(poolName);
    hikariConfig.setJdbcUrl(environment.getProperty(String.format("%s.url", dataSourcePrefix)));
    hikariConfig.setUsername(environment.getProperty(String.format("%s.username", dataSourcePrefix)));
    hikariConfig.setPassword(environment.getProperty(String.format("%s.password", dataSourcePrefix)));
    hikariConfig.setDriverClassName(environment.getProperty(String.format("%s.driver", dataSourcePrefix)));

    return new HikariDataSource(hikariConfig);
}

}

Но, похоже, это не работает.Если я регистрирую текущий DbType, все выглядит ужасно, но это не работает, так как я получаю следующую ошибку:

ОШИБКА: невозможно выполнить INSERT в транзакции только для чтения

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...