Mybatis-plus进阶之分布式id生成

修改MyEmp

package com.yyoo.boot.mp.beans;import com.baomidou.mybatisplus.annotation.IdType;import com.baomidou.mybatisplus.annotation.TableId;import com.baomidou.mybatisplus.annotation.TableName;import lombok.Data;@Data@TableName("t_my_emp")public class MyEmp { @TableId(type=IdType.ASSIGN_ID) private Long id; private String name; private int age; private int sex;}

其实就是在我们的主键字段，id上添加了@TableId注解

@TableId注解IdType属性说明 值I说明AUTO数据库 ID 自增NONE无状态，该类型为未设置主键类型（注解里等于跟随全局，全局里约等于 INPUT）默认为NONEINPUTnsert 前自行 set 主键值ASSIGN_ID分配 ID(主键类型为 Number(Long 和 Integer)或 String)(since 3.3.0),使用接口IdentifierGenerator的方法nextIdASSIGN_UUID分配 UUID,主键类型为 String(since 3.3.0),使用接口IdentifierGenerator的方法nextUUID配置IdentifierGenerator自动id生成器

@Bean public GlobalConfig globalConfig() { GlobalConfig conf = new GlobalConfig(); conf.setIdentifierGenerator(new DefaultIdentifierGenerator()); return conf; }

配置SqlSessionFactory设置GlobalConfig配置

@Bean(name = "sessionFactory") public SqlSessionFactory sqlSessionFactory(@Qualifier("hikariDataSource") DataSource ds) throws Exception { log.info("数据源：{}，aliasesPackage：{}",ds,this.aliasesPackage); MybatisSqlSessionFactoryBean factoryBean = new MybatisSqlSessionFactoryBean(); factoryBean.setDataSource(ds); if(this.aliasesPackage != null && !"".equals(this.aliasesPackage)) { factoryBean.setTypeAliasesPackage(this.aliasesPackage); } //指定mapper xml目录 Assert.notEmpty(this.mapperLocations,"扫描的Mapper xml不能为空"); factoryBean.setMapperLocations(resolveMapperLocations(this.mapperLocations)); // 添加分页插件 factoryBean.setPlugins(mybatisPlusInterceptor()); // 添加id生成器的配置 factoryBean.setGlobalConfig(globalConfig()); return factoryBean.getObject(); }

示例代码

@Test public void t1(){ int num = 100; List list = new ArrayList<>(); for (int i = 0;i < num;i++) { Random random = new Random(); MyEmp myEmp = new MyEmp(); myEmp.setName(AutoNameUtil.autoSurAndName()); myEmp.setAge(random.nextInt(50) + 15);// 15岁及以上 myEmp.setSex(random.nextInt(2)); list.add(myEmp); } long start = System.currentTimeMillis(); myEmpService.saveBatch(list); long end = System.currentTimeMillis(); System.out.println("执行时间："+(end -start)); }

执行结果 IdentifierGenerator生成器详解

IdentifierGenerator是生成器的接口，DefaultIdentifierGenerator是其默认的实现，DefaultIdentifierGenerator的nextId方法默认是使用雪花算法实现的分布式id，其nextUUID方法模式是生成不带中划线的 UUID。

我们示例中的配置只是非常简单粗暴的DefaultIdentifierGenerator配置，DefaultIdentifierGenerator还有两个重载的构造方法，请大家仔细阅读源码，以及去多了解雪花算法来合理的进行配置。本文不做过多讲解。

自定义IdentifierGenerator生成器

我们自定义IdentifierGenerator，只需实现IdentifierGenerator接口，重写nextId方法即可。然后再替换上面代码DefaultIdentifierGenerator的配置即可。以下是官网示例：

public class CustomIdGenerator implements IdentifierGenerator { @Override public Long nextId(Object entity) { //可以将当前传入的class全类名来作为bizKey,或者提取参数来生成bizKey进行分布式Id调用生成. String bizKey = entity.getClass().getName(); //根据bizKey调用分布式ID生成 long id = ....; //返回生成的id值即可. return id; }}

问题说明

在我们前面的示例中MyEmp对象的id属性的类型为long，使用id生成器就报错，而且id生成器根本没有生成id，需要将它修改为包装类才能成功。

实际使用中我们的JavaBean对应的数字类型建议都尽量使用其包装类型来作为JavaBean的字段类型，因为非包装类是不能表示null的，特别有些可为空的字段，需要使用null这样的表示的时候。

上一篇：Mybatis-plus进阶之分页
下一篇：待续