决战数据库-spring batch(4)数据库到数据库
tags：springbatch
1.引言
上一篇文章《快速使用组件-spring batch(3)读文件数据到数据库》对Spring Batch的读、处理、写组件进行了介绍，并且以实际案例使用了FlatFileItemReader读文本文件，并把每行数据映射为实体，然后使用JdbcBatchItemWriter把实体对象数据存储到MySQL中。但在数据集成的实际应用中，更多的工作是从A数据库到B数据库，数据库之间是异构的，数据与数据的字段定义是不尽相同的，因此，在数据同步、数据抽取时，需要从源数据库读取数据，经过校验、转换，过滤、清洗，然后把数据再写到目标数据库。本文将在上一篇的基础上，实现数据库到数据库的数据同步。简单来说，只需要把从文件读数据改为从数据库读即可。可下载完整示例工程代码参考
2.开发环境
- JDK: jdk1.8
- Spring Boot: 2.1.4.RELEASE
- Spring Batch:4.1.2.RELEASE
- 开发IDE: IDEA
- 构建工具Maven: 3.3.9
- 日志组件logback:1.2.3
- lombok:1.18.6
3.开发流程
上一篇文章中，已经把User数据存储在mytest数据库中，本文将以mytest数据库中的test_user表为源数据，使用Spring Batch把数据同步到目标数据库my_test1，实现MySQL到MySQL的同步。注，若是不同的数据库，在配置多数据源时可更改数据库驱动和连接信息即可。关键代码如下所示：
3.1 创建目标数据库
在MySQL中创建my_test1数据库作为目标数据库，执行示例工程中的sql/my_test1.sql创建用户表，为简单起见，目标数据表与源数据表数据表结构一样。创建后如下：
3.2 配置多数据源
至此，我们的程序涉及三个数据库，分别是：
- 用于
Spring Batch数据存储的my_spring_batch为 - 源数据库
mytest - 目标数据库
my_test1
因此，需要先配置多数据源，配置方法跟之前一样，配置properties文件的数据库连接信息和使用注解进行配置即可。如下：
application.properties
# spring batch db
spring.datasource.jdbc-url=jdbc:mysql://localhost:3310/my_spring_batch?serverTimezone=GMT%2B8&useUnicode=true&characterEncoding=utf8&useSSL=false
spring.datasource.username=root
spring.datasource.password=111111
# origin db
spring.origin-datasource.jdbc-url=jdbc:mysql://localhost:3310/mytest?serverTimezone=GMT%2B8&useUnicode=true&characterEncoding=utf8&useSSL=false
spring.origin-datasource.username=root
spring.origin-datasource.password=111111
# target db
spring.target-datasource.jdbc-url=jdbc:mysql://localhost:3310/my_test1?serverTimezone=GMT%2B8&useUnicode=true&characterEncoding=utf8&useSSL=false
spring.target-datasource.username=root
spring.target-datasource.password=111111
然后使用注解注入多数据源，如下：
DataSourceConfig.java
@Bean("datasource")
@ConfigurationProperties(prefix="spring.datasource")
@Primary
public DataSource batchDatasource() {
return DataSourceBuilder.create().build();
}
@Bean("originDatasource")
@ConfigurationProperties(prefix="spring.origin-datasource")
public DataSource originDatasource() {
return DataSourceBuilder.create().build();
}
@Bean("targetDatasource")
@ConfigurationProperties(prefix="spring.target-datasource")
public DataSource targetDatasource() {
return DataSourceBuilder.create().build();
}
3.3 添加读数据组件JdbcCursorItemReader
从数据库中读取数据，Spring Batch提供了组件JdbcCursorItemReader，通过它，可以把数据库的数据读取出来，然后映射为实体User，以供后续开发。创建方法如下：
@Bean
public ItemReader db2DbItemReader(@Qualifier("originDatasource") DataSource originDatasource) {
String readSql = " select * from test_user";
return new JdbcCursorItemReaderBuilder<User>()
.dataSource(originDatasource).sql(readSql)
.verifyCursorPosition(false).rowMapper(new UserRowMapper())
.build();
}
说明：
- 使用
@Qualifier("originDatasource")标识源数据库 JdbcCursorItemReaderBuilder用于构建JdbcCursorItemReader- 读数据的
sql语句根据实际情况编写即可，此处是读取整个表数据。 - 需要把数据库映射为实体
User，使用UserRowMapper，此mapper实现RowMapper接口，把从数据库读取的ResultSet映射为User的字段。
3.4 自定义处理组件Db2DbItemProcessor
读取到数据后，当前的处理是针对title字段，不为null的则转为大写即可。如下：
if(Objects.nonNull(title)){
user.setTitle(title.toUpperCase());
}
3.5 添加写数据组件JdbcBatchItemWriter
写入数据，同样使用JdbcBatchItemWriter组合，编写插入sql语句，把实体User数据插入到数据库即可，如下：
@Bean
public ItemWriter db2DbWriter(@Qualifier("targetDatasource") DataSource targetDatasource) {
String inserSql ="INSERT INTO test_user(id,name,phone,title,email,gender,date_of_birth,sys_create_time,sys_create_user,sys_update_time,sys_update_user) " +
"VALUES (:id,:name,:phone,:title,:email,:gender,:dateOfBirth,:sysCreateTime,:sysCreateUser,:sysUpdateTime,:sysUpdateUser)";
return new JdbcBatchItemWriterBuilder<User>()
.itemSqlParameterSourceProvider(new BeanPropertyItemSqlParameterSourceProvider<>())
.sql(inserSql)
.dataSource(targetDatasource)
.build();
}
说明：
- 使用
JdbcBatchItemWriterBuilder进行JdbcBatchItemWriter的创建，设置插入数据库的sql语句，同时指定数据源即可。 @Qualifier("targetDatasource") DataSource datasource用于指定数据源- 使用
BeanPropertyItemSqlParameterSourceProvider可以直接把读取的数据实体的属性数据作为参数填充到sql语句中，从而实现数据插入操作。
3.6 组装完整任务
经过上面的操作，可以使用一个java配置，把读、写、处理组装成完整的step和job，如下所示（详细可见示例工程文件）：
@Bean
public Job db2DbJob(Step db2DbStep,JobExecutionListener db2DbListener){
String funcName = Thread.currentThread().getStackTrace()[1].getMethodName();
return jobBuilderFactory.get(funcName)
.listener(db2DbListener)
.flow(db2DbStep)
.end().build();
}
@Bean
public Step db2DbStep(ItemReader db2DbItemReader ,ItemProcessor db2DbProcessor
,ItemWriter db2DbWriter){
String funcName = Thread.currentThread().getStackTrace()[1].getMethodName();
return stepBuilderFactory.get(funcName)
.<User,User>chunk(10)
.reader(db2DbItemReader)
.processor(db2DbProcessor)
.writer(db2DbWriter)
.build();
}
3.7 测试
参考上一文章的File2DbJobTest，编写Db2DbJobTest文件即可。如下：
@Test
public void testDb2DbJob() throws JobParametersInvalidException, JobExecutionAlreadyRunningException, JobRestartException, JobInstanceAlreadyCompleteException {
//构建job参数
JobParameters jobParameters = JobUtil.makeJobParameters();
//运行job
Map<String, Object> stringObjectMap = jobLauncherService.startJob(db2DbJob, jobParameters);
//测试结果
Assert.assertEquals(ExitStatus.COMPLETED,stringObjectMap.get(SyncConstants.STR_RETURN_EXITSTATUS));
}
经过此测试，可查看到源数据库mytest中的test_user表中的数据，已全部同步到目标库my_test1中的test_user中。完成数据库到数据库的数据同步。
4.总结
本文通过简单的示例，从源数据库中读取表数据，经过处理，写入到目标数据库，具体一定的通用性。希望让大家更深入的了解Spring Batch，并能用到实践中。