hibernate跨資料庫，json欄位處理方案，自定義擴充套件JsonStringType

·志堅行遠·發表於2023-03-08
資料庫JSON套件

一、背景

對於一些不經常更新的靜態資料，我們喜歡使用json格式儲存。推薦的做法是將json資料儲存在key-value資料庫，但這無疑增加了技術成本，所以我們通常還是儲存在RDB資料庫中。我們在使用hibernate，對json資料的存取期望是，存能自動轉換為json格式儲存，取能自動將json資料轉換為物件資料。在資料庫方面，Mysql5.7版本以後新增的功能，Mysql提供了一個原生的Json型別，Json值將不再以字串的形式儲存，而是採用一種允許快速讀取文字元素（document elements）的內部二進位制（internal binary）格式。在Json列插入或者更新的時候將會自動驗證Json文字，未透過驗證的文字將產生一個錯誤資訊。而除了mysql,pg極少部分支援json欄位型別的其他資料庫，還是需要大文字欄位進行儲存。

二、實現方案

關鍵技術： hibernate-types 將Java物件或Jackson JsonNode為JPA實體屬性。

  • 引入依賴-hibernate-types
<dependency>
   <groupId>com.vladmihalcea</groupId>
   <artifactId>hibernate-types-52</artifactId>
   <version>2.10.1</version>
</dependency>
  • 定義實體類
import com.alibaba.fastjson.JSONObject;
import com.vladmihalcea.hibernate.type.array.IntArrayType;
import com.vladmihalcea.hibernate.type.array.StringArrayType;
import com.vladmihalcea.hibernate.type.json.JsonBinaryType;
import com.vladmihalcea.hibernate.type.json.JsonStringType;
import lombok.Data;
import org.hibernate.annotations.Type;
import org.hibernate.annotations.TypeDef;
import org.hibernate.annotations.TypeDefs;

import javax.persistence.*;

@Data
@Entity
@Table(name = "parents")
@TypeDefs({
        @TypeDef(name = "string-array", typeClass = StringArrayType.class),
        @TypeDef(name = "json", typeClass = JsonStringType.class),
        @TypeDef(name = "jsonb", typeClass = JsonBinaryType.class)
})
public class TestEntity implements Serializable {
    @Id
    private Long id;

    @Column(name = "attr")
    @Type(type = "jsonb")
    private JSONObject attr;

    @Column(name = "user_list")
    @Type(type = "json")
    private List<User> userList;

    @Column(name = "ava_img")
    @Type(type = "string-array")
    private String[] avaImg;
}

以上為常規操作，在多資料庫使用時，你會發現，JsonStringType 型別會轉為資料庫預設的 varchar(255),根本就儲存不了長一點的json資料。

三、擴充套件JsonStringType,適配多資料庫json欄位儲存

也許網路上一搜JsonStringType適配性問題，答案基本都是建議用 @Column(columnDefinition = "json") 、 @Column(columnDefinition = "varchar(max)") 這種寫法，我在上一篇文章中說了，columnDefinition具有不可移植性，只能適配一種或一類資料庫。所以，需要自定義擴充套件JsonStringType。

  • 先分析JsonStringType原始碼
    image
    可以看到JsonStringType的型別描述時由JsonStringSqlTypeDescriptor提供的
    image
    而JsonStringSqlTypeDescriptor中，將資料庫對應的型別寫死為 12
    image

12對應資料庫欄位型別是varchar，這也太坑了，直接給JsonStringType對應的資料庫欄位型別寫死。

  • JsonStringType相容辦法
    既然我們找到了問題原因，那麼我們就容易解決，解決主要思路是將JsonStringType型別對應的資料庫欄位型別可配置
    接著看原始碼
    image
    JsonStringSqlTypeDescriptor 的父類是 AbstractJsonSqlTypeDescriptor
    image
    在AbstractJsonSqlTypeDescriptor中，將json格式寫死為1111，也就是 Types.OTHER型別
    我們重新定義一個JsonStringType，不將SqlType寫死，然後透過配置database-platform指定hibernate方言中，將1111轉換為我們需要的格式。
    原始碼如下：
MyJsonStringType （點選展開） 

import com.fasterxml.jackson.databind.ObjectMapper;
import com.vladmihalcea.hibernate.type.AbstractHibernateType;
import com.vladmihalcea.hibernate.type.json.internal.JsonTypeDescriptor;
import com.vladmihalcea.hibernate.type.util.Configuration;
import com.vladmihalcea.hibernate.type.util.ObjectMapperWrapper;
import org.hibernate.usertype.DynamicParameterizedType;

import java.lang.reflect.Type;
import java.util.Properties;

public class MyJsonStringType extends AbstractHibernateType<Object> implements DynamicParameterizedType {
    public static final MyJsonStringType INSTANCE = new MyJsonStringType();

    public MyJsonStringType() {
        super(MyJsonStringSqlTypeDescriptor.INSTANCE, new JsonTypeDescriptor(Configuration.INSTANCE.getObjectMapperWrapper()));
    }

    public MyJsonStringType(Type javaType) {
        super(MyJsonStringSqlTypeDescriptor.INSTANCE, new JsonTypeDescriptor(Configuration.INSTANCE.getObjectMapperWrapper(), javaType));
    }

    public MyJsonStringType(Configuration configuration) {
        super(MyJsonStringSqlTypeDescriptor.INSTANCE, new JsonTypeDescriptor(configuration.getObjectMapperWrapper()), configuration);
    }

    public MyJsonStringType(ObjectMapper objectMapper) {
        super(MyJsonStringSqlTypeDescriptor.INSTANCE, new JsonTypeDescriptor(new ObjectMapperWrapper(objectMapper)));
    }

    public MyJsonStringType(ObjectMapperWrapper objectMapperWrapper) {
        super(MyJsonStringSqlTypeDescriptor.INSTANCE, new JsonTypeDescriptor(objectMapperWrapper));
    }

    public MyJsonStringType(ObjectMapper objectMapper, Type javaType) {
        super(MyJsonStringSqlTypeDescriptor.INSTANCE, new JsonTypeDescriptor(new ObjectMapperWrapper(objectMapper), javaType));
    }

    public MyJsonStringType(ObjectMapperWrapper objectMapperWrapper, Type javaType) {
        super(MyJsonStringSqlTypeDescriptor.INSTANCE, new JsonTypeDescriptor(objectMapperWrapper, javaType));
    }

    public String getName() {
        return "json";
    }

    protected boolean registerUnderJavaType() {
        return true;
    }

    public void setParameterValues(Properties parameters) {
        ((JsonTypeDescriptor) this.getJavaTypeDescriptor()).setParameterValues(parameters);
    }
}
MyJsonStringSqlTypeDescriptor（點選展開） 
import com.vladmihalcea.hibernate.type.json.internal.AbstractJsonSqlTypeDescriptor;
import org.hibernate.type.descriptor.ValueBinder;
import org.hibernate.type.descriptor.WrapperOptions;
import org.hibernate.type.descriptor.java.JavaTypeDescriptor;
import org.hibernate.type.descriptor.sql.BasicBinder;

import java.sql.CallableStatement;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

public class MyJsonStringSqlTypeDescriptor extends AbstractJsonSqlTypeDescriptor {
    public static final MyJsonStringSqlTypeDescriptor INSTANCE = new MyJsonStringSqlTypeDescriptor();

    public MyJsonStringSqlTypeDescriptor() {
    }

    public <X> ValueBinder<X> getBinder(final JavaTypeDescriptor<X> javaTypeDescriptor) {
        return new BasicBinder<X>(javaTypeDescriptor, this) {
            protected void doBind(PreparedStatement st, X value, int index, WrapperOptions options) throws SQLException {
                st.setString(index, (String)javaTypeDescriptor.unwrap(value, String.class, options));
            }

            protected void doBind(CallableStatement st, X value, String name, WrapperOptions options) throws SQLException {
                st.setString(name, (String)javaTypeDescriptor.unwrap(value, String.class, options));
            }
        };
    }

    protected Object extractJson(ResultSet rs, String name) throws SQLException {
        return rs.getString(name);
    }

    protected Object extractJson(CallableStatement statement, int index) throws SQLException {
        return statement.getString(index);
    }

    protected Object extractJson(CallableStatement statement, String name) throws SQLException {
        return statement.getString(name);
    }
}

實體json定義改為
@TypeDef(name="json",typeClass = MyJsonStringType.class)
大功告成，下邊就可以根據不同資料庫自適應資料庫json欄位型別

  • mysql適配json欄位

    public class MySQLDialect extends MySQL5InnoDBDialect {
	public MySQLDialect() {
		registerColumnType(Types.OTHER, "json");
	}
}

    配置檔案：
    spring.jpa.database-platform: xxxx.xxxx.xx.MySQLDialect

  • 達夢資料庫適配json欄位

    public class DaMengDialect extends DmDialect {
	public DaMengDialect() {
		registerColumnType(Types.OTHER, "LONGVARCHAR");
	}
}

    配置檔案：
    spring.jpa.database-platform: xxxx.xxxx.xx.DaMengDialect

相關文章