MyBatis與Spring的讀寫分離實現

MyBatis 是一個優秀的持久層框架，它支持定制化 SQL、存儲過程以及高級映射。而 Spring 提供了強大的依賴注入和事務管理功能。將 MyBatis 與 Spring 結合使用，可以實現讀寫分離，提高系統的性能和可擴展性。下面是一個簡單的實現步驟：

1. 配置數據源

首先，需要配置兩個數據源，一個用于讀操作，一個用于寫操作。可以使用 Spring 的 DataSource 配置類來實現。

@Configuration
public class DataSourceConfig {

    @Bean
    @ConfigurationProperties(prefix = "spring.datasource.read")
    public DataSource readDataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean
    @ConfigurationProperties(prefix = "spring.datasource.write")
    public DataSource writeDataSource() {
        return DataSourceBuilder.create().build();
    }
}

在 application.yml 或 application.properties 中配置兩個數據源：

spring:
  datasource:
    read:
      url: jdbc:mysql://localhost:3306/db_read
      username: user_read
      password: password_read
    write:
      url: jdbc:mysql://localhost:3306/db_write
      username: user_write
      password: password_write

2. 配置 MyBatis

接下來，配置 MyBatis 使用不同的數據源進行讀寫操作。

@Configuration
public class MyBatisConfig {

    @Bean
    public SqlSessionFactory readSqlSessionFactory() throws Exception {
        SqlSessionFactoryBean factoryBean = new SqlSessionFactoryBean();
        factoryBean.setDataSource(readDataSource());
        return factoryBean.getObject();
    }

    @Bean
    public SqlSessionFactory writeSqlSessionFactory() throws Exception {
        SqlSessionFactoryBean factoryBean = new SqlSessionFactoryBean();
        factoryBean.setDataSource(writeDataSource());
        return factoryBean.getObject();
    }
}

3. 配置 MyBatis 的 Mapper

為讀操作和寫操作分別配置 Mapper。

@Mapper
public interface ReadMapper {
    @Select("SELECT * FROM table_read")
    List<Map<String, Object>> selectAll();
}

@Mapper
public interface WriteMapper {
    @Insert("INSERT INTO table_write (column1, column2) VALUES (#{column1}, #{column2})")
    int insert(Map<String, Object> record);
}

4. 使用 Spring 的依賴注入

在 Service 層使用 Spring 的依賴注入來獲取 Mapper 實例，并執行讀寫操作。

@Service
public class DataService {

    @Autowired
    private ReadMapper readMapper;

    @Autowired
    private WriteMapper writeMapper;

    public List<Map<String, Object>> getAll() {
        return readMapper.selectAll();
    }

    public int insert(Map<String, Object> record) {
        return writeMapper.insert(record);
    }
}

5. 配置事務管理器

為了確保寫操作的原子性，需要配置事務管理器。

@Configuration
@EnableTransactionManagement
public class TransactionConfig {

    @Autowired
    private PlatformTransactionManager writeTransactionManager;

    @Bean
    public PlatformTransactionManager transactionManager() {
        return writeTransactionManager;
    }
}

6. 測試讀寫分離

最后，編寫一個簡單的測試類來驗證讀寫分離是否生效。

@SpringBootTest
public class DataServiceTest {

    @Autowired
    private DataService dataService;

    @Test
    public void testGetAll() {
        List<Map<String, Object>> result = dataService.getAll();
        System.out.println(result);
    }

    @Test
    @Transactional
    public void testInsert() {
        Map<String, Object> record = new HashMap<>();
        record.put("column1", "value1");
        record.put("column2", "value2");
        int count = dataService.insert(record);
        System.out.println("Inserted record count: " + count);
    }
}

通過以上步驟，你就可以實現 MyBatis 與 Spring 的讀寫分離。讀操作會使用配置的讀數據源，而寫操作會使用配置的寫數據源。這樣可以有效地提高系統的性能和可擴展性。