3

我遇到了一个问题,即必须在数据库上运行许多不同的查询(不同的返回类型、不同的列数等)。在写这篇文章时,我开始怀疑是否有一种编写辅助函数的正确方法。看起来编写一个返回 a 的函数真的很容易ResultSet

但是,由于它 a) 没有关闭连接 b) 没有关闭结果集,因此它似乎是一个可能有效但不正确的解决方案。是否有任何地方可以转储所有结果,以便可以安全返回。

(我唯一能想到的就是返回一个二维字符串数组(在将所有数据转换为字符串之后),然后将其全部转换回来)

编辑:抱歉没有写清楚,想知道是否有任何方法可以按原样存储查询结果(不需要修改它),而无需为每种可能的返回类型编写单独的方法。

二维字符串列表背后的想法是能够按原样存储查询值。

Col1 Row1 | Col2 Row1 | Col3 Row1
Col1 Row2 | Col2 Row2 | Col3 Row2

编辑 2感谢您的回复,我想我会为它写一个小解析器。

4

3 回答 3

6

您不应该返回结果集,您应该将结果集中的结果读取到某种容器对象中。ResultSet 是数据库游标的包装器,当连接关闭时它会消失。这是您可以立即阅读并关闭的内容,而不是您可以在应用程序中传递的内容。

看看spring-jdbc是怎么做的。您实现了一个 resultSetMapper,它被传递给 JdbcTemplate 上的方法。

几个观察:

  • 你不需要使用 Spring 来使用 spring-jdbc。但是,我认为自己重新实现这些东西的价值很小。

  • 这不是读取 ResultSet 以打开和关闭连接的代码的工作,它需要在其他地方。

于 2013-06-13T14:18:52.283 回答
5

我建议查看 Spring JDBC。不要自己写这样的东西。它已经完成了,而且非常好。

例如,我不喜欢您返回 a Listof的想法Strings。这样你会丢失很多信息。我会返回一个MapLists列视图)或ListMaps行视图)。

如果必须,这里有一些数据库实用程序可以帮助您入门。

package persistence;

import java.sql.*;
import java.util.*;

/**
 * util.DatabaseUtils
 * User: Michael
 * Date: Aug 17, 2010
 * Time: 7:58:02 PM
 */
public class DatabaseUtils {
/*
    private static final String DEFAULT_DRIVER = "oracle.jdbc.driver.OracleDriver";
    private static final String DEFAULT_URL = "jdbc:oracle:thin:@host:1521:database";
    private static final String DEFAULT_USERNAME = "username";
    private static final String DEFAULT_PASSWORD = "password";
*/
/*
    private static final String DEFAULT_DRIVER = "org.postgresql.Driver";
    private static final String DEFAULT_URL = "jdbc:postgresql://localhost:5432/party";
    private static final String DEFAULT_USERNAME = "pgsuper";
    private static final String DEFAULT_PASSWORD = "pgsuper";
*/
    private static final String DEFAULT_DRIVER = "com.mysql.jdbc.Driver";
    private static final String DEFAULT_URL = "jdbc:mysql://localhost:3306/party";
    private static final String DEFAULT_USERNAME = "party";
    private static final String DEFAULT_PASSWORD = "party";

    public static void main(String[] args) {
        long begTime = System.currentTimeMillis();

        String driver = ((args.length > 0) ? args[0] : DEFAULT_DRIVER);
        String url = ((args.length > 1) ? args[1] : DEFAULT_URL);
        String username = ((args.length > 2) ? args[2] : DEFAULT_USERNAME);
        String password = ((args.length > 3) ? args[3] : DEFAULT_PASSWORD);

        Connection connection = null;

        try {
            connection = createConnection(driver, url, username, password);
            DatabaseMetaData meta = connection.getMetaData();
            System.out.println(meta.getDatabaseProductName());
            System.out.println(meta.getDatabaseProductVersion());

            String sqlQuery = "SELECT PERSON_ID, FIRST_NAME, LAST_NAME FROM PERSON ORDER BY LAST_NAME";
            System.out.println("before insert: " + query(connection, sqlQuery, Collections.EMPTY_LIST));

            connection.setAutoCommit(false);
            String sqlUpdate = "INSERT INTO PERSON(FIRST_NAME, LAST_NAME) VALUES(?,?)";
            List parameters = Arrays.asList("Foo", "Bar");
            int numRowsUpdated = update(connection, sqlUpdate, parameters);
            connection.commit();

            System.out.println("# rows inserted: " + numRowsUpdated);
            System.out.println("after insert: " + query(connection, sqlQuery, Collections.EMPTY_LIST));
        } catch (Exception e) {
            rollback(connection);
            e.printStackTrace();
        } finally {
            close(connection);
            long endTime = System.currentTimeMillis();
            System.out.println("wall time: " + (endTime - begTime) + " ms");
        }
    }

    public static Connection createConnection(String driver, String url, String username, String password) throws ClassNotFoundException, SQLException {
        Class.forName(driver);
        if ((username == null) || (password == null) || (username.trim().length() == 0) || (password.trim().length() == 0)) {
            return DriverManager.getConnection(url);
        } else {
            return DriverManager.getConnection(url, username, password);
        }
    }

    public static void close(Connection connection) {
        try {
            if (connection != null) {
                connection.close();
            }
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }


    public static void close(Statement st) {
        try {
            if (st != null) {
                st.close();
            }
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }

    public static void close(ResultSet rs) {
        try {
            if (rs != null) {
                rs.close();
            }
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }

    public static void rollback(Connection connection) {
        try {
            if (connection != null) {
                connection.rollback();
            }
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }

    public static List<Map<String, Object>> map(ResultSet rs) throws SQLException {
        List<Map<String, Object>> results = new ArrayList<Map<String, Object>>();
        try {
            if (rs != null) {
                ResultSetMetaData meta = rs.getMetaData();
                int numColumns = meta.getColumnCount();
                while (rs.next()) {
                    Map<String, Object> row = new HashMap<String, Object>();
                    for (int i = 1; i <= numColumns; ++i) {
                        String name = meta.getColumnName(i);
                        Object value = rs.getObject(i);
                        row.put(name, value);
                    }
                    results.add(row);
                }
            }
        } finally {
            close(rs);
        }
        return results;
    }

    public static List<Map<String, Object>> query(Connection connection, String sql, List<Object> parameters) throws SQLException {
        List<Map<String, Object>> results = null;
        PreparedStatement ps = null;
        ResultSet rs = null;
        try {
            ps = connection.prepareStatement(sql);

            int i = 0;
            for (Object parameter : parameters) {
                ps.setObject(++i, parameter);
            }
            rs = ps.executeQuery();
            results = map(rs);
        } finally {
            close(rs);
            close(ps);
        }
        return results;
    }

    public static int update(Connection connection, String sql, List<Object> parameters) throws SQLException {
        int numRowsUpdated = 0;
        PreparedStatement ps = null;
        try {
            ps = connection.prepareStatement(sql);

            int i = 0;
            for (Object parameter : parameters) {
                ps.setObject(++i, parameter);
            }
            numRowsUpdated = ps.executeUpdate();
        } finally {
            close(ps);
        }
        return numRowsUpdated;
    }
}
于 2013-06-13T14:20:10.643 回答
3

您可以编写帮助函数来解析 ResultSet 并将其转换为 ArrayList 或数组,甚至是对象的字段。例如,假设您有一个订单表,然后查询返回该表中特定用户(客户)的所有行。然后我们可以做这样的事情:

static List<Order> parseOrder(ResultSet rs) {
   ArrayList<Order> orderList = new ArrayList<>();
   while(rs.next() ) {
      Order order = new Order();
      order.setID(rs.getInt(1));
      order.setCustomerID(rs.getInt(2));
      order.setItemName(rs.getString(3));
      orderList.add(order);
  }
  return orderList;

}

简单地将结果集转换为对象数组的数组会更通用,但可能不太有用。

我会把它留给调用函数来关闭这个 ResultSet 以及可能的 PreparedStatement(或 Statement)和数据库连接。

于 2013-06-13T14:28:54.383 回答