MyBatis-Plus中的ASSIGN_ID生成问题及解决方案

id生成策略在分布式高并发环境下可能出现重复id的问题。MyBatis-Plus中对@TableId(type = IdType.ASSIGN_ID)生成默认使用com.baomidou.mybatisplus.core.toolkit.Sequence。然而,在该实现中,workid和datacenterid是根据网卡硬件地址生成的,而部署在Docker容器内的应用无法读取到宿主机的硬件地址。具体问题可参考 这里 。

雪花算法是Twitter开源的一种全局唯一ID生成算法,能够生成一个64位的整数ID。在雪花算法中,workerid是一个重要的参数,用于标识不同的机器。需要保证在不同的机器之间是唯一的,通常可以通过MAC地址或者IP地址来生成workerid。具体了解雪花算法可参考下图:

解决思路是,只要workid和datacenterid中的组合确保在所有节点中唯一就可以解决问题。下面选用了redis的自增值解决,当然也可以使用zookeeper、mysql等等,从已有的组件中选就好。具体实现代码如下:

/**
*  通过redis自增初始化的snowflake
*/
@ComPONEnt
public class SnowflakeUtil {

    private volatile Snowflake snowflake;

    @Resource
    private RedissonClient redissonClient;

    public long nextId() {
        return getSnowflake().nextId();
    }

    public String nextIdStr() {
        return String.valueOf(nextId());
    }

    public Snowflake getSnowflake() {

        if (snowflake == null) {
            synchronized (SnowflakeUtil.class) {
                if (snowflake == null) {
                    long workId = redissonClient.getAtomicLong("snowflake:worker").getAndIncrement();
                    snowflake = IdUtil.createSnowflake(workId % 32, 1);
                }
            }
        }

        return snowflake;
    }

}

/**
* 替换mybatis-plus ASSIGN_ID的生成器
*/
@Component
public class IdGenerator implements IdentifierGenerator {

    private final SnowflakeUtil snowflakeUtil;

    public IdGenerator(SnowflakeUtil snowflakeUtil) {
        this.snowflakeUtil = snowflakeUtil;
    }

    @Override
    public Number nextId(Object entity) {
        return snowflakeUtil.nextId();
    }
}

热门手游下载
下载排行榜