mybatis-plus实现mysql自定义IKeyGenerator

分享心得

已于 2023-08-10 21:09:40 修改

阅读量1.1k

点赞数

文章标签： mybatis mysql 数据库

于 2023-08-10 11:54:50 首次发布

本文链接：https://blog.csdn.net/m0_37539286/article/details/132206338

版权

本文介绍了在MybatisPlus插件中，如何使用IKeyGenerator接口实现MySQL的主键生成，通过新建表`dual_sequence`模拟序列功能，以及MysqlKeyGenerator类的具体代码实现。

摘要由CSDN通过智能技术生成

1. IKeyGenerator主键生成

内置实现类，缺少mysql，因为mysql不支持序列，只能通过表模拟序列
H2KeyGenerator
PostgreKeyGenerator
KingbaseKeyGenerator	人大金仓的KES数据库
DB2KeyGenerator
OracleKeyGenerator

2. 新建表模拟序列

CREATE TABLE `dual_sequence` (
  `sequence_key` varchar(100) NOT NULL COMMENT '序列key（主键）',
  `step` int NOT NULL DEFAULT '1' COMMENT '步长',
  `start_value` bigint NOT NULL DEFAULT '0' COMMENT '开始值',
  `end_value` bigint DEFAULT NULL COMMENT '结束值',
  `ver` int DEFAULT NULL COMMENT '版本号',
  PRIMARY KEY (`sequence_key`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci COMMENT='临时序列表 '

--添加一条记录
INSERT INTO dual_sequence
(sequence_key, step, start_value, end_value, ver)
VALUES('next.dbid', 1, 0, NULL, 0);

3.代码实现

MysqlKeyGenerator类

package com.yl.mybatisplus.plugin;

import com.baomidou.mybatisplus.core.incrementer.IKeyGenerator;
import org.springframework.stereotype.Component;
@Component
public class MysqlKeyGenerator implements IKeyGenerator {
    /**
     * 执行sql
     *
     * @param incrementerName 序列名称
     * @return
     */
    @Override
    public String executeSql(String incrementerName) {
        String sql = "update dual_sequence set start_value=start_value+step where sequence_key='"+ incrementerName +"';" +
                "select start_value as id from dual_sequence where sequence_key='" + incrementerName + "' for update;";
        return sql;
    }
}

实体类上添加@KeySequence，这里value是主键值，不是序列名

@Data
@TableName("luck_reward_emp")
@KeySequence(value = "next.dbid")
public class LuckRewardEmp {
    @TableId(value = "id", type = IdType.INPUT)
    private Long id;

}