Spring Boot 3整合ShardingSphere-Jdbc分库分表实战

<dependencies>
    <!-- ShardingSphere JDBC 核心依赖 -->
     <dependency>
         <groupId>org.apache.shardingsphere</groupId>
         <artifactId>shardingsphere-jdbc</artifactId>
         <version>5.5.2</version>
     </dependency>

    <!-- Druid 数据源 -->
    <dependency>
        <groupId>com.alibaba</groupId>
        <artifactId>druid-spring-boot-3-starter</artifactId>
        <version>1.2.27</version>
    </dependency>

    <!-- MySQL 驱动 -->
    <dependency>
        <groupId>mysql</groupId>
        <artifactId>mysql-connector-j</artifactId>
        <version>8.4.0</version>
    </dependency>
    
     <!-- springboot -->
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-web</artifactId>
        <version>3.5.5</version>
    </dependency>
</dependencies>

server:
  port: 9090

spring:
  datasource:
    driver-class-name: org.apache.shardingsphere.driver.ShardingSphereDriver # 指定了ShardingSphereDriver 为数据库驱动
    url: jdbc:shardingsphere:classpath:sharding-datasource.yml # 指定classpath的sharding-datasource.yml文件

mode:
  type: Standalone       # ShardingSphere 运行模式，Standalone 表示单机模式，不是集群模式
  repository:
    type: MEMORY         # 元数据存储类型，MEMORY 表示元数据存储在内存中（也可用 JDBC 存储）
dataSources:
  master:                # master名字可以自定义 数据库主库（普通表 写库）
    dataSourceClassName: com.alibaba.druid.pool.DruidDataSource  # 使用 Druid 作为真实数据源
    driverClassName: com.mysql.cj.jdbc.Driver                   # MySQL 驱动
    url: jdbc:mysql://localhost:3306/tsuki-chat?useSSL=false&serverTimezone=Asia/Shanghai&allowPublicKeyRetrieval=true&nullCatalogMeansCurrent=true&rewriteBatchedStatements=true
    username: root
    password: 123456
  slave:                 #slave名字可以自定义 数据库从库（普通表 从库）
    dataSourceClassName: com.alibaba.druid.pool.DruidDataSource
    driverClassName: com.mysql.cj.jdbc.Driver
    url: jdbc:mysql://localhost:3307/tsuki-chat?useSSL=false&serverTimezone=Asia/Shanghai&allowPublicKeyRetrieval=true&nullCatalogMeansCurrent=true&rewriteBatchedStatements=true
    username: root
    password: 123456
  users_master:          # users表数据库主库（用于 users 表分库分表写入操作）
    dataSourceClassName: com.alibaba.druid.pool.DruidDataSource
    driverClassName: com.mysql.cj.jdbc.Driver
    url: jdbc:mysql://localhost:3306/tsuki-users?useSSL=false&serverTimezone=Asia/Shanghai&allowPublicKeyRetrieval=true&nullCatalogMeansCurrent=true&rewriteBatchedStatements=true
    username: root
    password: 123456

  users_slave:          # users表数据库从库（用于 users 表分库分表读取操作）
    dataSourceClassName: com.alibaba.druid.pool.DruidDataSource
    driverClassName: com.mysql.cj.jdbc.Driver
    url: jdbc:mysql://localhost:3307/tsuki-users?useSSL=false&serverTimezone=Asia/Shanghai&allowPublicKeyRetrieval=true&nullCatalogMeansCurrent=true&rewriteBatchedStatements=true
    username: root
    password: 123456

rules:
  # 1️⃣ 全局表（除了 users 表）读写分离规则
  - !READWRITE_SPLITTING
    dataSourceGroups: # 必须用dataSourceGroups，用官网的dataSources会报错
      readwrite_ds:                     # 逻辑数据源名（可自定义）
        writeDataSourceName: master     # 写库
        readDataSourceNames:            # 读库集合
          - slave
        transactionalReadQueryStrategy: PRIMARY  # 事务读策略，PRIMARY 表示事务中读主库
        loadBalancerName: random        # 读库负载均衡策略
    loadBalancers:
      random:                           # 负载均衡算法配置
        type: RANDOM                    # 随机选择读库

  # 2️⃣ 针对users 表读写分离规则
  - !READWRITE_SPLITTING
    dataSourceGroups:
      users_group:                      # users 表逻辑数据源名（可自定义）
        writeDataSourceName: users_master  # 写库
        readDataSourceNames:
          - users_slave # 读库
        transactionalReadQueryStrategy: PRIMARY  # 事务读策略，PRIMARY 表示事务中读主库
        loadBalancerName: random        # 读库负载均衡策略
    loadBalancers:
      random:
        type: RANDOM                    # 随机负载均衡

  # 3️⃣ users 表分表规则
  - !SHARDING
    tables:
      users:
        actualDataNodes: users_group.users_${0..1}   # 物理表名列表，users_0 和 users_1
        tableStrategy:
          standard:
            shardingColumn: id                 # 分表字段
            shardingAlgorithmName: user_algo   # 使用分片算法 user_algo
    shardingAlgorithms:
      user_algo:
        type: INLINE
        props:
          algorithm-expression: users_${id % 2}  # 简单取模算法，将 id % 2 决定插入到 users_0 或 users_1

  # 4️⃣ 其他非分片表（比如 Test、Message 等）走默认读写分离数据源
  - !SINGLE
    tables:
      - "*.*"                             # 匹配所有其他表
    defaultDataSource: readwrite_ds       # 使用逻辑数据源 readwrite_ds（即 tsuki-chat 的读写分离）

props:
  sql-show: true                         # 打印 SQL，便于调试
  check-table: false                      # 不检查表是否存在，防止启动报错
  default-data-source-name: readwrite_ds  # 默认逻辑数据源

    @Test
    public void test3() {
        DbTest dbTest = new DbTest(1);
        testMapper.insert(dbTest);
    }

    @Test
    public void test4() {
        DbTest dbTest = testMapper.selectById(114514);
        log.info(dbTest.toString());
    }

    @Test
    public void test() {
        List<User> users = generateUsers(10); // 生成10条users数据
        List<BatchResult> insert = userMapper.insert(users);
        log.info("插入成功");
    }

    @Test
    public void test2() {
        Page<User> userPage = new Page<User>(1, 8);
        Page<User> selectedPage = userMapper.selectPage(userPage, null);
        selectedPage.getRecords().forEach(user -> log.info(String.valueOf(user.getId())));
    }