SpringBoot Mybatis 分库分表 sharding

开源  https://github.com/letian-tang/mybatis-sharding.git

史上最简洁的分库分表中间件 基于SpringBoot+Mybatis的分库分表 基于Mapper的切面 数据源默认使用HikariDataSource @Component public class ShardingStrategyImpl implements ShardingStrategy { @Override public Long getShardingKey() { // 例子：从ThreadLocal中获取当前卖家id，通过卖家id sharding return SellerIdHolder.getSellerId(); } }

mybatis-db-sharding 分库分表 mybatis-db-example OrderServiceTest 单元测试

如果有事务，需要在开启事务之前确定数据源 那么就需要对@Transactional 和 TransactionTemplate.execute做拦截 private static final String EXPRESSION_TRANSACTIONAL = "@annotation(org.springframework.transaction.annotation.Transactional)"; private static final String EXPRESSION_TRANSACTIONAL_TEMPLATE = "execution(* org.springframework.transaction.support.TransactionTemplate.execute(..))"; private final String[] expressions = new String[] {EXPRESSION_TRANSACTIONAL, EXPRESSION_TRANSACTIONAL_TEMPLATE}; /** * 拦截器 * * @return defaultPointcutAdvisor */ @Bean public DefaultPointcutAdvisor defaultPointcutAdvisor() { List<String> expressionList = shardingProperties.getMapperPointcuts(); Assert.notEmpty(expressionList, "Mapper Pointcut is null"); DataSourceAdvice interceptor = new DataSourceAdvice(shardStrategy, shardingProperties); AspectJExpressionPointcut pointcut = new AspectJExpressionPointcut(); expressionList.addAll(Arrays.asList(expressions)); String expression = String.join(" || ", expressionList); pointcut.setExpression(expression); DefaultPointcutAdvisor advisor = new DefaultPointcutAdvisor(); advisor.setPointcut(pointcut); advisor.setAdvice(interceptor); return advisor; } #开启 mybatis.db.sharding.enabled=true #Mapper切面，多个 逗号分隔 如 execution(* com.tdy..*Mapper.*(..)),execution(* com.tdy..*Mapper.*(..)) mybatis.db.sharding.mapper-pointcuts=execution(* com.tdy..*Mapper.*(..)) #数据库1 mybatis.db.sharding.data-sources[0].jdbc-url=jdbc:mysql://localhost:3306/sophon_0?serverTimezone=Asia/Shanghai&allowMultiQueries=true&useSSL=false&zeroDateTimeBehavior=convertToNull&useAffectedRows=true mybatis.db.sharding.data-sources[0].username=username mybatis.db.sharding.data-sources[0].password=password mybatis.db.sharding.data-sources[0].maximum-pool-size=10 mybatis.db.sharding.data-sources[0].idle-timeout=10000 #数据库2 mybatis.db.sharding.data-sources[1].jdbc-url=jdbc:mysql://localhost:3306/sophon_1?serverTimezone=Asia/Shanghai&allowMultiQueries=true&useSSL=false&zeroDateTimeBehavior=convertToNull&useAffectedRows=true mybatis.db.sharding.data-sources[1].username=username mybatis.db.sharding.data-sources[1].password=password mybatis.db.sharding.data-sources[1].maximum-pool-size=3 mybatis.db.sharding.data-sources[1].idle-timeout=10000 #表名t_order，nums分表数量4，不配置的表不进行分表 mybatis.db.sharding.tables.t_order.nums=4

开源  https://github.com/letian-tang/mybatis-sharding.git

@Service public class OrderServiceImpl implements OrderService { @Autowired private TransactionTemplate transactionTemplate; @Autowired private OrderMapper orderMapper; /** * 单表支持 * * @param order */ @Override public void insert(Order order) { orderMapper.insert(order); } /** * @Transactional 注解事务支持 * @param orders */ @Override @Transactional public void insertBatch(List<Order> orders) { for (Order order : orders) { orderMapper.insert(order); } } /** * 事务模板 transactionTemplate 支持 * * @param orders */ @Override public void insertBatchTemplate(List<Order> orders) { transactionTemplate.execute(new TransactionCallbackWithoutResult() { @Override protected void doInTransactionWithoutResult(TransactionStatus status) { for (Order order : orders) { orderMapper.insert(order); } } }); } /** * 查询支持 * * @return */ @Override public List<Order> select() { return orderMapper.selectList(new LambdaQueryWrapper<Order>()); } }

单元测试

@RunWith(SpringRunner.class) @SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT, classes = ShardingApplication.class) public class OrderServiceTest { @Autowired private OrderService orderService; @Before public void init() { SellerIdHolder.setSellerId(12L); } @After public void after() { SellerIdHolder.clear(); } /** * 单表 */ @Test public void saveOrder() { orderService.insert(Order.builder().name("11111").build()); } /** * 事务方式 @Transactional */ @Test public void saveOrder2() { orderService.insertBatch(Lists.newArrayList(Order.builder().name("22222").build(), Order.builder().name("33333").build())); } /** * 事务 transactionTemplate方式 */ @Test public void saveOrder3() { orderService.insertBatchTemplate(Lists.newArrayList(Order.builder().name("44444").build(), Order.builder().name("55555").build())); } /** * 查询 */ @Test public void selectOrder() { orderService.select(); } /** * 测试 多条sql语句，分号;分隔 */ @Test public void saveOrder4() { orderService.insertBatchList(Lists.newArrayList(Order.builder().name("66666").build(), Order.builder().name("77777").build())); } }

开源  https://github.com/letian-tang/mybatis-sharding.git