Spring事务路由
为了将读写事务路由到主节点并将只读事务路由到副本节点，我们可以定义一个连接到主节点的ReadWriteDataSource和一个连接到副本节点的ReadOnlyDataSource。
读写和只读事务路由是通过Spring AbstractRoutingDataSource抽象实现的，该抽象由TransactionRoutingDatasource实现，如下图所示：
TransactionRoutingDataSource非常易于实现，其示例如下：
public class TransactionRoutingDataSource
extends AbstractRoutingDataSource {
@Nullable
@Override
protected Object determineCurrentLookupKey() {
return TransactionSynchronizationManager
.isCurrentTransactionReadOnly() ?
DataSourceType.READ_ONLY :
DataSourceType.READ_WRITE;
}
}
基本上，我们检查Spring TransactionSynchronizationManager类，该类存储当前事务上下文，以检查当前正在运行的Spring事务是否为只读。
determineCurrentLookupKey方法返回将用于选择读写或只读JDBC DataSource的鉴别器值。
DataSourceType只是一个基本的Java枚举，定义了我们的事务路由选项：
public enum DataSourceType {
READ_WRITE,
READ_ONLY
}
Spring读写和只读JDBC DataSource配置
DataSource的配置如下：
@Configuration
@ComponentScan(
basePackages = "com.vladmihalcea.book.hpjp.util.spring.routing"
)
@PropertySource(
"/META-INF/jdbc-postgresql-replication.properties"
)
public class TransactionRoutingConfiguration
extends AbstractJPAConfiguration {
@Value("${jdbc.url.primary}")
private String primaryUrl;
@Value("${jdbc.url.replica}")
private String replicaUrl;
@Value("${jdbc.username}")
private String username;
@Value("${jdbc.password}")
private String password;
@Bean
public DataSource readWriteDataSource() {
PGSimpleDataSource dataSource = new PGSimpleDataSource();
dataSource.setURL(primaryUrl);
dataSource.setUser(username);
dataSource.setPassword(password);
return connectionPoolDataSource(dataSource);
}
@Bean
public DataSource readOnlyDataSource() {
PGSimpleDataSource dataSource = new PGSimpleDataSource();
dataSource.setURL(replicaUrl);
dataSource.setUser(username);
dataSource.setPassword(password);
return connectionPoolDataSource(dataSource);
}
@Bean
public TransactionRoutingDataSource actualDataSource() {
TransactionRoutingDataSource routingDataSource =
new TransactionRoutingDataSource();
Map<Object, Object> dataSourceMap = new HashMap<>();
dataSourceMap.put(
DataSourceType.READ_WRITE,
readWriteDataSource()
);
dataSourceMap.put(
DataSourceType.READ_ONLY,
readOnlyDataSource()
);
routingDataSource.setTargetDataSources(dataSourceMap);
return routingDataSource;
}
@Override
protected Properties additionalProperties() {
Properties properties = super.additionalProperties();
properties.setProperty(
"hibernate.connection.provider_disables_autocommit",
Boolean.TRUE.toString()
);
return properties;
}
@Override
protected String[] packagesToScan() {
return new String[]{
"com.vladmihalcea.book.hpjp.hibernate.transaction.forum"
};
}
@Override
protected String databaseType() {
return Database.POSTGRESQL.name().toLowerCase();
}
protected HikariConfig hikariConfig(
DataSource dataSource) {
HikariConfig hikariConfig = new HikariConfig();
int cpuCores = Runtime.getRuntime().availableProcessors();
hikariConfig.setMaximumPoolSize(cpuCores * 4);
hikariConfig.setDataSource(dataSource);
hikariConfig.setAutoCommit(false);
return hikariConfig;
}
protected HikariDataSource connectionPoolDataSource(
DataSource dataSource) {
return new HikariDataSource(hikariConfig(dataSource));
}
}
/META-INF/jdbc-postgresql-replication.properties 资源文件提供了读写和只读 JDBC DataSource 组件的配置：
hibernate.dialect=org.hibernate.dialect.PostgreSQL10Dialect
jdbc.url.primary=jdbc:postgresql://localhost:5432/high_performance_java_persistence
jdbc.url.replica=jdbc:postgresql://localhost:5432/high_performance_java_persistence_replica
jdbc.username=postgres
jdbc.password=admin
jdbc.url.primary属性定义了主节点的URL，而jdbc.url.replica则定义了副本节点的URL。
readWriteDataSource Spring组件定义了读写JDBC DataSource，而readOnlyDataSource组件则定义了只读JDBC DataSource。
请注意，读写和只读数据源都使用HikariCP进行连接池管理。有关使用数据库连接池的好处的更多详细信息，请参见相关文档。
actualDataSource作为读写和只读数据源的外观，使用TransactionRoutingDataSource实用程序实现。
readWriteDataSource使用DataSourceType.READ_WRITE键进行注册，而readOnlyDataSource则使用DataSourceType.READ_ONLY键进行注册。
因此，在执行读写@Transactional方法时，将使用readWriteDataSource，而在执行@Transactional（readOnly = true）方法时，则将使用readOnlyDataSource。
请注意，
additionalProperties方法定义了
hibernate.connection.provider_disables_autocommit Hibernate属性，我将其添加到Hibernate中，以延迟RESOURCE_LOCAL JPA事务的数据库获取。
hibernate.connection.provider_disables_autocommit不仅可以让您更好地使用数据库连接，而且它是我们使此示例工作的唯一方式，因为如果没有此配置，则会在调用
determineCurrentLookupKey方法
TransactionRoutingDataSource之前获取连接。
构建JPA
EntityManagerFactory所需的其余Spring组件由
AbstractJPAConfiguration基类定义。
基本上，
actualDataSource被DataSource-Proxy进一步包装并提供给JPA的
ENtityManagerFactory。您可以查看
GitHub上的源代码以获取更多详细信息。
测试时间
为了检查事务路由是否起作用，我们将通过在
postgresql.conf配置文件中设置以下属性来启用PostgreSQL查询日志：
log_min_duration_statement = 0
log_line_prefix = '[%d] '
log_min_duration_statement属性设置用于记录所有PostgreSQL语句，而第二个属性将数据库名称添加到SQL日志中。
因此，在调用newPost和findAllPostsByTitle方法时，应像这样：
Post post = forumService.newPost(
"High-Performance Java Persistence",
"JDBC", "JPA", "Hibernate"
);
List<Post> posts = forumService.findAllPostsByTitle(
"High-Performance Java Persistence"
);
我们可以看到，PostgreSQL 记录了以下消息：
[high_performance_java_persistence] LOG: execute <unnamed>:
BEGIN
[high_performance_java_persistence] DETAIL:
parameters: $1 = 'JDBC', $2 = 'JPA', $3 = 'Hibernate'
[high_performance_java_persistence] LOG: execute <unnamed>:
select tag0_.id as id1_4_, tag0_.name as name2_4_
from tag tag0_ where tag0_.name in ($1 , $2 , $3)
[high_performance_java_persistence] LOG: execute <unnamed>:
select nextval ('hibernate_sequence')
[high_performance_java_persistence] DETAIL:
parameters: $1 = 'High-Performance Java Persistence', $2 = '4'
[high_performance_java_persistence] LOG: execute <unnamed>:
insert into post (title, id) values ($1, $2)
[high_performance_java_persistence] DETAIL:
parameters: $1 = '4', $2 = '1'
[high_performance_java_persistence] LOG: execute <unnamed>:
insert into post_tag (post_id, tag_id) values ($1, $2)
[high_performance_java_persistence] DETAIL:
parameters: $1 = '4', $2 = '2'
[high_performance_java_persistence] LOG: execute <unnamed>:
insert into post_tag (post_id, tag_id) values ($1, $2)
[high_performance_java_persistence] DETAIL:
parameters: $1 = '4', $2 = '3'
[high_performance_java_persistence] LOG: execute <unnamed>:
insert into post_tag (post_id, tag_id) values ($1, $2)
[high_performance_java_persistence] LOG: execute S_3:
COMMIT
[high_performance_java_persistence_replica] LOG: execute <unnamed>:
BEGIN
[high_performance_java_persistence_replica] DETAIL:
parameters: $1 = 'High-Performance Java Persistence'
[high_performance_java_persistence_replica] LOG: execute <unnamed>:
select post0_.id as id1_0_, post0_.title as title2_0_
from post post0_ where post0_.title=$1
[high_performance_java_persistence_replica] LOG: execute S_1:
COMMIT
使用 high_performance_java_persistence 前缀的日志语句在主节点上执行，而使用 high_performance_java_persistence_replica 的日志语句则在副本节点上执行。
所以，一切都运作得很顺利！
所有源代码都可以在我的 High-Performance Java Persistence GitHub 存储库中找到，因此您也可以尝试它。
结论
这种要求非常有用，因为 单主数据库复制 架构不仅提供了容错性和更好的可用性，还允许我们通过添加更多的副本节点来扩展读操作。
hibernate.connection.provider_disables_autocommit。 - Vlad Mihalcea