时间戳

首页 > 解决方案 > 如何从 jdbc 结果集中返回 java8 Stream

java-8 - 如何从 jdbc 结果集中返回 java8 Stream

问题描述

为了充分利用 java8 流和 Spring4，我在来自 Springs jsdbRestTemplate 的 JDBC resultSet 上使用 Stream API，如下所示（代码缩短和简化）：

public <T> T consumeResultStream(
    String query, 
    Function<Stream<String>, T> extractorFunction
) {
    return jdbcTemplate.query(
        query, 
        resultSet -> {
            Spliterator<String> spliterator = 
                Spliterators.spliteratorUnknownSize(
                    new Iterator<String>() {
                      @Override public boolean hasNext() {
                        return !resultSet.isAfterLast();
                      }
                      @Override public String next() {
                        String result = resultSet.getString(0);
                        resultSet.next();
                        return result;
                      }
                    }, 
                    Spliterator.IMMUTABLE);
           resultStream = StreamSupport.stream(
               spliterator, /*parallel*/ false);
       }
       return extractorFunction.apply(resultStream);
    });
}

这似乎工作正常。客户端可以像这样使用流Api而不用担心jdbc类

List<T> myResult = consumeResultStream("SELECT ...", stream -> 
    stream.filter((String s) -> ...)
        .map(String s -> toT(s))
        .collect(toList()));

但是，当我重构时（尝试将流提供给客户端方法），如下所示：

    final Stream<String> stream = 
        jdbcTemplate.query(query, resultSet -> {
          // ... same as above
          return resultStream;
        });
    return extractorFunction.apply(stream);

我明白了

org.springframework.jdbc.InvalidResultSetAccessException: 
  The object is already closed [90007-199]

所以看起来数据只能在jdbcTemplate.query()方法内读取。有没有一种干净的方法可以规避这个问题并返回来自数据库的元素的惰性流？假设由于结果的大小而无法选择实现结果和流式传输（尽管分页可能是更好的模式）。

标签： java-8java-streamspring-jdbcspring-4

解决方案

JdbcTemplateJPA与 Spring特性相反，不处理超出其调用的事务。
要不允许关闭数据库连接，请从客户端打开一个操作返回的惰性结果的事务。
用注释@Transactional通常就足够了：

@Transactional
public void findLazyData(){
   Stream<String> result = dataAccessService.find(...);
   // where find() contains the JdbcTemplate invocation
}

注意包的来源：org.springframework.transaction.annotation.Transactional.

推荐阅读