Springboot+Mybatis+通用Mapper多数据源实现数据同步

Architect_home 2019-04-19

展开全文

有个需求需要查询远程数据库，然后将远程数据库中的数据抓取到本地，远程数据库中的数据是每天都增加的，所以就需要写个程序自动实现实时抓取。

这里我用到的框架是Springboot2.0+Mybatis+Mapper，涉及的数据库有SqlServer，Oracle，Mysql

添加依赖

<dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-quartz</artifactId>
        </dependency>
        <dependency>
            <groupId>org.quartz-scheduler</groupId>
            <artifactId>quartz</artifactId>
            <version>2.3.0</version>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-jdbc</artifactId>
        </dependency>
        <dependency>
            <groupId>org.mybatis.spring.boot</groupId>
            <artifactId>mybatis-spring-boot-starter</artifactId>
            <version>1.3.1</version>
        </dependency>
        <!--<dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-data-jpa</artifactId>
        </dependency>-->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-devtools</artifactId>
            <scope>runtime</scope>
        </dependency>
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <optional>true</optional>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
            <scope>test</scope>
        </dependency>
        <dependency>
            <groupId>commons-lang</groupId>
            <artifactId>commons-lang</artifactId>
            <version>2.6</version>
        </dependency>
         <!-- sqlserver -->
        <dependency>
            <groupId>com.microsoft.sqlserver</groupId>
            <artifactId>sqljdbc4</artifactId>
            <version>4.4</version>
        </dependency>
        <!-- oracle -->
        <dependency>
            <groupId>com.oracle</groupId>
            <artifactId>ojdbc14</artifactId>
            <version>10.2.0.4.0</version>
        </dependency>
        <!-- 通用Maper -->
        <dependency>
            <groupId>tk.mybatis</groupId>
            <artifactId>mapper-spring-boot-starter</artifactId>
            <version>1.2.4</version>
        </dependency>
        <!--druid -->
        <dependency>
            <groupId>com.alibaba</groupId>
            <artifactId>druid</artifactId>
            <version>1.0.28</version>
        </dependency>
        <!-- log4j -->
        <dependency>
            <groupId>org.slf4j</groupId>
            <artifactId>slf4j-api</artifactId>
            <version>1.7.25</version>
        </dependency>

如果没有log包，可能在配置文件中的DruidDataSource会出现错误提示，但实际上不会影响到程序。

application.yml项目配置

也可以用application.properties格式配置，但是感觉yml格式看起来比较直观.

##端口
server:
  port:8081
logging:
  level:
    cn.javabb.bootdemo.mapper: info
spring:
  remote-datasource:
    url: jdbc:sqlserver://66.66.66.66:1433;DatabaseName=dbName
    username: sa
    password: 123
    #使用Druid的数据源
    type: com.alibaba.druid.pool.DruidDataSource
    driver-class-name: com.microsoft.sqlserver.jdbc.SQLServerDriver
    filters: stat
    maxActive: 20
    initialSize: 1
    maxWait: 60000
    minIdle: 1
    timeBetweenEvictionRunsMillis: 60000
    minEvictableIdleTimeMillis: 300000
    validationQuery: select 'x'
    testWhileIdle: true
    testOnBorrow: false
    testOnReturn: false
    poolPreparedStatements: true
    maxOpenPreparedStatements: 20
  local-datasource:
    url: jdbc:sqlserver://77.77.77.77:1433;DatabaseName=dbName
    username: sa
    password: 1234
    #使用Druid的数据源
    type: com.alibaba.druid.pool.DruidDataSource
    driver-class-name: com.microsoft.sqlserver.jdbc.SQLServerDriver
    filters: stat
    maxActive: 20
    initialSize: 1
    maxWait: 60000
    minIdle: 1
    timeBetweenEvictionRunsMillis: 60000
    minEvictableIdleTimeMillis: 300000
    validationQuery: select 'x'
    testWhileIdle: true
    testOnBorrow: false
    testOnReturn: false
    poolPreparedStatements: true
    maxOpenPreparedStatements: 20
  oracle-datasource:
    url: jdbc:oracle:thin:@88.88.88.88:1521/sid
    username: tiger
    password: tiger
    #使用Druid的数据源
    type: com.alibaba.druid.pool.DruidDataSource
    driver-class-name: oracle.jdbc.driver.OracleDriver
    filters: stat
    maxActive: 20
    initialSize: 1
    maxWait: 60000
    minIdle: 1
    timeBetweenEvictionRunsMillis: 60000
    minEvictableIdleTimeMillis: 300000
    validationQuery: select 'x'
    testWhileIdle: true
    testOnBorrow: false
    testOnReturn: false
    poolPreparedStatements: true
    maxOpenPreparedStatements: 20
#Mybatis
mybatis:
  configuration:
    map-underscore-to-camel-case: true
  mapper-locations: classpath:mapper/*.xml
  typeAliasesPackage: cn.javabb.**.entity
#Mapper
mapper:
  mappers:
  - cn.javabb.bootdemo.base.BaseMapper
  not-empty: false
  identity: sqlserver
  before: true

其中remote-datasource,oracle-datasource和local-datasource为两个不同的数据源，mybatis配置和通用mapper配置一样的,如果不需要用到通用mapper可以去掉配置。

application启动类：

@EnableAsync
@EnableScheduling
@ServletComponentScan
@SpringBootApplication(exclude = {DataSourceAutoConfiguration.class,DataSourceTransactionManagerAutoConfiguration.class, MybatisAutoConfiguration.class})
public class BootDemoApplication {
    public static void main(String[] args) {
        SpringApplication.run(BootDemoApplication.class, args);
    }
}

注解说明：

@EnableAsync：开启异步任务支持，如果不需要可以去掉

@EnableScheduling：开启定时任务支持

@ServletComponentScan：用来扫描mapper

@SpringBootApplication:springboot注解支持

注意：如果项目依赖中有，mybatis-spring-boot-starter，会默认自动根据配置文件配置mybatis,如果刚开始项目没有配置DataSource，启动项目就会报错，在这里去掉DataSource的默认自动配置类不加载，使项目在使用的时候选择去DataSource。

数据源配置

有几个数据源配置几个DataSource

配置第一个数据源：local-datasource

import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.SqlSessionFactoryBean;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.autoconfigure.jdbc.DataSourceBuilder;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;
import tk.mybatis.spring.annotation.MapperScan;
import javax.sql.DataSource;
@Configuration
@MapperScan(basePackages = "cn.javabb.bootdemo.mapper.local",sqlSessionFactoryRef = "localSqlSessionFactory")
public class LocalDataSourceConfig {
    @Bean(name = "localDataSource")
    @ConfigurationProperties("spring.local-datasource")
    public DataSource localDataSource(){
        return DataSourceBuilder.create().build();
    }
    @Bean(name = "localSqlSessionFactory")
    public SqlSessionFactory sqlSessionFactory(@Qualifier("localDataSource") DataSource dataSource) throws Exception {
        SqlSessionFactoryBean sessionFactoryBean = new SqlSessionFactoryBean();
        sessionFactoryBean.setDataSource(dataSource);
        sessionFactoryBean.setMapperLocations(new PathMatchingResourcePatternResolver()
                .getResources("classpath*:mapper/local/*.xml"));
        return sessionFactoryBean.getObject();
    }
    @Bean(name = "localTransactionManager")
    public DataSourceTransactionManager localTransactionManager(){
        return new DataSourceTransactionManager(localDataSource());
    }
}

spring.local-datasource 为application.yml中配置的数据源名称

classpath*:mapper/local/*.xml 指定local数据源的mybatis的xml文件位置

原本一个数据源的时候，可以在启动程序的时候加载DataSource，但是这里是多数据源，需要为每个数据源配置DataSource，并且需要添加MapperScan注解，（因为这里是使用的通用Mapper，所以引入的class应该是mapper的，import tk.mybatis.spring.annotation.MapperScan;，如果没用就引入org的MapperScan）

basePackages这里的包路径是需要用到这个数据源的路径，这个包下所有的接口文件都会使用这个local数据源

配置第二个数据源：remote-datasource

@Configuration
@MapperScan(basePackages = "cn.javabb.bootdemo.mapper.remote",sqlSessionFactoryRef = "remoteSqlSessionFactory")
public class RemoteDataSourceConfig {
    @Bean(name = "remoteDataSource")
    @Primary
    @ConfigurationProperties("spring.remote-datasource")
    public DataSource remoteDataSource(){
        return DataSourceBuilder.create().build();
    }
    @Bean(name = "remoteSqlSessionFactory")
    @Primary
    public SqlSessionFactory sqlSessionFactory(@Qualifier("remoteDataSource") DataSource dataSource) throws Exception {
        SqlSessionFactoryBean sessionFactoryBean = new SqlSessionFactoryBean();
        sessionFactoryBean.setDataSource(dataSource);
        sessionFactoryBean.setMapperLocations(new PathMatchingResourcePatternResolver()
                .getResources("classpath*:mapper/remote/*.xml"));
        return sessionFactoryBean.getObject();
    }
    @Bean(name = "remoteTransactionManager")
    @Primary
    public DataSourceTransactionManager remoteTransactionManager(){
        return new DataSourceTransactionManager(remoteDataSource());
    }
}

@Primary指定默认数据源。

配置第三个数据源：oracle-datasource

@Configuration
@MapperScan(basePackages = "cn.javabb.bootdemo.mapper.oracle",sqlSessionFactoryRef = "oracleSqlSessionFactory")
public class OracleDataSourceConfig {
    @Bean(name = "oracleDataSource")
    @ConfigurationProperties("spring.oracle-datasource")
    public DataSource oracleDataSource(){
        return DataSourceBuilder.create().build();
    }
    @Bean(name = "oracleSqlSessionFactory")
    public SqlSessionFactory sqlSessionFactory(@Qualifier("oracleDataSource") DataSource dataSource) throws Exception {
        SqlSessionFactoryBean sessionFactoryBean = new SqlSessionFactoryBean();
        sessionFactoryBean.setDataSource(dataSource);
        sessionFactoryBean.setMapperLocations(new PathMatchingResourcePatternResolver()
                .getResources("classpath*:mapper/oracle/*.xml"));
        return sessionFactoryBean.getObject();
    }
    @Bean(name = "oracleTransactionManager")
    public DataSourceTransactionManager oracleTransactionManager(){
        return new DataSourceTransactionManager(oracleDataSource());
    }
}

接下来所有的接口都应该按照不同的数据源放置，对应的xml也需要按照不同的数据源分开。

按照我项目的实际例子：

Mapper（使用local数据源，下面所有的方法都会在local数据源中查询）

import cn.javabb.bootdemo.entity.*;
import org.apache.ibatis.annotations.Mapper;
import org.springframework.dao.DataAccessException;
import org.springframework.stereotype.Component;
/**
 * @Author QINB imqinbao@163.com
 * @CreateDate 2019/1/18/018 13:08
 * @Since V1.0
 */
@Component
@Mapper
public interface StLocalMapper {
    int insertPptnOne(StPptn pptn);
    int insertRiverOne(StRiver river);
    int insertRsvrOne(StRsvr rsvr);
}

StLocalMapper

<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper PUBLIC "-////DTD Mapper 3.0//EN" "http:///dtd/mybatis-3-mapper.dtd">
<mapper namespace="cn.javabb.bootdemo.mapper.local.StLocalMapper">
    <insert id="insertBatch" parameterType="java.util.List">
        insert into ST_PPTN_R_copy1(stcd,tm,drp,intv,pdr,dyp,wth,tmdown)
        values
        <foreach collection="list" item="item" index="index" separator=",">
        (#{item.stcd},#{item.tm},#{item.drp},#{item.intv},#{item.pdr},#{item.dyp},#{item.wth},#{item.tmdown})
        </foreach>
    </insert>
    <insert id="insertPptnOne" parameterType="cn.javabb.bootdemo.entity.StPptn">
    insert into ST_pptn_R(stcd,tm,drp,intv,pdr,dyp,wth,tmdown)
    values (#{stcd},#{tm},#{drp},#{intv},#{pdr},#{dyp},#{wth},#{tmdown})
    </insert>
    <insert id="insertRiverOne" parameterType="cn.javabb.bootdemo.entity.StRiver">
    insert into ST_river_R(STCD,TM,Z,Q,XSA,XSAVV,XSMXV,FLWCHRCD,WPTN,MSQMT,MSAMT,MSVMT,TMDOWN)
    values (#{STCD},#{TM},#{Z},#{Q},#{XSA},#{XSAVV},#{XSMXV},#{FLWCHRCD},#{WPTN},#{MSQMT},#{MSAMT},#{MSVMT},#{TMDOWN})
    </insert>
    <insert id="insertRsvrOne" parameterType="cn.javabb.bootdemo.entity.StRsvr">
    insert into ST_rsvr_R(STCD,TM,RZ,INQ,W,BLRZ,OTQ,RWCHRCD,RWPTN,INQDR,MSQMT,TMDOWN)
    values (#{STCD},#{TM},#{RZ},#{INQ},#{W},#{BLRZ},#{OTQ},#{RWCHRCD},#{RWPTN},#{INQDR},#{MSQMT},#{TMDOWN})
    </insert>
</mapper>

编写一个service调用方法就行了

@Slf4j
@Service
public class StDataSynService {
    @Autowired
    StLocalMapper stLocalMapper;
    @Async
    public void pptnSyn(){
        String nowDate = DateFormatUtils.format(new Date(),"yyyy-MM-dd HH:mm:ss");
        String yesterDay = DateFormatUtils.format(DateUtils.addDays(new Date(),-day),"yyyy-MM-dd HH:mm:ss");
        log.info("[{}]扫描Pptn同步数据....",nowDate);
        //需要同步的数据,往前推一个小时
        List<StPptn> remoteList = stRemoteMapper.listPptnRemote(nowDate,yesterDay);
        int total = 0;
        if(null!=remoteList && remoteList.size()>0){
            total = remoteList.size();
            log.info("获取Pptn数据：{}条。",total);
            Long startTM = System.currentTimeMillis();
            //执行同步
            int n=0;
            for(StPptn pptn:remoteList){
                n += stLocalMapper.insertPptnOne(pptn);
            }
            if(n>0){
                if(n == total){
                    log.info("Pptn同步成功.用时：{}秒,需要同步记录:{},成功:{}",(System.currentTimeMillis()-startTM)/1000,total,n);
                }else{
                    log.info("Pptn同步成功.用时：{}秒,需要同步记录:{},成功:{},失败:{}",(System.currentTimeMillis()-startTM)/1000,total,n,(total-n));
                }
            }
        }else{
            log.info("Pptn无数据同步");
        }
    }
}