SpringBoot 2.0 整合sharding-jdbc中间件实现数据分库分表

下面是详细的攻略。

1. 简介

sharding-jdbc 是一个基于 Java 开发的开源分布式中间件，可以提供数据分库、数据分表等功能。而SpringBoot 2.0 框架已经支持了 sharding-jdbc。

2. 步骤

2.1 添加 Maven 依赖

在 pom.xml 文件中添加 sharding-jdbc 的 Maven 依赖：

<!-- 引入 sharding-jdbc 依赖 -->
<dependency>
    <groupId>io.shardingsphere</groupId>
    <artifactId>sharding-jdbc-spring-boot-starter</artifactId>
    <version>4.1.1</version>
</dependency>

2.2 添加配置文件

在 application.yml 文件中添加 sharding-jdbc 的配置：

spring:
  shardingsphere:
    datasource:
      names: ds_0,ds_1
      ds_0:
        driver-class-name: com.mysql.cj.jdbc.Driver
        url: jdbc:mysql://localhost:3306/db_test_0?useUnicode=true&characterEncoding=UTF-8&useSSL=false
        username: root
        password: root
      ds_1:
        driver-class-name: com.mysql.cj.jdbc.Driver
        url: jdbc:mysql://localhost:3306/db_test_1?useUnicode=true&characterEncoding=UTF-8&useSSL=false
        username: root
        password: root
    sharding:
      tables:
        user_info:
          actualDataNodes: ds_${0..1}.user_info_${0..1}
          tableStrategy:
            inline:
              shardingColumn: user_id
              algorithmExpression: user_info_${user_id % 2}
          keyGenerateStrategy:
            column: user_id
            keyGeneratorName: snowflake

这里配置了两个数据源（ds_0 和 ds_1），以及一个分片表（user_info）。user_info 表根据 user_id 字段进行分片，分为 user_info_0 和 user_info_1 两个表，实际的表名在 actualDataNodes 中定义。

2.3 编写代码

在需要访问数据库的代码中，只需要注入 ShardingDataSource 即可使用分库分表的功能：

@Autowired
private DataSource dataSource;

// 使用 JdbcTemplate 进行查询
JdbcTemplate jdbcTemplate = new JdbcTemplate(dataSource);
List<Map<String, Object>> userList = jdbcTemplate.queryForList("SELECT * FROM user_info WHERE user_id=?", 1);

以上代码使用 JdbcTemplate 查询了一张分片表，并且只查询了 user_id = 1 的记录。

3. 示例

下面给出两个示例，一个是根据用户 ID 查询订单信息（分表），另一个是根据订单 ID 查询商品信息（分库）。

3.1 根据用户 ID 查询订单信息

订单表 order_info 分为 order_info_0 和 order_info_1 两个表，根据 user_id 字段进行分片。

@Autowired
private DataSource dataSource;

public List<Map<String, Object>> listOrderInfoByUserId(int userId) {
    // 使用 JdbcTemplate 进行查询
    JdbcTemplate jdbcTemplate = new JdbcTemplate(dataSource);
    String sql = "SELECT * FROM order_info WHERE user_id=?";
    List<Map<String, Object>> orderList = jdbcTemplate.queryForList(sql, userId);
    return orderList;
}

3.2 根据订单 ID 查询商品信息

订单库 order_db 中有 order_detail 和 order_goods 两个表，根据 order_id 字段进行分片。

@Autowired
private DataSource dataSource;

public List<Map<String, Object>> listGoodsInfoByOrderId(long orderId) {
    // 获取 OrderDetailMapper 实例
    SqlSession sqlSession = sqlSessionFactory.openSession();
    OrderDetailMapper mapper = sqlSession.getMapper(OrderDetailMapper.class);

    // 查询订单详情
    List<OrderDetail> detailList = mapper.selectByOrderId(orderId);

    // 查询订单商品信息（订单商品表在不同数据库中）
    List<Map<String, Object>> goodsList = new ArrayList<>();
    for (OrderDetail detail : detailList) {
        // 获取商品信息
        JdbcTemplate jdbcTemplate = new JdbcTemplate(dataSource);
        String sql = "SELECT * FROM order_goods WHERE id=?";
        Map<String, Object> goodsInfo = jdbcTemplate.queryForMap(sql, detail.getGoodsId());
        goodsList.add(goodsInfo);
    }
    return goodsList;
}

以上代码先通过 OrderDetailMapper 查询了订单详情，然后根据订单详情中的商品 ID 查询了商品信息。其中，订单详情表和订单商品表在不同的数据库中。