jdbc如何写csv文件_Java：将JDBC结果集作为CSV流化

最新推荐文章于 2024-03-14 16:17:24 发布

dnc8371

最新推荐文章于 2024-03-14 16:17:24 发布

阅读量860

点赞数

文章标签： java csv json spring jdbc

原文链接：https://www.javacodegeeks.com/2018/12/java-streaming-jdbc-resultset-csv.html

版权

本文介绍如何在Java中将JDBC查询结果集转换为CSV格式，并流式传输数据，避免一次性加载到内存。提供了一个实现Spring ResourceOutputStreamResult的类，适用于从ResultSet生成CSV。示例展示了在JAX-RS Web服务中的应用。

摘要由CSDN通过智能技术生成

jdbc如何写csv文件

在上一篇文章中，我展示了如何将java.sql.ResultSet转换为JSON并将其流回调用方。这篇文章是关于以CSV格式流式传输。流式传输使您可以一点一点地传输数据，而不必将所有数据都加载到服务器的内存中。

例如，考虑以下ResultSet ：

+---------+-----+
| Name    | Age |
+---------+-----+
| Alice   |  20 |
| Bob     |  35 |
| Charles |  50 |
+---------+-----+

相应的CSV为：

name,age
Alice,20
Bob,35
Charles,50

以下类（在我的GitHub Repository中也可用）可用于将ResultSet转换为CSV。请注意，该类实现Spring的ResultSetExtractor ， JdbcTemplate可以使用它从ResultSet提取结果。

/**
 * Streams a ResultSet as CSV.
 */
public class StreamingCsvResultSetExtractor
                         implements ResultSetExtractor<Void> {

  private static char DELIMITER = ',';

  private final OutputStream os;

  /**
   * @param os the OutputStream to stream the CSV to
   */
  public StreamingCsvResultSetExtractor(final OutputStream os) {
    this.os = os;
  }

  @Override
  public Void extractData(final ResultSet rs) {
    try (var pw = new PrintWriter(os, true)) {
      final var rsmd = rs.getMetaData();
      final var columnCount = rsmd.getColumnCount();
      writeHeader(rsmd, columnCount, pw);
      while (rs.next()) {
        for (var i = 1; i <= columnCount; i++) {
          final var value = rs.getObject(i);
          pw.write(value == null ? "" : value.toString());
          if (i != columnCount) {
            pw.append(DELIMITER);
          }
        }
        pw.println();
      }
      pw.flush();
    } catch (final SQLException e) {
      throw new RuntimeException(e);
    }
    return null;
  }

  private static void writeHeader(final ResultSetMetaData rsmd,
      final int columnCount, final PrintWriter pw) throws SQLException {
    for (var i = 1; i <= columnCount; i++) {
      pw.write(rsmd.getColumnName(i));
      if (i != columnCount) {
        pw.append(DELIMITER);
      }
    }
    pw.println();
  }
}

要将其用于带有JAX-RS的Web服务中：

import javax.ws.rs.core.StreamingOutput;

@GET
@Path("runQuery")
@Produces("text/csv")
public StreamingOutput runQuery() {
  return new StreamingOutput() {
    @Override
    public void write(final OutputStream os)
        throws IOException, WebApplicationException {
      jdbcTemplate.query("select name, age from person",
                   new StreamingCsvResultSetExtractor(os));
    }
  };
}

相关文章：

流式传输JDBC ResultSet作为JSON