Question

例如，在PostgreSQL中有一个表，该表的列为text[]类型：

CREATE TABLE t
(
    id integer,
    name text,
    tags text[],
    PRIMARY KEY (id)
)

现在，我想通过两种方式选择tags：

使用主键tags选择id，结果应为List<String>类型
使用tags选择name，结果应为List<List<String>>类型

我应该如何编写MyBatis映射器来实现这一目标？

Answer 1

您不是第一个遇到此问题的人。

项目common-mybatis的类型处理程序正是针对此用例的：StringArrayTypeHandler

只需将其添加到MyBatis配置中即可：

    <typeHandlers>
        <typeHandler handler="org.gbif.mybatis.type.StringArrayTypeHandler"/>
    </typeHandlers>

...然后进行映射就很简单：

    <select id="getTagsById" resultType="java.util.List">
        SELECT tags FROM t WHERE id = #{id}
    </select>

    <select id="getTagsByName" resultType="java.util.List">
        SELECT tags FROM t WHERE name = #{name}
    </select>

...并在代码中：

try (SqlSession session = sessionFactory.openSession()) {
    List<String> tags = session.selectOne("[...].getTagsById", 1);
    System.out.println("Tags: " + tags);

    List<List<String>> multiTags = session.selectList("[...].getTagsByName", "test");
    System.out.println("Tags: " + multiTags);
}

已针对具有JDBC驱动程序版本42.2.5的PostgreSQL 11和以下测试数据进行了测试：

select * from t;
 id |  name   |                tags                
----+---------+------------------------------------
  1 | test    | {Thriller,Drama}
  2 | my name | {Science-Fiction,Adventure,Horror}
  3 | test    | {Comedy,Adventure}
(3 rows)

...产生：

Tags: [Thriller, Drama]
Tags: [[Thriller, Drama], [Comedy, Adventure]]

Answer 2

您仍然可以使用Java映射器，但是，当返回类型为SqlSession#selectList时，MyBatis在内部调用List，而这并不是您想要的。
因此，您需要使用Object作为返回类型。

@Select("select tags from t where id = #{id}")
Object getTagById(Integer id);

@Select("select tags from t where name = #{name}")
List<Object> getTagByName(String name);

然后在配置中全局注册您的类型处理程序。即

<typeHandlers>
  <typeHandler handler="xxx.yyy.ListArrayTypeHandler" />
</typeHandlers>

或

configuration.getTypeHandlerRegistry()
  .register(ListArrayTypeHandler.class);

为完整起见，这是一个示例类型处理程序实现。

import java.sql.Array;
import java.sql.CallableStatement;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

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

@MappedJdbcTypes({ JdbcType.ARRAY })
@MappedTypes({ Object.class })
public class ListArrayTypeHandler extends BaseTypeHandler<List<?>> {

  @Override
  public void setNonNullParameter(PreparedStatement ps, int i,
      List<?> parameter, JdbcType jdbcType) throws SQLException {
    //  JDBC type is required
    Array array = ps.getConnection().createArrayOf("TEXT", parameter.toArray());
    try {
      ps.setArray(i, array);
    } finally {
      array.free();
    }
  }

  @Override
  public List<?> getNullableResult(ResultSet rs, String columnName) throws SQLException {
    return extractArray(rs.getArray(columnName));
  }

  @Override
  public List<?> getNullableResult(ResultSet rs, int columnIndex) throws SQLException {
    return extractArray(rs.getArray(columnIndex));
  }

  @Override
  public List<?> getNullableResult(CallableStatement cs, int columnIndex) throws SQLException {
    return extractArray(cs.getArray(columnIndex));
  }

  protected List<?> extractArray(Array array) throws SQLException {
    if (array == null) {
      return null;
    }
    Object javaArray = array.getArray();
    array.free();
    return new ArrayList<>(Arrays.asList((Object[])javaArray));
  }
}

仅供参考，要将List存储到tags列中，您可能必须明确指定类型处理程序。

insert into t (...) values (#{id}, #{name},
  #{tags,typeHandler=xxx.yyy.ListArrayTypeHandler})

如何在mybatis中选择文本数组类型的结果？

2 个答案: