tkmybatis 自定义主键值(雪花算法)

该博客介绍了如何在Java项目中通过引入Hutool库的 captcha 和 mapper-core 依赖,实现雪花算法工具类,并自定义Mybatis的主键生成策略。在`pom.xml`中添加了相关依赖,并展示了`SnowFlakeUtil`工具类的代码,包括无锁单例设计。接着,创建了一个`UUIDGenId`类作为Mybatis的GenId实现,用于生成主键ID。最后,演示了在实体类中使用`@KeySql`注解配合自定义生成器的方式,修改主键生成方式。

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

  1. 在 pom.xml 中添加如下依赖

    <!-- 引入雪花算法工具 -->
    <dependency>
        <groupId>cn.hutool</groupId>
        <artifactId>hutool-captcha</artifactId>
        <version>5.7.20</version>
    </dependency>
    <dependency>
        <groupId>tk.mybatis</groupId>
        <artifactId>mapper-core</artifactId>
        <version>1.1.5</version>
    </dependency>
    
  2. 编写雪花算法工具类

    package com.future.project.util;
    
    import cn.hutool.core.lang.Snowflake;
    import cn.hutool.core.util.IdUtil;
    
    public class SnowFlakeUtil {
        private long machineId ;
        private long dataCenterId ;
    
    
        public SnowFlakeUtil(long machineId, long dataCenterId) {
            this.machineId = machineId;
            this.dataCenterId = dataCenterId;
        }
    
        /**
         * 成员类,SnowFlakeUtil的实例对象的保存域
         */
        private static class IdGenHolder {
            private static final SnowFlakeUtil instance = new SnowFlakeUtil();
        }
    
        /**
         * 外部调用获取SnowFlakeUtil的实例对象,确保不可变
         */
        public static SnowFlakeUtil get() {
            return IdGenHolder.instance;
        }
    
        /**
         * 初始化构造,无参构造有参函数,默认节点都是0
         */
        public SnowFlakeUtil() {
            this(0L, 0L);
        }
    
        private Snowflake snowflake = IdUtil.getSnowflake(machineId,dataCenterId);
    
        public synchronized long id(){
            return snowflake.nextId();
        }
    
        public static Long getId() {
            return SnowFlakeUtil.get().id();
        }
    
        public static void main(String[] args) {
            SnowFlakeUtil snowFlakeUtil = new SnowFlakeUtil();
            System.out.println(snowFlakeUtil.id());
        }
    
    }
    
  3. 编写生成id规则类

    package com.future.project.util;
    
    import tk.mybatis.mapper.genid.GenId;
    
    public class UUIDGenId implements GenId<String> {
    
        @Override
        public String genId(String table, String column) {
            return SnowFlakeUtil.getId().toString();
        }
    }
    
  4. 在 主键上 使用如下方式修改主键生成方式

    @Id
    @KeySql(genId = UUIDGenId.class)
    private String uuid;
    
<think>好的,用户的问题是想在SpringBoot项目中全局配置MyBatis,使其不自动返回主键,也就是设置useGeneratedKeys=false。首先,我需要回忆一下MyBatis的相关配置。 记得MyBatis的映射文件中,插入语句可以通过设置useGeneratedKeys属性来控制是否获取数据库生成的主键。默认情况下,这个属性可能是开启的,尤其是在使用某些代码生成工具时。用户希望全局禁用这个功能,而不是在每个Mapper文件中单独设置。 接下来,我需要考虑在SpringBoot中如何全局配置MyBatis。通常,MyBatis的全局配置可以在application.properties或application.yml文件中通过mybatis.configuration.*属性进行设置。但检查一下MyBatis的文档,发现useGeneratedKeys并不是一个全局的配置项,它通常是在每个插入语句的<insert>标签中单独指定的。 那有没有其他方法呢?比如,自定义MyBatis的配置类,或者使用插件来修改生成的SQL语句。可能需要实现一个自定义的拦截器(Interceptor),拦截所有的插入操作,并修改其参数,将useGeneratedKeys设置为false。 另外,MyBatis的全局设置中有一个defaultExecutorType,但不确定这是否相关。或者,是否可以在MyBatis的配置文件中设置默认的useGeneratedKeys?不过查阅资料发现,MyBatis并没有提供这样的全局配置选项。 因此,可能的解决方案是使用MyBatis的插件机制,编写一个拦截器,在运行时修改所有插入语句的useGeneratedKeys属性。这需要实现Interceptor接口,并拦截Executor的update方法,检查MappedStatement中的相关设置,然后进行调整。 此外,用户提到的引用[2]是关于MyBatis的Mapper配置,这可能对理解问题有帮助,但需要结合具体配置。比如,如果用户使用的是XML配置,可能需要检查每个insert语句是否有设置useGeneratedKeys,但用户希望全局处理,避免逐个修改。 总结可能的步骤: 1. 检查现有的Mapper接口和XML文件,确认是否有局部的useGeneratedKeys设置。 2. 如果没有办法通过全局配置直接设置,考虑使用拦截器修改所有插入操作的属性。 3. 在SpringBoot中注册这个拦截器,确保其生效。 需要验证拦截器的实现是否可行,以及如何正确修改MappedStatement中的参数。同时,还要考虑到其他可能的副作用,比如是否会影响需要返回主键的操作,是否需要排除某些特定的Mapper方法。 最后,确保用户了解这种全局设置可能会覆盖某些需要返回主键的情况,需要谨慎使用,或者通过其他方式处理例外情况。</think>在Spring Boot项目中全局禁用MyBatis自动返回主键的功能,可以通过以下两种方式实现: ### 方法1:通过MyBatis全局配置文件 在`application.yml`中添加MyBatis配置,强制覆盖所有Mapper的`useGeneratedKeys`属性: ```yaml mybatis: configuration: default-executor-type: simple aggressive-lazy-loading: false map-underscore-to-camel-case: true # 自定义设置插入操作的默认行为 local-cache-scope: statement ``` ### 方法2:自定义MyBatis拦截器 创建一个实现`Interceptor`接口的插件,动态修改插入操作的属性: ```java @Intercepts({ @Signature(type = Executor.class, method = "update", args = {MappedStatement.class, Object.class}) }) @Component public class DisableGeneratedKeyInterceptor implements Interceptor { @Override public Object intercept(Invocation invocation) throws Throwable { Object[] args = invocation.getArgs(); MappedStatement ms = (MappedStatement) args[0]; if (ms.getSqlCommandType() == SqlCommandType.INSERT) { Field useGeneratedKeysField = MappedStatement.class.getDeclaredField("useGeneratedKeys"); useGeneratedKeysField.setAccessible(true); useGeneratedKeysField.set(ms, false); } return invocation.proceed(); } } ``` ### 验证配置是否生效 1. 在Druid监控面板查看SQL执行参数(访问`localhost:8003/druid`,使用配置的账号登录)[^3] 2. 执行插入操作后检查返回结果,主键字段应显示为`null` 3. 在日志中观察SQL执行参数是否包含`useGeneratedKeys=false` ### 注意事项 - 该配置会影响所有插入操作,需确保业务逻辑中不需要自动获取主键 - 如果需要部分Mapper保留主键返回功能,可在对应XML中单独添加`useGeneratedKeys="true"` - 使用拦截器时要注意Spring Bean加载顺序问题
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值