是否可以在 java 中传递结果集？

Question

我正在尝试创建一个命令，我可以将不同的查询发送到 SQLite / Mysql DB，并将结果集返回给正在调用的任何函数。需要能够处理2列还是15列。

以下不起作用 - 大概是因为它关闭了结果集/连接，但我不知道该怎么做。

想法？

public static ResultSet queryDB(String query) {
    try {
        Connection connection = DriverManager.getConnection("jdbc:sqlite:" + Settings.SQLITE_DB_PATH);
        Statement statement = connection.createStatement();
        ResultSet resultSet = statement.executeQuery(query);
        resultSet.close();
        statement.close();
        connection.close();
        return resultSet;
    } catch (SQLException ex) {
        Logger.getLogger(SQLInterp.class.getName()).log(Level.SEVERE, null, ex);
    }
    return null;
}

Answer 1

你基本上有3个选择：

1. 不要关闭方法中的ResultSet、Statement和Connection，将执行此操作的责任交给调用者。

   不推荐，因为它容易出错，并且会破坏格式良好的代码结构范例。

2. 传入具有处理数据所需逻辑的对象，如suggested by Jacob G.。

   例如使用 Java 8+ Consumer:

   public static void queryDB(String query, Consumer<ResultSet> processor) {
       try (
           Connection connection = DriverManager.getConnection("jdbc:sqlite:" + Settings.SQLITE_DB_PATH);
           Statement statement = connection.createStatement();
           ResultSet resultSet = statement.executeQuery(query);
       ) {
           processor.accept(resultSet);
       } catch (SQLException ex) {
           Logger.getLogger(SQLInterp.class.getName()).log(Level.SEVERE, null, ex);
       }
   }
   

   然后这样称呼它：

   SQLInterp.queryDB("SELECT * FROM foo", rs -> {
       while (rs.next()) {
           // process data here
       }
   });
   

3. 以通用数据结构将所有数据读入内存，例如List<Map<String, Object>>:

   这当然假设查询对每一列都有很好的唯一名称。

   public static List<Map<String, Object>> queryDB2(String query) {
       try (
           Connection connection = DriverManager.getConnection("jdbc:sqlite:" + Settings.SQLITE_DB_PATH);
           Statement statement = connection.createStatement();
           ResultSet resultSet = statement.executeQuery(query);
       ) {
           ResultSetMetaData metaData = resultSet.getMetaData();
           int columnCount = metaData.getColumnCount();
           String[] name = new String[columnCount];
           for (int i = 0; i < columnCount; i++)
               name[i] = metaData.getColumnLabel(i + 1);
           List<Map<String, Object>> rows = new ArrayList<>();
           while (resultSet.next()) {
               Map<String, Object> row = new LinkedHashMap<>();
               for (int i = 0; i < columnCount; i++)
                   row.put(name[i], resultSet.getObject(i + 1));
               rows.add(row);
           }
           return rows;
       } catch (SQLException ex) {
           Logger.getLogger(SQLInterp.class.getName()).log(Level.SEVERE, null, ex);
       }
   }