jpa使用uuid策略后无法手动设置id的问题及解决

 更新时间:2024年08月12日 09:12:09   作者:流年ln  
这篇文章主要介绍了jpa使用uuid策略后无法手动设置id的问题及解决,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教

jpa使用uuid策略后无法手动设置id

实体对象定义如下:

@Data
@Entity
@Table(name = "sys_user")
public class UserDO {

	/** 用户id */
	@Id
	@GenericGenerator(name = "uuid", strategy = "org.hibernate.id.UUIDGenerator")
	@GeneratedValue(generator = "uuid")
	@Column(length = 36, nullable = false)
	private String userId;
	....
}

调用jpa的save方法,即使手动设置了id,手动设置的这个id也会被覆盖。

要求

当用户手动设置了id,以用户设置的id为准,否则使用uuid

解决方案

使用自定义id生成策略,判断id非空

import cn.hutool.core.util.StrUtil;
import lombok.extern.slf4j.Slf4j;
import org.hibernate.engine.spi.SharedSessionContractImplementor;
import org.hibernate.id.UUIDGenerator;

import javax.persistence.Id;
import java.io.Serializable;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;

/**
 * 自定义uuid生成策略, 扩展 {@link org.hibernate.id.UUIDGenerator}
 * 当用户设置了id,以用户设置的id为主,为空则生成uuid
 * {@link javax.persistence.Id} 注解的字段只处理第一个
 *
 * @author qiudw
 * @date 7/7/2023
 */
@Slf4j
public class CustomerUuidGenerator extends UUIDGenerator {

	@Override
	public Serializable generate(SharedSessionContractImplementor session, Object obj) {
		Class<?> clazz = obj.getClass();
		log.debug("generate id class: {}", clazz.getName());
		Field[] fields = clazz.getDeclaredFields();
		try {
			for (Field field : fields) {
				Id idAnnotation = field.getAnnotation(Id.class);
				if (idAnnotation == null) {
					continue;
				}
				String fieldName = field.getName();
				log.debug("field name: {}", fieldName);
				String getMethodName = generateGetMethodName(fieldName);
				log.debug("get method name: {}", getMethodName);
				Method declaredMethod = clazz.getDeclaredMethod(getMethodName);
				String idValue = (String) declaredMethod.invoke(obj);
				if (StrUtil.isNotBlank(idValue)) {
					log.debug("use set value: {}", idValue);
					return idValue;
				}
			}
		} catch (NoSuchMethodException e) {
			log.error("no such method", e);
		} catch (InvocationTargetException | IllegalAccessException e) {
			log.error("invoke exception", e);
		}

		return super.generate(session, obj);
	}

	/**
	 * 生成get和set的方法名称
	 * type + 首字母大写(fieldName)
	 *
	 * @param fieldName userId
	 * @return type = set, fieldName = userId, 返回 setUserId
	 */
	private String generateGetMethodName(String fieldName) {
		return "get" + fieldName.substring(0, 1).toUpperCase() + fieldName.substring(1);
	}

}

实体对象使用自定的策略

@Data
@Entity
@Table(name = "sys_user")
public class UserDO {

	/** 用户id */
	@Id
	@GenericGenerator(name = "uuid", strategy = "com.user.CustomerUuidGenerator")
	@GeneratedValue(generator = "uuid")
	@Column(length = 36, nullable = false)
	private String userId;
	....
}

现在可以手动设置id了

不设置则自动生成uuid

总结

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

相关文章

  • java中mybatis和hibernate的用法总结

    java中mybatis和hibernate的用法总结

    在本篇文章里小编给大家整理的是一篇关于java中mybatis和hibernate的用法总结内容,有兴趣的朋友们可以学习参考下。
    2021-01-01
  • JpaRepository如何实现增删改查并进行单元测试

    JpaRepository如何实现增删改查并进行单元测试

    这篇文章主要介绍了JpaRepository如何实现增删改查并进行单元测试,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教
    2021-11-11
  • Java 解析线程的几种状态详解

    Java 解析线程的几种状态详解

    这篇文章主要为大家详细介绍了Java 解析线程的几种状态,文中示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下,希望能够给你带来帮助
    2022-03-03
  • Java8新特性之重复注解(repeating annotations)浅析

    Java8新特性之重复注解(repeating annotations)浅析

    这篇文章主要介绍了Java8新特性之重复注解(repeating annotations)浅析,这个新特性只是修改了程序的可读性,是比较小的一个改动,需要的朋友可以参考下
    2014-06-06
  • SpringBoot项目Jar包使用systemctl运行过程

    SpringBoot项目Jar包使用systemctl运行过程

    这篇文章主要介绍了SpringBoot项目Jar包使用systemctl运行过程,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
    2025-03-03
  • Java构造器与传值学习总结

    Java构造器与传值学习总结

    这篇文章主要为大家详细介绍了Java构造器与传值学习总结,文中示例介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2022-01-01
  • Java初学者必会的Map集合及其原理

    Java初学者必会的Map集合及其原理

    这篇文章主要给大家介绍Map集合及其原理,该集合中的信息是key-value形式,Map集合与Collection集合又有什么不同呢,要想搞清楚以上问题,下面跟着小编一起来看看吧
    2023-06-06
  • Spring Security 使用 OncePerRequestFilter 过滤器校验登录过期、请求日志等操作

    Spring Security 使用 OncePerRequestFilter 

    OncePerRequestFilter是一个过滤器,每个请求都会执行一次;一般开发中主要是做检查是否已登录、Token是否过期和授权等操作,而每个操作都是一个过滤器,下面介绍Spring Security 使用 OncePerRequestFilter 过滤器校验登录过期、请求日志等操作方法,感兴趣的朋友一起看看吧
    2024-06-06
  • Spring MVC 图片的上传和下载功能

    Spring MVC 图片的上传和下载功能

    SSM 框架是一种基于Java的Web开发框架,其中Spring作为控制层、SpringMVC作为视图层、MyBatis作为持久层,这个框架非常适合Web应用程序的开发,这篇文章主要介绍了Spring MVC 图片的上传和下载功能,需要的朋友可以参考下
    2023-03-03
  • idea 安装 Mybatis 开发帮助插件 MyBatisCodeHelper-Pro 插件破解版的方法

    idea 安装 Mybatis 开发帮助插件 MyBatisCodeHelper-Pro 插件破解版的方法

    MyBatisCodeHelper-Pro 插件可以帮助我们快速的开发 mybatis,这篇文章给大家介绍idea 安装 Mybatis 开发帮助插件 MyBatisCodeHelper-Pro 插件破解版的相关知识,感兴趣的朋友跟随小编一起看看吧
    2020-09-09

最新评论