JDBCTemplate使用BeanPropertyRowMapper设置嵌套的POJO

Question

给定以下示例POJO ：（假定所有属性的Getter和Setter）

class User {
    String user_name;
    String display_name;
}

class Message {
    String title;
    String question;
    User user;
}

可以轻松地查询数据库（在我的情况下为postgres），并使用BeanPropertyRowMapper填充Message类的列表，其中db字段与POJO中的属性匹配：（假定DB表具有与POJO属性对应的字段）。

NamedParameterDatbase.query("SELECT * FROM message", new BeanPropertyRowMapper(Message.class));

我想知道-是否有一种方便的方法来构造单个查询和/或以这种方式创建行映射器，以也在消息中填充内部“用户” POJO的属性。

也就是说，一些语法上的查询在查询中每个结果行：

SELECT * FROM message, user WHERE user_id = message_id

产生消息列表，并填充相关的用户

用例：

最终，这些类作为一个序列化对象从Spring Controller传递回去，这些类被嵌套，以使生成的JSON / XML具有良好的结构。

目前，通过执行两个查询并在循环中手动设置每个消息的用户属性可以解决这种情况。可用，但我认为应该可以使用更优雅的方式。

更新：使用的解决方案-

@Will Keeling的荣誉来自使用自定义行映射器的答案的启发-我的解决方案增加了bean属性映射的添加，以使字段分配自动化。

需要说明的是，结构化查询是为了给相关的表名加上前缀（但是由于没有标准约定可以这样做，因此查询是通过编程方式构建的）：

SELECT title AS "message.title", question AS "message.question", user_name AS "user.user_name", display_name AS "user.display_name" FROM message, user WHERE user_id = message_id

然后，自定义行映射器创建几个bean映射，并根据列的前缀设置其属性：（使用元数据获取列名）。

public Object mapRow(ResultSet rs, int i) throws SQLException {

    HashMap<String, BeanMap> beans_by_name = new HashMap();

    beans_by_name.put("message", BeanMap.create(new Message()));
    beans_by_name.put("user", BeanMap.create(new User()));

    ResultSetMetaData resultSetMetaData = rs.getMetaData();

    for (int colnum = 1; colnum <= resultSetMetaData.getColumnCount(); colnum++) {

        String table = resultSetMetaData.getColumnName(colnum).split("\\.")[0];
        String field = resultSetMetaData.getColumnName(colnum).split("\\.")[1];

        BeanMap beanMap = beans_by_name.get(table);

        if (rs.getObject(colnum) != null) {
            beanMap.put(field, rs.getObject(colnum));
        }
    }

    Message m = (Task)beans_by_name.get("message").getBean();
    m.setUser((User)beans_by_name.get("user").getBean());

    return m;
}

同样，对于两个类的连接来说，这似乎有些过头了，但是IRL用例涉及具有数十个字段的多个表。

Answer 1

Spring AutoGrowNestedPaths在BeanMapper接口中引入了一个新属性。

只要SQL查询使用来格式化列名称。分隔符（如前），那么行映射器将自动定位内部对象。

这样，我创建了一个新的通用行映射器，如下所示：

查询：

SELECT title AS "message.title", question AS "message.question", user_name AS "user.user_name", display_name AS "user.display_name" FROM message, user WHERE user_id = message_id

行映射器：

package nested_row_mapper;

import org.springframework.beans.*;
import org.springframework.jdbc.core.RowMapper;
import org.springframework.jdbc.support.JdbcUtils;

import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;

public class NestedRowMapper<T> implements RowMapper<T> {

  private Class<T> mappedClass;

  public NestedRowMapper(Class<T> mappedClass) {
    this.mappedClass = mappedClass;
  }

  @Override
  public T mapRow(ResultSet rs, int rowNum) throws SQLException {

    T mappedObject = BeanUtils.instantiate(this.mappedClass);
    BeanWrapper bw = PropertyAccessorFactory.forBeanPropertyAccess(mappedObject);

    bw.setAutoGrowNestedPaths(true);

    ResultSetMetaData meta_data = rs.getMetaData();
    int columnCount = meta_data.getColumnCount();

    for (int index = 1; index <= columnCount; index++) {

      try {

        String column = JdbcUtils.lookupColumnName(meta_data, index);
        Object value = JdbcUtils.getResultSetValue(rs, index, Class.forName(meta_data.getColumnClassName(index)));

        bw.setPropertyValue(column, value);

      } catch (TypeMismatchException | NotWritablePropertyException | ClassNotFoundException e) {
         // Ignore
      }
    }

    return mappedObject;
  }
}