在 MP框架中优雅实现 MySQL Sequence 主键策略

本文链接：https://blog.youkuaiyun.com/m0_46109609/article/details/127891031

本文介绍了如何创建和管理数据库序列，包括DROPTABLE、CREATE TABLE用于序列管理的SQL语句，以及currval、nextval和setval三个函数的定义，用于获取当前值、下一个值和设置当前值。这些函数在数据库操作中常用于自增ID的生成和管理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

前言

在项目中，我们通常会利用Mysql的自增主键来确保表数据的唯一性。但是随着数据量的增多，需要对数据做水平分表，这种利用自增主键的方式就存在不足之处了。现在比较好的分布式id策略有雪花算法、Redis生成ID，以及利用数据库生成相同业务的Sequence主键。我这里给大家分享一种，基于sql函数生成Sequence主键的写法，并且可以很方便的集成到MybatisPlus框架中。

步骤

首先在数据库执行sequence表和函数的创建

-- Sequence 管理表
DROP TABLE IF EXISTS sequence;
CREATE TABLE sequence (
      name VARCHAR(50) NOT NULL COMMENT '序列名',
      current_value BIGINT NOT NULL COMMENT '当前序列值',
      increment INT NOT NULL DEFAULT 1 COMMENT '步长',
      PRIMARY KEY (name)
) ENGINE=InnoDB;

-- 取当前值的函数
DROP FUNCTION IF EXISTS currval;
DELIMITER $
CREATE FUNCTION currval (seq_name VARCHAR(50))
    RETURNS BIGINT
    LANGUAGE SQL
    DETERMINISTIC
    CONTAINS SQL
    SQL SECURITY DEFINER
    COMMENT '取当前的sequence'
BEGIN
    DECLARE value BIGINT;
    SET value = 0;
    SELECT current_value INTO value
    FROM sequence
    WHERE name = seq_name;
    RETURN value;
END
$
DELIMITER ;

-- 取下一个值的函数
DROP FUNCTION IF EXISTS nextval;
DELIMITER $
CREATE FUNCTION nextval (seq_name VARCHAR(50))
    RETURNS BIGINT
    LANGUAGE SQL
    DETERMINISTIC
    CONTAINS SQL
    SQL SECURITY DEFINER
    COMMENT '取下一个sequence'
BEGIN
    UPDATE sequence
    SET current_value = current_value + increment
    WHERE name = seq_name;
    RETURN currval(seq_name);
END
$
DELIMITER ;

-- 更新当前值的函数
DROP FUNCTION IF EXISTS setval;
DELIMITER $
CREATE FUNCTION setval (seq_name VARCHAR(50), value BIGINT)
    RETURNS BIGINT
    LANGUAGE SQL
    DETERMINISTIC
    CONTAINS SQL
    SQL SECURITY DEFINER
    COMMENT '更新当前sequence'
BEGIN
    UPDATE sequence
    SET current_value = value
    WHERE name = seq_name;
    RETURN currval(seq_name);
END
$
DELIMITER ;

插入业务Key，并且设置初始值

INSERT INTO sequence VALUES ('DOC_CHANNEL_SEQUENCE', 0, 1);

配置mp的相关配置

    @Bean
    public SequenceKeyGenerator sequenceKeyGenerator() {
        return new SequenceKeyGenerator();
    }

    public static class SequenceKeyGenerator implements IKeyGenerator {

        @Override
        public String executeSql(String incrementerName) {
            return "SELECT NEXTVAL('" + incrementerName + "')";
        }

        @Override
        public DbType dbType() {
            return DbType.MYSQL;
        }
    }