JPA使用pg数据库时，bool字段不能跨库迁移的解决方案

首先，大多数人的印象里，hibernate作为一个笨重学习成本高的近乎全自动的框架它的优点就是可以支持很多数据库，但是最近研究发现，java中的boolean类型的字段，在mariadb/mysql 中为bit 0/1,在sqlserver/oracle中为bit 0/1 numeric(1,0) check * in (0,1),然而在PG数据库中却是bool，因为pg就是支持boolean到bool的映射

所以，java中的boolean 在除PG的数据库之外的几个常见数据库中存储的都是 0/1，这就导致了数据迁移时会出现不兼容的问题。

在GPT的帮助下，经研究hibernate的数据库方言发现，可以通过修改官方的方言以达到实现兼容的目的。

相关代码如下：

package com.example.demo.jpa.dialect;

import java.sql.Types;

import org.hibernate.boot.model.TypeContributions;
import org.hibernate.dialect.PostgreSQLDialect;
import org.hibernate.service.ServiceRegistry;
import org.hibernate.type.descriptor.sql.internal.DdlTypeImpl;
import org.hibernate.type.descriptor.sql.spi.DdlTypeRegistry;

/***
 * 将java中的bool 处理成pg中的numeric(1,0)
 * 
 *
 */
public class PgDialect extends PostgreSQLDialect {

    @Override
    public int getPreferredSqlTypeCodeForBoolean() {
        return Types.NUMERIC;
    }

    @Override
    public String toBooleanValueString(boolean bool) {
        // 将布尔值映射为 numeric(1, 0)
        return bool ? "1" : "0";
    }

    @Override
    public void contributeTypes(TypeContributions typeContributions, ServiceRegistry serviceRegistry) {
        super.contributeTypes(typeContributions, serviceRegistry);
        // 把java中的Boolean转为0/1 把数据库中的0/1 转为Boolean
        typeContributions.getTypeConfiguration().getJavaTypeRegistry().addDescriptor(new PgBooleanJavaType());
        final DdlTypeRegistry ddlTypeRegistry = typeContributions.getTypeConfiguration().getDdlTypeRegistry();
        ddlTypeRegistry.addDescriptor(new DdlTypeImpl(Types.BOOLEAN, "numeric(1,0) ", this));
    }

}

 

package com.example.demo.jpa.dialect;

import java.math.BigDecimal;
import java.util.Objects;

import org.hibernate.type.descriptor.WrapperOptions;
import org.hibernate.type.descriptor.java.BooleanJavaType;

/**
 * 当从数据库的字段类型为Boolean时，如何与java中的boolean互相映射
 *
 */
public class PgBooleanJavaType extends BooleanJavaType {

    private static final long serialVersionUID = 1L;

    @SuppressWarnings("unchecked")
    @Override
    public <X> X unwrap(Boolean value, Class<X> type, WrapperOptions options) {
        if (value == null) {
            return null;
        } else if (BigDecimal.class.isAssignableFrom(type)) {
            return (X) (Objects.equals(true, value) ? new BigDecimal(1) : new BigDecimal(0));
        } else if (Boolean.class.equals(type)) {
            return (X) value;
        } else if (Byte.class.equals(type)) {
            return (X) toByte(value);
        }
        if (Short.class.equals(type)) {
            return (X) toShort(value);
        } else if (Integer.class.equals(type)) {
            return (X) toInteger(value);
        } else if (Long.class.equals(type)) {
            return (X) toLong(value);
        } else if (Character.class.equals(type)) {
            return (X) Character.valueOf(value ? '1' : '0');
        } else if (String.class.equals(type)) {
            return (X) (value ? "1" : "0");
        }
        throw unknownUnwrap(type);
    }

}