博客

在实际开发中遇到postgresql中定义的数组字段，下面解决两个问题，如何定义数组字段的默认值为空格数组，以及如何再java实体类中直接使用数组对象接受数据或把数据存入数据库。

1.在postgresql中定义数组对象及默认值

以字符串你数组为例：

比如一个字段用于存储多张图片的url，可以使用一下sql定义

pictures _varchar NOT NUll default ARRAY[]::character varying[]

2.实体类存入数组到数据库并接受数据库的数组数据

直接定义数组字段并无法实现数据存入和接受。还需要一个typeHandler。这个typeHandler是自定的，而且整型数组和字符串数组不一样，下面以字符串数组为例。

typeHandler类如下：

import org.apache.ibatis.type.BaseTypeHandler;
import org.apache.ibatis.type.JdbcType;

import java.sql.*;
import java.util.List;

public class StringListArrayTypeHandler extends BaseTypeHandler<List<String>> {

    @Override
    public void setNonNullParameter(PreparedStatement preparedStatement, int i, List<String> strings, JdbcType jdbcType) throws SQLException {
        if (strings != null) {
            Array array = preparedStatement.getConnection().createArrayOf(JdbcType.VARCHAR.name(), strings.toArray(new String[0]));
            preparedStatement.setArray(i, array);
        }
    }

    @Override
    public List<String> getNullableResult(ResultSet resultSet, String s) throws SQLException {
        Array array = resultSet.getArray(s);
        if (array == null) {
            return null;
        }
        String[] result = (String[]) array.getArray();
        array.free();
        return List.of(result);
    }

    @Override
    public List<String> getNullableResult(ResultSet resultSet, int i) throws SQLException {
        Array array = resultSet.getArray(i);
        if (array == null) {
            return null;
        }
        String[] result = (String[]) array.getArray();
        array.free();
        return List.of(result);
    }

    @Override
    public List<String> getNullableResult(CallableStatement callableStatement, int i) throws SQLException {
        Array array = callableStatement.getArray(i);
        if (array == null) {
            return null;
        }
        String[] result = (String[]) array.getArray();
        array.free();
        return List.of(result);
    }
}

typeHandler一般放在项目的config目录下。

实体类的定义如下：

@TableName(autoResultMap = true)
public class Entity implements Serializable {

    private static final long serialVersionUID = 1L;

    @TableId(type = IdType.AUTO)
    private Long id;
    
    private String code;

    private String name;

    /**
     * 图片
     */
    @TableField(typeHandler = StringListArrayTypeHandler.class)
    private List<String> pictures;
}

注意注解@TableName(autoResultMap = true)也要加上。

附加整型数组的typeHandler：

import org.apache.ibatis.type.BaseTypeHandler;
import org.apache.ibatis.type.JdbcType;

import java.sql.*;
import java.util.Collections;
import java.util.List;

public class IntegerListArrayTypeHandler extends BaseTypeHandler<List<Integer>> {

    @Override
    public void setNonNullParameter(PreparedStatement preparedStatement, int i, List<Integer> integers, JdbcType jdbcType) throws SQLException {
        if (integers != null) {
            Array array = preparedStatement.getConnection().createArrayOf(JdbcType.INTEGER.name(), integers.toArray(new Integer[0]));
            preparedStatement.setArray(i, array);
        }
    }

    @Override
    public List<Integer> getNullableResult(ResultSet resultSet, String s) throws SQLException {
        Array array = resultSet.getArray(s);
        if (array == null) {
            return null;
        }
        Integer[] result = (Integer[]) array.getArray();
        array.free();
        return List.of(result);
    }

    @Override
    public List<Integer> getNullableResult(ResultSet resultSet, int i) throws SQLException {
        Array array = resultSet.getArray(i);
        if (array == null) {
            return null;
        }
        Integer[] result = (Integer[]) array.getArray();
        array.free();
        return List.of(result);
    }

    @Override
    public List<Integer> getNullableResult(CallableStatement callableStatement, int i) throws SQLException {
        Array array = callableStatement.getArray(i);
        if (array == null) {
            return null;
        }
        Integer[] result = (Integer[]) array.getArray();
        array.free();
        return List.of(result);
    }
}