如何从jdbc结果集中返回java8流

时间:2019-07-05 06:21:22

标签: java-8 java-stream spring-jdbc spring-4

为了充分利用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()方法中读取数据。有没有一种干净的方法可以绕过这个问题并返回来自数据库的元素的惰性流?假定实现结果并实现流式传输是不可取的,因为结果的大小(尽管分页可能是更好的模式)。

1 个答案:

答案 0 :(得分:1)

JdbcTemplate不能处理与Spring JPA功能相反的事务。
要不允许关闭数据库连接,请从客户端打开一个事务来操纵返回的惰性结果。
使用@Transactional对其进行注释通常就足够了:

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

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