分库分表

阅读本指南前，请先阅读快速起步。本文档使用更复杂的场景进一步介绍Sharding-JDBC的分库分表能力。

数据库模式

本文档中提供了两个数据源db0和db1，每个数据源之中包含了两组表t_order_0和t_order_1，t_order_item_0和t_order_item_1 。这两组表的建表语句为：

  1. CREATE TABLE IF NOT EXISTS t_order_x (
  2.   order_id INT NOT NULL,
  3.   user_id  INT NOT NULL,
  4.   PRIMARY KEY (order_id)
  5. );
  6. CREATE TABLE IF NOT EXISTS t_order_item_x (
  7.   item_id  INT NOT NULL,
  8.   order_id INT NOT NULL,
  9.   user_id  INT NOT NULL,
  10.   PRIMARY KEY (item_id)
  11. );

逻辑表与实际表映射关系

均匀分布

数据表在每个数据源内呈现均匀分布的态势

  1. db0
  2.   ├── t_order_0 
  3.   └── t_order_1 
  4. db1
  5.   ├── t_order_0 
  6.   └── t_order_1

表规则可以使用默认的配置

  1.  TableRule orderTableRule = TableRule.builder("t_order").actualTables(Arrays.asList("t_order_0", "t_order_1")).dataSourceRule(dataSourceRule).build();

自定义分布

数据表呈现有特定规则的分布

  1. db0
  2.   ├── t_order_0 
  3.   └── t_order_1 
  4. db1
  5.   ├── t_order_2
  6.   ├── t_order_3
  7.   └── t_order_4

表规则可以指定每张表在数据源中的分布情况

  1.  TableRule orderTableRule = TableRule.builder("t_order").actualTables(Arrays.asList("db0.t_order_0", "db0.t_order_1", "db1.t_order_2", "db1.t_order_3", "db1.t_order_4")).dataSourceRule(dataSourceRule).build();

本教程采用的数据分布例子

  1. db0
  2.   ├── t_order_0               user_id为偶数   order_id为偶数
  3.   ├── t_order_1               user_id为偶数   order_id为奇数
  4.   ├── t_order_item_0          user_id为偶数   order_id为偶数
  5.   └── t_order_item_1          user_id为偶数   order_id为奇数
  6. db1
  7.   ├── t_order_0               user_id为奇数   order_id为偶数
  8.   ├── t_order_1               user_id为奇数   order_id为奇数
  9.   ├── t_order_item_0          user_id为奇数   order_id为偶数
  10.   └── t_order_item_1          user_id为奇数   order_id为奇数

逻辑表与实际表

配置分库分表的目的是将原有一张表的数据分散到不同库不同表中，且不改变原有SQL语句的情况下来使用这一张表。那么从一张表到多张的映射关系需要使用逻辑表与实际表这两种概念。下面通过一个例子来解释一下。假设在使用PreparedStatement访问数据库，SQL如下：

  1. select * from t_order where user_id = ? and order_id = ?;

当user_id=0且order=0时，Sharding-JDBC将会将SQL语句转换为如下形式：

  1. select * from db0.t_order_0 where user_id = ? and order_id = ?;

其中原始SQL中的t_order就是 逻辑表，而转换后的db0.t_order_0就是 实际表

规则配置

以上分库分表的形式Sharding-JDBC是通过规则配置来进行的描述的，下面讲通过几个小节来描述规则的详细配置：

  1.  ShardingRule shardingRule = ShardingRule.builder()
  2.         .dataSourceRule(dataSourceRule)
  3.         .tableRules(Arrays.asList(orderTableRule, orderItemTableRule))
  4.         .databaseShardingStrategy(new DatabaseShardingStrategy("user_id", new ModuloDatabaseShardingAlgorithm()))
  5.         .tableShardingStrategy(new TableShardingStrategy("order_id", new ModuloTableShardingAlgorithm()))
  6.         .build();

数据源配置

首先我们来构造DataSourceRule对象，它是来描述数据源的分布规则的。

  1.  DataSourceRule dataSourceRule = new DataSourceRule(dataSourceMap);

这里构造器需要一个入参：数据源名称与真实数据源之间的映射关系，这个关系的构造方法如下

  1. Map<String, DataSource> dataSourceMap = new HashMap<>(2);
  2. dataSourceMap.put("ds_0", createDataSource("ds_0"));
  3. dataSourceMap.put("ds_1", createDataSource("ds_1"));

真实的数据源可以使用任意一种数据库连接池，这里使用DBCP来举例

  1. private static DataSource createDataSource(final String dataSourceName) {
  2.     BasicDataSource result = new BasicDataSource();
  3.     result.setDriverClassName(com.mysql.jdbc.Driver.class.getName());
  4.     result.setUrl(String.format("jdbc:mysql://localhost:3306/%s", dataSourceName));
  5.     result.setUsername("root");
  6.     result.setPassword("");
  7.     return result;
  8. }

策略配置

数据源策略与表策略

策略类图Sharding-JDBC认为对于分片策略存有两种维度- 数据源分片策略DatabaseShardingStrategy：数据被分配的目标数据源- 表分片策略TableShardingStrategy：数据被分配的目标表，该目标表存在与该数据的目标数据源内。故表分片策略是依赖与数据源分片策略的结果的这里注意的是两种策略的API完全相同，以下针对策略API的讲解将适用于这两种策略

全局默认策略与特定表策略

策略是作用在特定的表规则上的，数据源策略与表策略与特定表相关

  1.  TableRule orderTableRule = TableRule.builder("t_order")
  2.          .actualTables(Arrays.asList("t_order_0", "t_order_1")
  3.          .dataSourceRule(dataSourceRule)
  4.          .databaseShardingStrategy(new DatabaseShardingStrategy("user_id", new ModuloDatabaseShardingAlgorithm()))
  5.          .tableShardingStrategy(new TableShardingStrategy("order_id", new ModuloTableShardingAlgorithm())))
  6.          .build();

如果分片规则中的所有表或大部分表的分片策略相同，可以使用默认策略来简化配置。以下两种配置是等价的:

  1.   //使用了默认策略配置
  2.   TableRule orderTableRule = TableRule.builder("t_order")
  3.           .actualTables(Arrays.asList("t_order_0", "t_order_1"))
  4.           .dataSourceRule(dataSourceRule)
  5.           .build();
  6.   TableRule orderItemTableRule = TableRule.builder("t_order_item")
  7.             .actualTables(Arrays.asList("t_order_item_0", "t_order_item_1"))
  8.             .dataSourceRule(dataSourceRule)
  9.             .build();
  10.   ShardingRule shardingRule = ShardingRule.builder()
  11.             .dataSourceRule(dataSourceRule)
  12.             .tableRules(Arrays.asList(orderTableRule, orderItemTableRule))
  13.             .databaseShardingStrategy(new DatabaseShardingStrategy("user_id", new ModuloDatabaseShardingAlgorithm()))
  14.             .tableShardingStrategy(new TableShardingStrategy("order_id", new ModuloTableShardingAlgorithm()))
  15.             .build();
  1.   //未使用默认策略配置
  2.   TableRule orderTableRule = TableRule.builder("t_order")
  3.           .actualTables(Arrays.asList("t_order_0", "t_order_1"))
  4.           .dataSourceRule(dataSourceRule)
  5.           .build();
  6.   TableRule orderItemTableRule = TableRule.builder("t_order_item")
  7.             .actualTables(Arrays.asList("t_order_item_0", "t_order_item_1"))
  8.             .dataSourceRule(dataSourceRule)
  9.             .databaseShardingStrategy(new DatabaseShardingStrategy("user_id", new ModuloDatabaseShardingAlgorithm()))
  10.             .tableShardingStrategy(new TableShardingStrategy("order_id", new ModuloTableShardingAlgorithm()))
  11.             .build();
  12.   ShardingRule shardingRule = ShardingRule.builder()
  13.             .dataSourceRule(dataSourceRule)
  14.             .tableRules(Arrays.asList(orderTableRule, orderItemTableRule))
  15.             .databaseShardingStrategy(new DatabaseShardingStrategy("user_id", new ModuloDatabaseShardingAlgorithm()))
  16.             .tableShardingStrategy(new TableShardingStrategy("order_id", new ModuloTableShardingAlgorithm()))
  17.             .build();

分片键

分片键是分片策略的第一个参数。分片键表示的是SQL语句中WHERE中的条件列。分片键可以配置多个

  • 单分片策略
  1. new TableShardingStrategy("order_id", new SingleKeyShardingAlgorithm()))
  • 多分片策略
  1. new TableShardingStrategy(Arrays.asList("order_id", "order_type", "order_date"), new MultiKeyShardingAlgorithm()))

分片算法

分片算法接口类图关系如下：

算法

绑定表

绑定表代表一组表，这组表的逻辑表与实际表之间的映射关系是相同的。比如t_order与t_order_item就是这样一组绑定表关系,它们的分库与分表策略是完全相同的,那么可以使用它们的表规则将它们配置成绑定表

  1. new BindingTableRule(Arrays.asList(orderTableRule, orderItemTableRule))

那么在进行SQL路由时，如果SQL为

  1. SELECT i.* FROM t_order o JOIN t_order_item i ON o.order_id=i.order_id WHERE o.user_id=? AND o.order_id=?

其中t_order在FROM的最左侧，Sharding-JDBC将会以它作为整个绑定表的主表。所有路由计算将会只使用主表的策略，那么t_order_item表的分片计算将会使用t_order的条件。故绑定表之间的分区键要完全相同。

分片算法详解

单分片键算法与多分片键算法

这两种算法从名字上就可以知道前者是针对只有一个分片键，后者是针对有多个分片键的。单分片键算法是多分片键算法的一种简便形式，所以完全可以使用多分片算法去替代单分片键算法。下面两种形式是等价的

  1. new TableShardingStrategy("order_id", new SingleKeyShardingAlgorithm()))
  2. new TableShardingStrategy(Arrays.asList("order_id"), new MultiKeyShardingAlgorithm()))

同时在算法内部，doSharding等方法的shardingValue入参根据使用算法类型不同而不同单分片键算法，方法签名

  1. public String doEqualSharding(final Collection<String> dataSourceNames, final ShardingValue<Integer> shardingValue)

多分片键算法，方法签名

  1. public Collection<String> doSharding(final Collection<String> availableTargetNames, final Collection<ShardingValue<?>> shardingValues)

分片键算法类型

根据数据源策略与表策略、单分片与多分片，这两种组合，一共产生了4种可供实现的分片算法的接口

  • 单分片键数据源分片算法SingleKeyDatabaseShardingAlgorithm
  • 单分片表分片算法SingleKeyTableShardingAlgorithm
  • 多分片键数据源分片算法MultipleKeyDatabaseShardingAlgorithm
  • 多分片表分片算法MultipleKeyTableShardingAlgorithm

单分片键算法

单分片键算法需要实现三个方法，下面以”单分片键数据源分片算法“举例

  1. @Override
  2. public String doEqualSharding(final Collection<String> availableTargetNames, final ShardingValue<Integer> shardingValue)
  4. @Override
  5. public Collection<String> doInSharding(final Collection<String> availableTargetNames, final ShardingValue<Integer> shardingValue)
  7. @Override
  8. public Collection<String> doBetweenSharding(final Collection<String> availableTargetNames, final ShardingValue<Integer> shardingValue)

这三种算法作用如下- doEqualSharding在WHERE使用=作为条件分片键。算法中使用shardingValue.getValue()获取等=后的值- doInSharding在WHERE使用IN作为条件分片键。算法中使用shardingValue.getValues()获取IN后的值- doBetweenSharding在WHERE使用BETWEEN作为条件分片键。算法中使用shardingValue.getValueRange()获取BETWEEN后的值

下面是一个余2的算法的例子，当分片键的值除以2余数就是实际表的结尾。注意注释中提供了一些算法生成SQL的结果，参数tableNames集合中有两个参数t_order_0和t_order_1

  1.  public final class ModuloTableShardingAlgorithm implements SingleKeyTableShardingAlgorithm<Integer> {
  3.     /**
  4.     *  select * from t_order from t_order where order_id = 11 
  5.     *          └── SELECT *  FROM t_order_1 WHERE order_id = 11
  6.     *  select * from t_order from t_order where order_id = 44
  7.     *          └── SELECT *  FROM t_order_0 WHERE order_id = 44
  8.     */
  9.     public String doEqualSharding(final Collection<String> tableNames, final ShardingValue<Integer> shardingValue) {
  10.         for (String each : tableNames) {
  11.             if (each.endsWith(shardingValue.getValue() % 2 + "")) {
  12.                 return each;
  13.             }
  14.         }
  15.         throw new IllegalArgumentException();
  16.     }
  18.     /**
  19.     *  select * from t_order from t_order where order_id in (11,44)  
  20.     *          ├── SELECT *  FROM t_order_0 WHERE order_id IN (11,44) 
  21.     *          └── SELECT *  FROM t_order_1 WHERE order_id IN (11,44) 
  22.     *  select * from t_order from t_order where order_id in (11,13,15)  
  23.     *          └── SELECT *  FROM t_order_1 WHERE order_id IN (11,13,15)  
  24.     *  select * from t_order from t_order where order_id in (22,24,26)  
  25.     *          └──SELECT *  FROM t_order_0 WHERE order_id IN (22,24,26) 
  26.     */
  27.     public Collection<String> doInSharding(final Collection<String> tableNames, final ShardingValue<Integer> shardingValue) {
  28.         Collection<String> result = new LinkedHashSet<>(tableNames.size());
  29.         for (Integer value : shardingValue.getValues()) {
  30.             for (String tableName : tableNames) {
  31.                 if (tableName.endsWith(value % 2 + "")) {
  32.                     result.add(tableName);
  33.                 }
  34.             }
  35.         }
  36.         return result;
  37.     }
  39.     /**
  40.     *  select * from t_order from t_order where order_id between 10 and 20 
  41.     *          ├── SELECT *  FROM t_order_0 WHERE order_id BETWEEN 10 AND 20 
  42.     *          └── SELECT *  FROM t_order_1 WHERE order_id BETWEEN 10 AND 20 
  43.     */
  44.     public Collection<String> doBetweenSharding(final Collection<String> tableNames, final ShardingValue<Integer> shardingValue) {
  45.         Collection<String> result = new LinkedHashSet<>(tableNames.size());
  46.         Range<Integer> range = (Range<Integer>) shardingValue.getValueRange();
  47.         for (Integer i = range.lowerEndpoint(); i <= range.upperEndpoint(); i++) {
  48.             for (String each : tableNames) {
  49.                 if (each.endsWith(i % 2 + "")) {
  50.                     result.add(each);
  51.                 }
  52.             }
  53.         }
  54.         return result;
  55.     }
  56. }

多分片键算法

多分片键试用于使用场景比较复杂，为了能提供更高的灵活性，故只提供实现一个方法。

  1. @Override
  2. public Collection<String> doSharding(final Collection<String> availableTargetNames, final Collection<ShardingValue<?>> shardingValues)

算法实现的时候根据shardingValue.getType()来获取条件是=，IN或者BETWEEN。然后根据业务进行灵活的实现。

如果表的数据分布如下

  1. db0
  2.   ├── t_order_00               user_id以a偶数   order_id为偶数
  3.   ├── t_order_01               user_id以a偶数   order_id为奇数
  4.   ├── t_order_10               user_id以b奇数   order_id为偶数
  5.   └── t_order_11               user_id以b奇数   order_id为奇数

算法实现如下:

  1. public final class MultipleKeysModuloTableShardingAlgorithm implements MultipleKeysTableShardingAlgorithm {
  3.     @Override
  4.     public Collection<String> doSharding(final Collection<String> availableTargetNames, final Collection<ShardingValue<?>> shardingValues) {
  5.         Set<Integer> orderIdValueSet = getShardingValue(shardingValues, "order_id");
  6.         Set<Integer> userIdValueSet = getShardingValue(shardingValues, "user_id");
  8.         List<String> result = new ArrayList<>();
  9.         /*
  10.         userIdValueSet[10,11] + orderIdValueSet[101,102] => valueResult[[10,101],[10,102],[11,101],[11,102]]
  11.          */
  12.         Set<List<Integer>> valueResult = Sets.cartesianProduct(userIdValueSet, orderIdValueSet);
  13.         for (List<Integer> value : valueResult) {
  14.             String suffix = Joiner.on("").join(value.get(0) % 2, value.get(1) % 2);
  15.             for (String tableName : availableTargetNames) {
  16.                 if (tableName.endsWith(suffix)) {
  17.                     result.add(tableName);
  18.                 }
  19.             }
  21.         }
  22.         return result;
  23.     }
  25.     private Set<Integer> getShardingValue(final Collection<ShardingValue<?>> shardingValues, final String shardingKey) {
  26.         Set<Integer> valueSet = new HashSet<>();
  27.         ShardingValue<Integer> shardingValue = null;
  28.         for (ShardingValue<?> each : shardingValues) {
  29.             if (each.getColumnName().equals(shardingKey)) {
  30.                 shardingValue = (ShardingValue<Integer>) each;
  31.                 break;
  32.             }
  33.         }
  34.         if (null == shardingValue) {
  35.             return valueSet;
  36.         }
  37.         switch (shardingValue.getType()) {
  38.             case SINGLE:
  39.                 valueSet.add(shardingValue.getValue());
  40.                 break;
  41.             case LIST:
  42.                 valueSet.addAll(shardingValue.getValues());
  43.                 break;
  44.             case RANGE:
  45.                 for (Integer i = shardingValue.getValueRange().lowerEndpoint(); i <= shardingValue.getValueRange().upperEndpoint(); i++) {
  46.                     valueSet.add(i);
  47.                 }
  48.                 break;
  49.             default:
  50.                 throw new UnsupportedOperationException();
  51.         }
  52.         return valueSet;
  53.     }
  54. }

构造ShardingDataSource

完成规则配置后，我们可以通过ShardingDataSourceFactory工厂得到ShardingDataSource

  1. DataSource dataSource = ShardingDataSourceFactory.createDataSource(shardingRule);

使用ShardingDataSource

通过一个例子来看看如何使用该数据源

  1. String sql = "SELECT i.* FROM t_order o JOIN t_order_item i ON o.order_id=i.order_id WHERE o.user_id=? AND o.order_id=?";
  2.         try (
  3.                 Connection conn = dataSource.getConnection();
  4.                 PreparedStatement preparedStatement = conn.prepareStatement(sql);
  5.                 ) {
  6.             preparedStatement.setInt(1, 10);
  7.             preparedStatement.setInt(2, 1001);
  8.             ResultSet rs = preparedStatement.executeQuery();
  9.             while (rs.next()) {
  10.                 System.out.println(rs.getInt(1));
  11.                 System.out.println(rs.getInt(2));
  12.                 System.out.println(rs.getInt(3));
  13.             }
  14.             rs.close();
  15.         }

该数据源与普通数据源完全相同，你可以通过上例的API形式来使用，也可以将其配置在Spring，Hibernate等框架中使用。

如果希望不依赖于表中的列传入分片键值，参考：强制路由