DbUnit和JSON列类型支持

时间:2018-10-16 22:50:58

标签: java mysql testing dbunit spring-test-dbunit

我想知道使DbUnit与MySQL的JSON(B)类型列一起使用的最佳方法是什么?我们确实到处都有这样的列,每当我尝试通过@DatabaseSetup从XML文件中馈送测试数据时,我都会收到NoSuchColumnException,这阻止了我使用与JSON相关的实体或存储库进行任何明智的集成测试形式:

org.dbunit.dataset.NoSuchColumnException: assistant_event.AEV_CONTEXT -  (Non-uppercase input column: aev_context) in ColumnNameToIndexes cache map. Note that the map's column names are NOT case sensitive.

我发现这是因为我的AEV_CONTEXT列未被识别是因为它是一个JSON列:

@Type(type = "json")
@Column(name = "aev_context", columnDefinition = "json")
private Context context;

但是,尝试解决该问题时遇到了困难。而且,更奇怪的是,我在这里的任何地方都找不到这种解决方法!实际上,我不知道这是Hibernate还是DbUnit。

到目前为止,真的只有我遇到这个问题吗?任何建议将不胜感激!

哦,如果您想知道,这就是我获得Hibernate的JSON支持的方式:

https://vladmihalcea.com/how-to-map-json-objects-using-generic-hibernate-types/

2 个答案:

答案 0 :(得分:1)

一种可能的解决方案是定义您自己的DataTypeFactory并使用它。我这样做是为了支持PostgreSQL JSONB数据类型,对于MySQL可以做类似的事情。

class NewPostgresqlDataTypeFactory : PostgresqlDataTypeFactory() {
    override fun createDataType(sqlType: Int, sqlTypeName: String?): DataType {
        return when (sqlTypeName) {
            "jsonb" -> return JsonbDataType()
            else -> super.createDataType(sqlType, sqlTypeName)
        }
    }

    class JsonbDataType : AbstractDataType("jsonb", Types.OTHER, String::class.java, false) {
        override fun typeCast(obj: Any?): Any {
            return obj.toString()
        }

        override fun getSqlValue(column: Int, resultSet: ResultSet): Any {
            return resultSet.getString(column)
        }

        override fun setSqlValue(value: Any?, column: Int, statement: PreparedStatement) {
            val jsonObj = PGobject()
            jsonObj.type = "json"
            jsonObj.value = value?.toString()
            statement.setObject(column, jsonObj)
        }
    }
}

答案 1 :(得分:0)

万一有人需要,这是PostgreSQL JSONB数据类型的Java版本this

import org.dbunit.dataset.datatype.AbstractDataType;
import org.dbunit.dataset.datatype.DataType;
import org.dbunit.dataset.datatype.DataTypeException;
import org.dbunit.dataset.datatype.TypeCastException;
import org.dbunit.ext.postgresql.PostgresqlDataTypeFactory;
import org.postgresql.util.PGobject;

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

public class NewPostgresqlDataTypeFactory extends PostgresqlDataTypeFactory {
    @Override
    public DataType createDataType(int sqlType, String sqlTypeName) throws DataTypeException {
        if (sqlTypeName.equals("jsonb")) {
            return new JsonbDataType();
        } else {
            return super.createDataType(sqlType, sqlTypeName);
        }
    }

    public static class JsonbDataType extends AbstractDataType {

        public JsonbDataType() {
            super("jsonb", Types.OTHER, String.class, false);
        }

        @Override
        public Object typeCast(Object obj) throws TypeCastException {
            return obj.toString();
        }

        @Override
        public Object getSqlValue(int column, ResultSet resultSet) throws SQLException, TypeCastException {
            return resultSet.getString(column);
        }

        @Override
        public void setSqlValue(Object value,
                                int column,
                                PreparedStatement statement) throws SQLException, TypeCastException {
            final PGobject jsonObj = new PGobject();
            jsonObj.setType("json");
            jsonObj.setValue(value == null ? null : value.toString());

            statement.setObject(column, jsonObj);
        }
    }
}