SpringBoot集成Sharding-JDBC实现分库分表

本文已收录于专栏《中间件合集》

版本介绍

SpringBoot的版本是： 2.3.12
ShardingSphere-jdbc的版本是：5.1.2
Mysql-Connector的版本是：8.0.19
Druid的版本是：1.2.8

背景介绍

分库分表是数据库设计中的一种常见策略，主要用于解决在高并发、大数据量场景下的性能瓶颈和管理问题。本文章采用Sharding-JDBC以水平分配的方式来实现分库分表。
ShardingSphere官网：https://shardingsphere.apache.org/document/5.1.2/cn/overview/
分库分表的原因：

性能提升：单一数据库在高并发情况下容易发生性能瓶颈，分库分表可以将请求分散到多个数据库或表上，降低单个数据库的压力。
容量管理：随着数据量的增长，单个数据库的存储能力可能不足。分库分表可以将数据分散到多个数据库中，以支撑更大的数据存储需求。
效率提升：在查询时，分表可以将查询范围缩小，只需要访问更少的数据，从而提高查询效率。
可维护性：分库分表可以更方便地进行数据备份、恢复和维护，提高系统的整体可维护性。
横向扩展：分库分表支持横向扩展，允许在现有架构中增加更多的数据库服务器，以适应业务增长。

拆分方式

1. 分库
垂直分库：将不同功能模块或业务逻辑的数据存储在不同的数据库中。例如，用户数据和订单数据分别存放在不同的数据库中。
水平分库：类似于分表，但是将同一类型的数据分散到不同的数据库中。例如，将用户数据按ID范围分散到不同的数据库。
2. 分表
按范围分表：根据某个字段（如时间、ID等）的范围，将数据划分到不同的表中，每个表存储特定范围的数据。
按哈希分表：对某个字段（如用户ID）进行哈希运算，根据哈希值将数据分配到不同的表中。

集成并测试

1.引入依赖

<dependency><groupId>com.alibaba</groupId><artifactId>druid-spring-boot-starter</artifactId><version>1.2.8</version></dependency><!-- Database Driver --><dependency><groupId>mysql</groupId><artifactId>mysql-connector-java</artifactId><version>8.0.19</version></dependency><dependency><groupId>org.projectlombok</groupId><artifactId>lombok</artifactId></dependency><dependency><groupId>org.apache.shardingsphere</groupId><artifactId>shardingsphere-jdbc-core-spring-boot-starter</artifactId><version>5.1.2</version></dependency>

2.创建库和表

在这里插入图片描述
在每个数据库中都需要创建多个相同的表，例如order_info_0、order_info_2、order_info_3等等，这样在配置分片规则的时候才会根据不同的计算方式映射到不同的库和表里面。

3.pom文件配置

server:port: 8081
spring:autoconfigure: # 排除druid 否则报错exclude: com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceAutoConfigureshardingsphere:# 开启sql打印enabled: trueprops:#　是否显示sqlsql-show: truedatasource:#数据源名称names: sharding0,sharding1#数据源实例： 如果这里还有mastersharding0:type: com.alibaba.druid.pool.DruidDataSourcedriver-class: com.mysql.cj.jdbc.Driver#使用Druid，不能使用jdbc-url 得使用urlurl: jdbc:mysql://localhost:3307/budget-lim-dev?useUnicode=true&characterEncoding=utf-8&useSSL=false&serverTimezone=Asia/Shanghaiusername: rootpassword: 123456sharding1:type: com.alibaba.druid.pool.DruidDataSourcedriver-class: com.mysql.cj.jdbc.Driverurl: jdbc:mysql://localhost:3306/budget-lim-dev?useUnicode=true&characterEncoding=utf-8&useSSL=false&serverTimezone=Asia/Shanghaiusername: rootpassword: 123456#分片规则rules:sharding:#对表进行分片tables:#逻辑表名，代表的是需要分表的名称order_info:#实际节点：这里代表的是 会使用sharding数据源中 order_info表 细分为0~3 4个表actual-data-nodes: sharding$->{0..1}.order_info_$->{0..1}#库策略database-strategy:standard:#根据num来进行分库sharding-column: numsharding-algorithm-name: alg_round-robin#表策略table-strategy:#标准表策略standard:#分表的列sharding-column: id#分片算法名称： 来源于下面的sharding-algorithmssharding-algorithm-name: alg_randomkey-generate-strategy: # 主键生成策略column: id  # 主键列key-generator-name: snowflake  # 策略算法名称(推荐使用雪花算法)#主键生成规则，SNOWFLAKE 雪花算法key-generators:snowflake:type: SNOWFLAKE#分片算法sharding-algorithms:alg_hash_mod:#类型：hash取余  类似于获取一个列的数type: HASH_MOD#分片的数量，因为是2个表，所以是2props:sharding-count: 2mybatis:#映射文件 配置之后，mybatis会去扫描该路径下的xml文件，才会与Mapper对应起来mapper-locations: classpath:mapper/*.xml#别名类(实体类)所在包type-aliases-package: com.wzil.simplesharding.entityconfiguration:#驼峰转换map-underscore-to-camel-case: true

3.编写测试类

Entity层

/*** @Author: wzil* @DATE: 2024年7月24日* @Description:**/
@Data
public class OrderInfo {private Long id;private String name;private Integer num;private Date createTime;
}

Mapper接口

/*** @Author: wzil* @DATE: 2024年7月24日15:26:30* @Description: 创建mapper接口，**/@Mapper
public interface OrderInfoMapper {List<OrderInfo> list();void save(OrderInfo orderInfo);void deleteById(@Param("id") Long id);void updateNameById(@Param("id") Long id, @Param("name") String name);OrderInfo getById(@Param("id") Long id);List<OrderInfo> limitOrder();List<OrderInfo> limitOrderWithOffset();}

MapperXML文件

<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<!--namespace：命名空间，用来映射对应的mapper
相当于将mapper和mapper.xml连接起来，这一步很重要-->
<mapper namespace="com.wzil.simplesharding.mapper.OrderInfoMapper"><insert id="save">insert into order_info(`name`,num,create_time) values(#{name},#{num},#{createTime})</insert><update id="updateNameById">update order_info set name=#{name}where id=#{id}</update><delete id="deleteById">delete from order_info where id=#{id}</delete><select id="list" resultType="com.wzil.simplesharding.entity.OrderInfo">select * from order_info</select><select id="getById" resultType="com.wzil.simplesharding.entity.OrderInfo">select * from order_infowhere id=#{id}</select><select id="limitOrder" resultType="com.wzil.simplesharding.entity.OrderInfo">select * from order_infoorder by num desclimit 10</select><select id="limitOrderWithOffset" resultType="com.wzil.simplesharding.entity.OrderInfo">select * from order_infoorder by num desclimit 2,10</select>
</mapper>

测试类

@Slf4j
@SpringBootTest
class SimpleShardingApplicationTests {@Autowiredprivate OrderInfoMapper orderInfoMapper;@Testvoid add() {for (int i = 0; i < 100; i++) {OrderInfo orderInfo = new OrderInfo();orderInfo.setName("wzil"+i);orderInfo.setCreateTime(new Date());orderInfo.setNum(i);orderInfoMapper.save(orderInfo);}}@Testvoid delete() {Long id=1022869460031111169L;orderInfoMapper.deleteById(id);}@Testvoid update() {Long id=1022869460031111169L;String name="hello";orderInfoMapper.updateNameById(id,name);}@Testvoid list() {List<OrderInfo> list = orderInfoMapper.list();System.out.println(list);}@Testvoid getById() {Long id=1022869460031111169L;OrderInfo info=orderInfoMapper.getById(id);System.out.println(info);}
}

4.运行结果

在这里插入图片描述
此实例运行的是更新的语句，可以看到是对sharding1数据库中的order_info_0进行更新操作的。

自定义分片规则

定义分片类

package com.wzil.simplesharding.config;import org.apache.shardingsphere.sharding.api.sharding.standard.PreciseShardingValue;
import org.apache.shardingsphere.sharding.api.sharding.standard.RangeShardingValue;
import org.apache.shardingsphere.sharding.api.sharding.standard.StandardShardingAlgorithm;import java.util.Collection;
import java.util.Properties;/*** @BelongsProject: simple_sharding* @BelongsPackage: com.wzil.simplesharding.config* @Author: Wuzilong* @Description: 轮询分片算法* @CreateTime: 2024-07-24 14:21* @Version: 1.0*/public class RoundRobinShardingAlgorithm implements StandardShardingAlgorithm<String> {int currentIndex=0;@Overridepublic String doSharding(Collection<String> collection, PreciseShardingValue<String> preciseShardingValue) {// 计算当前的索引位置currentIndex = (currentIndex + 1) % collection.size();// 根据当前索引返回数据源名称return (String) collection.toArray()[currentIndex];}@Overridepublic Collection<String> doSharding(Collection<String> collection, RangeShardingValue<String> rangeShardingValue) {return null;}@Overridepublic Properties getProps() {return null;}@Overridepublic void init(Properties properties) {}
}

编写一个分片规则的类去实现StandardShardingAlgorithm接口，去重写doSharding方法。doSharding(Collection collection, PreciseShardingValue preciseShardingValue)这个方法是确定值的分片规则。 doSharding(Collection collection, RangeShardingValue rangeShardingValue)这个方法是范围值的分片规则。根据业务的需求来重写不同的方法。

编写pom文件

        #分片算法sharding-algorithms:alg_round-robin:#指定了算法的类型type: CLASS_BASEDprops:#标准的分片策略strategy: standard#算法类的全限定名algorithmClassName: com.wzil.simplesharding.config.RoundRobinShardingAlgorithm

编写完之后就可以在上面配置的分库或者分表的规则去指定我们自己定义的alg_round-robin这个类型了。
在这里插入图片描述