概 述

ID号生成器(或:全局唯一ID生成器)是服务端系统的基础设施,而且ID号这个东西基本搞后端开发的程序员天天都要接触。而关于ID生成的算法现在业界首屈一指的当属 Snowflake雪花算法。

UidGenerator正是百度开源的一款基于 Snowflake雪花算法实现的高性能唯一ID生成器。在本号前文中已经详细使用过 UidGenerator,但使用过程还是比较繁杂,还需要自己去引 UidGenerator组件的源码,感觉有点不方便。为此本文基于 UidGenerator,再来封装一套更利于 SpringBoot项目使用的 ID号生成组件,命名为 id-spring-boot-starter,一看名字就知道是开箱即用的。

用 法

  • 首先导入SQL脚本
DROP TABLE IF EXISTS WORKER_NODE;
CREATE TABLE WORKER_NODE
(
ID BIGINT NOT NULL AUTO_INCREMENT COMMENT 'auto increment id',
HOST_NAME VARCHAR(64) NOT NULL COMMENT 'host name',
PORT VARCHAR(64) NOT NULL COMMENT 'port',
TYPE INT NOT NULL COMMENT 'node type: ACTUAL or CONTAINER',
LAUNCH_DATE DATE NOT NULL COMMENT 'launch date',
MODIFIED TIMESTAMP NOT NULL COMMENT 'modified time',
CREATED TIMESTAMP NOT NULL COMMENT 'created time',
PRIMARY KEY(ID)
)
 COMMENT='DB WorkerID Assigner for UID Generator',ENGINE = INNODB;

这一步肯定是省不掉,毕竟 UidGenerator需要数据库支持

  • pom中加入依赖
  <dependency>
            <artifactId>id-spring-boot-starter</artifactId>
            <groupId>org.example</groupId>
            <version>1.0-SNAPSHOT</version>
        </dependency>
  • 配置数据库连接
server:
  port: 9999

#mybatis配置
mybatis:

  mapper-locations: classpath:mapper/*.xml
  configuration:
    map-underscore-to-camel-case: true
#Spring配置
spring:
  application:
    name: springcloud-provider-dept  #项目名字
  datasource:
    type: com.alibaba.druid.pool.DruidDataSource
    driver-class-name: com.mysql.jdbc.Driver
    username: root
    password: 123456
    url: jdbc:mysql://localhost:3306/db01?useUnicode=true&characterEncoding=utf-8&allowMultiQueries=true&useSSL=false&serverTimezone=Asia/Shanghai

还是因为 UidGenerator需要数据库支持

  • 修改Spring Boot主类
    Spring Boot应用主类上加上 mybatis注解即可:
@MapperScan("com.baidu.fsg.uid")
  • 代码使用方式
@RestController
public class UidController {

    @Autowired
    private UidGenService uidGenService;

    @GetMapping("/uidGenerator")
    public String UidGenerator() {
        return String.valueOf(uidGenService.getUid());
    }
}

怎么样,用法超级简单吧:

  1. 首先用 Autowired的方式引入 UidGenService类;
  2. 直接调用 UidGenService类的 getUid()方法即可获得一个 Long型的 ID

源码下载

需要组件 id-spring-boot-starter源码进行自定制的,可以去 github上自行下载源代码,地址在此:
https://gitee.com/inverse-star/test/tree/master/springboot-demo/springboot_uid_generator/id-spring-boot-starter

几个注意点

  1. 由于 UidGenerator需要数据库支持,因此使用前一定要导数据表,并且要配置 MapperScan
  2. 需要高度定制 UidGenerator组件详情的,可以自行修改 id-spring-boot-starter内部的 cached-uid-spring.xml文件,然后重新打 jar包
  3. 由于 ID号生成一般属于系统基础服务,因此可以独立成一个微服务,供其他微服务调用
Logo

欢迎加入 MCP 技术社区!与志同道合者携手前行,一同解锁 MCP 技术的无限可能!

更多推荐