springboot(七):springboot+mybatis多資料

dunne21發表於2021-09-09

springboot+mybatis多資料來源最簡解決方案


說起多資料來源,一般都來解決那些問題呢,主從模式或者業務比較複雜需要連線不同的分庫來支援業務。我們專案是後者的模式,網上找了很多,大都是根據jpa來做多資料來源解決方案,要不就是老的spring多資料來源解決方案,還有的是利用aop動態切換,感覺有點小複雜,其實我只是想找一個簡單的多資料支援而已,折騰了兩個小時整理出來,供大家參考。

廢話不多說直接上程式碼吧

配置檔案

pom包就不貼了比較簡單該依賴的就依賴,主要是資料庫這邊的配置:

mybatis.config-locations=classpath:mybatis/mybatis-config.xml

spring.datasource.test1.driverClassName = com.mysql.jdbc.Driver
spring.datasource.test1.url = jdbc:mysql://localhost:3306/test1?useUnicode=true&characterEncoding=utf-8
spring.datasource.test1.username = root
spring.datasource.test1.password = root

spring.datasource.test2.driverClassName = com.mysql.jdbc.Driver
spring.datasource.test2.url = jdbc:mysql://localhost:3306/test2?useUnicode=true&characterEncoding=utf-8
spring.datasource.test2.username = root
spring.datasource.test2.password = root

一個test1庫和一個test2庫,其中test1位主庫,在使用的過程中必須指定主庫,不然會報錯。

資料來源配置
@Configuration
@MapperScan(basePackages = "com.neo.mapper.test1", sqlSessionTemplateRef  = "test1SqlSessionTemplate")
public class DataSource1Config {

    @Bean(name = "test1DataSource")
    @ConfigurationProperties(prefix = "spring.datasource.test1")
    @Primary
    public DataSource testDataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean(name = "test1SqlSessionFactory")
    @Primary
    public SqlSessionFactory testSqlSessionFactory(@Qualifier("test1DataSource") DataSource dataSource) throws Exception {
        SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
        bean.setDataSource(dataSource);
        bean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath:mybatis/mapper/test1/*.xml"));
        return bean.getObject();
    }

    @Bean(name = "test1TransactionManager")
    @Primary
    public DataSourceTransactionManager testTransactionManager(@Qualifier("test1DataSource") DataSource dataSource) {
        return new DataSourceTransactionManager(dataSource);
    }

    @Bean(name = "test1SqlSessionTemplate")
    @Primary
    public SqlSessionTemplate testSqlSessionTemplate(@Qualifier("test1SqlSessionFactory") SqlSessionFactory sqlSessionFactory) throws Exception {
        return new SqlSessionTemplate(sqlSessionFactory);
    }

}

最關鍵的地方就是這塊了,一層一層注入,首先建立DataSource,然後建立SqlSessionFactory再建立事務,最後包裝到SqlSessionTemplate中。其中需要指定分庫的mapper檔案地址,以及分庫dao層程式碼

@MapperScan(basePackages = "com.neo.mapper.test1", sqlSessionTemplateRef  = "test1SqlSessionTemplate")

這塊的註解就是指明瞭掃描dao層,並且給dao層注入指定的SqlSessionTemplate。所有@Bean都需要按照命名指定正確。

dao層和xml層

dao層和xml需要按照庫來分在不同的目錄,比如:test1庫dao層在com.neo.mapper.test1包下,test2庫在com.neo.mapper.test1

public interface User1Mapper {

    List getAll();

    UserEntity getOne(Long id);

    void insert(UserEntity user);

    void update(UserEntity user);

    void delete(Long id);

}

xml層


        id, userName, passWord, user_sex, nick_name
    
       INSERT INTO 
            users
            (userName,passWord,user_sex) 
        VALUES
            (#{userName}, #{passWord}, #{userSex})
    
       UPDATE 
            users 
       SET 
        userName = #{userName},passWord = #{passWord},
        nick_name = #{nickName}
       WHERE 
            id = #{id}
    
       DELETE FROM
             users 
       WHERE 
             id =#{id}
    
測試

測試可以使用SpringBootTest,也可以放到Controller中,這裡只貼Controller層的使用

@RestController
public class UserController {

    @Autowired
    private User1Mapper user1Mapper;

    @Autowired
    private User2Mapper user2Mapper;

    @RequestMapping("/getUsers")
    public List getUsers() {
        List users=user1Mapper.getAll();
        return users;
    }

    @RequestMapping("/getUser")
    public UserEntity getUser(Long id) {
        UserEntity user=user2Mapper.getOne(id);
        return user;
    }

    @RequestMapping("/add")
    public void save(UserEntity user) {
        user2Mapper.insert(user);
    }

    @RequestMapping(value="update")
    public void update(UserEntity user) {
        user2Mapper.update(user);
    }

    @RequestMapping(value="/delete/{id}")
    public void delete(@PathVariable("id") Long id) {
        user1Mapper.delete(id);
    }

}


作者:純潔的微笑
版權所有,歡迎保留原文連結進行轉載:)

來自 “ ITPUB部落格 ” ,連結:http://blog.itpub.net/1600/viewspace-2808476/,如需轉載,請註明出處,否則將追究法律責任。

相關文章