Mybatis自定义插件生成雪花ID做为主键项目
先附上项目项目GitHub地址 spring-boot-mybatis-interceptor
有关Mybatis雪花ID主键插件前面写了两篇博客作为该项目落地的铺垫。
该插件项目可以直接运用于实际开发中，作为分布式数据库表主键ID使用。
一、项目概述
1、项目背景
在生成表主键ID时，我们可以考虑主键自增 或者 UUID,但它们都有很明显的缺点
主键自增：1、自增ID容易被爬虫遍历数据。2、分表分库会有ID冲突。
UUID: 1、太长，并且有索引碎片，索引多占用空间的问题 2、无序。
雪花算法就很适合在分布式场景下生成唯一ID,它既可以保证唯一又可以排序,该插件项目的原理是
通过拦截器拦截Mybatis的insert语句,通过自定义注解获取到主键，并为该主键赋值雪花ID,插入数据库中。
2、技术架构
项目总体技术选型
SpringBoot2.1.7 + Mybatis + Maven3.5.4 + Mysql + lombok(插件)
3、使用方式
在你需要做为主键的属性上添加@AutoId注解,那么通过插件可以自动为该属性赋值主键ID。
public class TabUser {
/**
* id(添加自定义注解)
*/
@AutoId
private Long id;
/**
* 姓名
*/
private String name;
//其它属性 包括get，set方法
}
4、项目测试
配置好数据库连接信息,直接启动Springboot启动类Application.java,访问localhost:8080/save-foreach-user就可以看到数据库数据已经有雪花ID了。
如图
二、项目代码说明
在正式环境中只要涉及到插入数据的操作都被该插件拦截,并发量会很大。所以该插件代码即要保证线程安全又要保证高可用。所以在代码设计上做一些说明。
1、线程安全
这里的线程安全主要是考虑产生雪花ID的时候必须是线程安全的，不能出现同一台服务器同一时刻出现了相同的雪花ID,这里是通过
静态内部类单例模式 + synchronized
来保证线程安全的，具体有关生成雪花ID的代码这里就不粘贴。
2、高可用
我们去思考消耗性能比较大的地方可能出要出现在两个地方
1）雪花算法生成雪花ID的过程。
2）通过类的反射机制找到哪些属性带有@AutoId注解的过程。
第一点
其实在静态内部类实现雪花算法这篇博客已经简单测试过，生成20万条数据，大约在1.7秒能满足实际开发中我们的需要。
第二点
这里是有比较好的解决方案的，可以通过两点去改善它。
1)、在插件中添加了一个Map处理器
/**
* key值为Class对象 value可以理解成是该类带有AutoId注解的属性，只不过对属性封装了一层。
* 它是非常能够提高性能的处理器 它的作用就是不用每一次一个对象经来都要看下它的哪些属性带有AutoId注解
* 毕竟类的反射在性能上并不友好。只要key包含该Class,那么下次同样的class进来，就不需要检查它哪些属性带AutoId注解。
*/
private Map<Class, List<Handler>> handlerMap = new ConcurrentHashMap<>();
插件部分源码
public class AutoIdInterceptor implements Interceptor {
/**
* Map处理器
*/
private Map<Class, List<Handler>> handlerMap = new ConcurrentHashMap<>();
/**
* 某某方法
*/
private void process(Object object) throws Throwable {
Class handlerKey = object.getClass();
List<Handler> handlerList = handlerMap.get(handlerKey);
//先判断handlerMap是否已存在该class，不存在先找到该class有哪些属性带有@AutoId
if (handlerList == null) {
handlerMap.put(handlerKey, handlerList = new ArrayList<>());
// 通过反射 获取带有AutoId注解的所有属性字段,并放入到handlerMap中
}
//为带有@AutoId赋值ID
for (Handler handler : handlerList) {
handler.accept(object);
}
}
}
2）添加break label(标签)
这个就比较细节了,因为上面的process方法不是线程安全的，也就是说可能存在同一时刻有N个线程进入process方法，那么这里可以优化如下：
//添加了SYNC标签
SYNC:
if (handlerList == null) {
//此时handlerList确实为null,进入这里
synchronized (this) {
handlerList = handlerMap.get(handlerKey);
//但到这里发现它已经不是为null了，因为可能被其它线程往map中插入数据,那说明其实不需要在执行下面的逻辑了，直接跳出if体的SYNC标签位置。
//那么也就不会执行 if (handlerList == null) {}里面的逻辑。
if (handlerList != null) {
break SYNC;
}
}
}
这里虽然很细节，但也是有必要的，毕竟这里并发量很大,这样设计能一定程度提升性能。
我相信，无论今后的道路多么坎坷，只要抓住今天，迟早会在奋斗中尝到人生的甘甜。抓住人生中的一分一秒，胜过虚度中的一月一年！(7）