|
|
@@ -0,0 +1,162 @@
|
|
|
+package com.baomidou.mybatisplus.test.sharding;
|
|
|
+
|
|
|
+import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
|
|
|
+import com.baomidou.mybatisplus.extension.plugins.handler.sharding.ShardingProcessor;
|
|
|
+import com.baomidou.mybatisplus.extension.plugins.handler.sharding.ShardingRuleEnum;
|
|
|
+import com.baomidou.mybatisplus.extension.plugins.handler.sharding.ShardingStrategy;
|
|
|
+import com.baomidou.mybatisplus.extension.plugins.inner.ShardingInnerInterceptor;
|
|
|
+import com.baomidou.mybatisplus.test.BaseDbTest;
|
|
|
+import com.google.common.collect.Lists;
|
|
|
+import lombok.extern.slf4j.Slf4j;
|
|
|
+import org.apache.ibatis.plugin.Interceptor;
|
|
|
+import org.junit.jupiter.api.MethodOrderer;
|
|
|
+import org.junit.jupiter.api.Test;
|
|
|
+import org.junit.jupiter.api.TestMethodOrder;
|
|
|
+
|
|
|
+import java.time.LocalDateTime;
|
|
|
+import java.util.Collections;
|
|
|
+import java.util.List;
|
|
|
+import java.util.Map;
|
|
|
+
|
|
|
+/**
|
|
|
+ * @author zengzhihong
|
|
|
+ */
|
|
|
+@Slf4j
|
|
|
+@TestMethodOrder(MethodOrderer.OrderAnnotation.class)
|
|
|
+public class ShardingTest extends BaseDbTest<ShardingOrderMapper> {
|
|
|
+
|
|
|
+ private final LocalDateTime date1 = LocalDateTime.of(2021, 1, 1, 0, 0, 0);
|
|
|
+ private final LocalDateTime date2 = LocalDateTime.of(2021, 2, 1, 0, 0, 0);
|
|
|
+ private final LocalDateTime date3 = LocalDateTime.of(2021, 3, 1, 0, 0, 0);
|
|
|
+
|
|
|
+ @Test
|
|
|
+ void insertTest() {
|
|
|
+ ShardingOrder shardingOrder1 = new ShardingOrder().setOrderId(1L).setSubject("test1").setCreateTime(date1);
|
|
|
+ doTestAutoCommit(i -> {
|
|
|
+ i.insert(shardingOrder1);
|
|
|
+ i.insert1(2L);
|
|
|
+ });
|
|
|
+ }
|
|
|
+
|
|
|
+// @Test
|
|
|
+// void insertBatchTest() {
|
|
|
+// ShardingOrder shardingOrder1 =
|
|
|
+// new ShardingOrder().setOrderId(1L).setSubject("test1").setCreateTime(date1);
|
|
|
+// ShardingOrder shardingOrder2 =
|
|
|
+// new ShardingOrder().setOrderId(2L).setSubject("test2").setCreateTime(date2);
|
|
|
+// ShardingOrder shardingOrder3 =
|
|
|
+// new ShardingOrder().setOrderId(3L).setSubject("test3").setCreateTime(date3);
|
|
|
+// List<ShardingOrder> list = new ArrayList<>();
|
|
|
+// list.add(shardingOrder1);
|
|
|
+// list.add(shardingOrder2);
|
|
|
+// list.add(shardingOrder3);
|
|
|
+// String sqlStatement = SqlHelper.getSqlStatement(ShardingOrderMapper.class, SqlMethod.INSERT_ONE);
|
|
|
+// SqlHelper.executeBatch(ShardingOrder.class, log, list, 1000,
|
|
|
+// (sqlSession, entity) -> sqlSession.insert(sqlStatement, entity));
|
|
|
+//
|
|
|
+// }
|
|
|
+
|
|
|
+// @Test
|
|
|
+// void selectTest() {
|
|
|
+// insertBatchTest();
|
|
|
+// shardingOrderMapper.selectPage(new Page<>(),
|
|
|
+// Wrappers.<ShardingOrder>lambdaQuery().eq(ShardingOrder::getOrderId, 1L));
|
|
|
+//
|
|
|
+// Map<String, Object> params = new HashMap<>();
|
|
|
+// params.put("order_id", 1L);
|
|
|
+// params.put("create_time", date1);
|
|
|
+// shardingOrderMapper.selectByMap(params);
|
|
|
+//
|
|
|
+// shardingOrderMapper.selectById(1L);
|
|
|
+//
|
|
|
+// shardingOrderMapper.selectList(Wrappers.<ShardingOrder>lambdaQuery().eq(ShardingOrder::getCreateTime, date1)
|
|
|
+// .in(ShardingOrder::getOrderId, Arrays.asList(1L, 2L)));
|
|
|
+//
|
|
|
+// shardingOrderMapper.selectMaps(Wrappers.<ShardingOrder>lambdaQuery().in(ShardingOrder::getOrderId, 3L
|
|
|
+// ).eq(ShardingOrder::getCreateTime,
|
|
|
+// date1));
|
|
|
+//
|
|
|
+// shardingOrderMapper.selectBatchIds(Arrays.asList(1L, 2L));
|
|
|
+//
|
|
|
+// final Integer count =
|
|
|
+// shardingOrderMapper.selectCount(Wrappers.<ShardingOrder>lambdaQuery().eq(ShardingOrder::getOrderId, 1L
|
|
|
+// ).eq(ShardingOrder::getCreateTime,
|
|
|
+// date1));
|
|
|
+//
|
|
|
+// shardingOrderMapper.selectMapsPage(new Page<>(),
|
|
|
+// Wrappers.<ShardingOrder>lambdaQuery().eq(ShardingOrder::getOrderId, 1L));
|
|
|
+//
|
|
|
+// shardingOrderMapper.selectOne(Wrappers.<ShardingOrder>lambdaQuery().eq(ShardingOrder::getOrderId, 1L));
|
|
|
+//
|
|
|
+// shardingOrderMapper.selectPage(new Page<>(),
|
|
|
+// Wrappers.<ShardingOrder>lambdaQuery().eq(ShardingOrder::getOrderId, 1L));
|
|
|
+//
|
|
|
+// }
|
|
|
+//
|
|
|
+//
|
|
|
+// @Test
|
|
|
+// void updateTest() {
|
|
|
+// ShardingOrder order = new ShardingOrder();
|
|
|
+// order.setOrderId(1L);
|
|
|
+// order.setSubject("66");
|
|
|
+// shardingOrderMapper.update(order, Wrappers.<ShardingOrder>lambdaQuery().eq(ShardingOrder::getOrderId, 1L));
|
|
|
+// shardingOrderMapper.updateById(order);
|
|
|
+// }
|
|
|
+//
|
|
|
+// @Test
|
|
|
+// void deleteTest() {
|
|
|
+// shardingOrderMapper.delete(Wrappers.<ShardingOrder>lambdaQuery().eq(ShardingOrder::getOrderId, 1L));
|
|
|
+// shardingOrderMapper.deleteById(1L);
|
|
|
+// shardingOrderMapper.deleteBatchIds(Arrays.asList(1L, 2L));
|
|
|
+// Map<String, Object> params = new HashMap<>();
|
|
|
+// params.put("order_id", 1L);
|
|
|
+// shardingOrderMapper.deleteByMap(params);
|
|
|
+// }
|
|
|
+
|
|
|
+
|
|
|
+ @Override
|
|
|
+ protected List<Interceptor> interceptors() {
|
|
|
+ MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
|
|
|
+ final ShardingStrategy orderShardingHashStrategy = new ShardingStrategy("sharding_order",
|
|
|
+ "order_id", ShardingRuleEnum.ABSOLUTE, OrderShardingProcessor.class);
|
|
|
+ interceptor.addInnerInterceptor(new ShardingInnerInterceptor(orderShardingHashStrategy));
|
|
|
+ return Collections.singletonList(interceptor);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Override
|
|
|
+ protected List<String> tableSql() {
|
|
|
+ return Lists.newArrayList("CREATE TABLE IF NOT EXISTS sharding_order_01\n" +
|
|
|
+ " (\n" +
|
|
|
+ " order_id BIGINT(20) NOT NULL AUTO_INCREMENT,\n" +
|
|
|
+ " subject VARCHAR(30) NULL DEFAULT NULL,\n" +
|
|
|
+ " create_time DATETIME NULL,\n" +
|
|
|
+ " PRIMARY KEY (order_id)\n" +
|
|
|
+ " )",
|
|
|
+ "CREATE TABLE IF NOT EXISTS sharding_order_02\n" +
|
|
|
+ " (\n" +
|
|
|
+ " order_id BIGINT(20) NOT NULL AUTO_INCREMENT,\n" +
|
|
|
+ " subject VARCHAR(30) NULL DEFAULT NULL,\n" +
|
|
|
+ " create_time DATETIME NULL,\n" +
|
|
|
+ " PRIMARY KEY (order_id)\n" +
|
|
|
+ ")",
|
|
|
+ "CREATE TABLE IF NOT EXISTS sharding_order_03\n" +
|
|
|
+ " (\n" +
|
|
|
+ " order_id BIGINT(20) NOT NULL AUTO_INCREMENT,\n" +
|
|
|
+ " subject VARCHAR(30) NULL DEFAULT NULL,\n" +
|
|
|
+ " create_time DATETIME NULL,\n" +
|
|
|
+ " PRIMARY KEY (order_id)\n" +
|
|
|
+ ")");
|
|
|
+ }
|
|
|
+
|
|
|
+ static class OrderShardingProcessor implements ShardingProcessor {
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public String doSharding(ShardingStrategy strategy, Map<String, List<Object>> shardingValues) {
|
|
|
+ final List<Object> values = shardingValues.get("order_id");
|
|
|
+ // value是一个集合 比如 in查询
|
|
|
+ // 不管有几个value 此处最终return一个真实表名
|
|
|
+ // 未携带分配字段会报错 所以shardingValues 一定是 notEmpty
|
|
|
+ return "sharding_order_" + String.format("%02d", ((Long) values.get(0) % 3 + 1));
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|
miemie