Hibernate4中让SequenceGenerator支持字符类型的方法
Hibernate4中让SequenceGenerator支持字符类型的方法
李玉珏 发表于3年前
Hibernate4中让SequenceGenerator支持字符类型的方法
  • 发表于 3年前
  • 阅读 918
  • 收藏 3
  • 点赞 1
  • 评论 0

腾讯云 新注册用户 域名抢购1元起>>>   

摘要: 在Hibernate4中,如果要用SequenceGenerator,其要求数据库中的对应字段为NUMBER等数值类型,但是在具体项目中,因为序列对应的字段往往是主键,虽然是数值但并不参与计算,还有一些其他的原因,往往希望字段定义为字符类型,比如VARCHAR2等,这是合理的需求,本文将提供解决该问题的方法。

        在Hibernate4中,如果要用SequenceGenerator,其要求数据库中的对应字段为NUMBER等数值类型,但是在具体项目中,因为序列对应的字段往往是主键,虽然是数值但并不参与计算,还有一些其他的原因,往往希望字段定义为字符类型,比如VARCHAR2等,这是合理的需求,本文将提供解决该问题的方法。

        经过研究源代码发现,该问题通过Hibernate扩展的方式,会比较麻烦,修改源代码是比较简单方式,下面介绍具体需要修改的源文件,及具体修改方法。

        需要修改的源文件为org.hibernate.id.SequenceGenerator和org.hibernate.id.IdentifierGeneratorHelper。

        IdentifierGeneratorHelper需要修改的地方较多:

    public static IntegralDataTypeHolder getIntegralDataTypeHolder(Class integralType) {
        if ( integralType == Long.class
                || integralType == Integer.class
                || integralType == Short.class
                || integralType == String.class) {//增加对String的判断
            return new BasicHolder( integralType );
        }
        else if ( integralType == BigInteger.class ) {
            return new BigIntegerHolder();
        }
        else if ( integralType == BigDecimal.class ) {
            return new BigDecimalHolder();
        }
        else {
            throw new IdentifierGenerationException(
                    "Unknown integral data type for ids : " + integralType.getName()
            );
        }
    }

        BasicHolder的构造方法:

        public BasicHolder(Class exactType) {
            this.exactType = exactType;
            if ( exactType != Long.class && exactType != Integer.class && exactType != Short.class && exactType != String.class) {//增加对String的判断
                throw new IdentifierGenerationException( "Invalid type for basic integral holder : " + exactType );
            }
        }

        BasicHolder的makeValue方法:

        public Number makeValue() {
            // TODO : should we check for truncation?
            checkInitialized();
            if ( exactType == Long.class || exactType == String.class) {
                return value;
            }
            else if ( exactType == Integer.class ) {
                return ( int ) value;
            }
            else {
                return ( short ) value;
            }
        }

        

        SequenceGenerator的修改比较简单,修改generate方法即可:

    @Override
    public Serializable generate(SessionImplementor session, Object obj) {
        Number n =generateHolder( session ).makeValue();
        if (identifierType.getReturnedClass() == String.class){//增加对String的判断
            return n.toString();
        }else{
            return n;
        }
    }

        这个修改在4.2.12以上版本(小版本)中测试通过,其他版本未验证。

共有 人打赏支持
李玉珏
粉丝 200
博文 43
码字总数 64437
×
李玉珏
如果觉得我的文章对您有用,请随意打赏。您的支持将鼓励我继续创作!
* 金额(元)
¥1 ¥5 ¥10 ¥20 其他金额
打赏人
留言
* 支付类型
微信扫码支付
打赏金额:
已支付成功
打赏金额: