jpa插入让主键id使用自定义值

最新推荐文章于 2024-05-07 16:42:46 发布

qazwsx081

最新推荐文章于 2024-05-07 16:42:46 发布

阅读量7k

点赞数 1

本文链接：https://blog.csdn.net/qazwsx081/article/details/88427794

版权

    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY, generator = "idOrGenerate")
    @GenericGenerator(name = "idOrGenerate", strategy = "com.sunlands.util.IdOrGenerate")
    Integer id;

package com.sunlands.util;

import com.sunlands.domain.ReturnMessageDO;
import org.hibernate.HibernateException;
import org.hibernate.engine.spi.SessionImplementor;
import org.hibernate.id.IdentityGenerator;
import java.io.Serializable;

public class IdOrGenerate extends IdentityGenerator {
    @Override
    public Serializable generate(SessionImplementor session, Object obj) throws HibernateException {
        if (obj == null){
            throw new HibernateException(new NullPointerException()) ;
        }
        if ((((ReturnMessageDO) obj).getId()) == null) {//id is null it means generate ID
            Serializable id = super.generate(session, obj) ;
            return id;
        } else {
            return ((ReturnMessageDO) obj).getId();//id is not null so using assigned id.

        }
    }
}