Java JDBC在ArrayList中存储ResultSet

时间:2017-10-05 14:52:44

标签: java sqlite jdbc arraylist

我正努力在Java中使用SQLite数据库。

(我也在使用sqlite-jdbc-3.20.0 library,但我认为这个问题不受此影响。)

我已经创建了一些我想稍后使用的功能来简化我的工作流程。 为此我创建了类 DBInterface.java ,我在 Main.java 类中创建了一个对象。

DBInterface.java 的简化内容:

/**
 * Executes a sql String and returns it's results as ResultSet[].
 *
 * @param sql as String holidng the SQL statement to execute
 * @return ResultSet[] containing the results of the query, or empty ResultSet[] if no results
 */
public ArrayList<ResultSet> excecute(String sql)
{
    if(!isConnected()) return null;
    try
    {
        return getResultSets(statement.execute(sql));
    }
    catch (SQLException e)
    {
        Main.log(tag, "An SQLException has occured for statement '"+sql+"':");
        Main.log(tag, e);
    }
    return null;
}

/**
 * Returns an array containing all resultSets of the last executed statement.
 *
 * @param isResultSet as boolean defining if the statement resulted in an result set
 * @return
 */
private ArrayList<ResultSet> getResultSets(boolean isResultSet)
{
    ArrayList<ResultSet> results = new ArrayList<>();

    try
    {
        int count = 0;
        while(true) {
            if(isResultSet)
            {
                ResultSet set = statement.getResultSet();
                results.add(set);
                /*
                while(set.next())
                {
                    //Here the information can still be retrieved
                    Main.log(tag, ""+set.getString("name")+", row:"+set.getRow());
                }
                */

            }
            else
            {
                if(statement.getUpdateCount() == -1)
                {
                    break;
                }

                Main.log(tag, "Result '"+count+"' is just a count: '"+statement.getUpdateCount()+"'");
            }

            count ++;
            isResultSet = statement.getMoreResults();
        }
    }
    catch(SQLException e)
    {
        Main.log(tag, "An SQLException has occured while processing a request:");
        Main.log(tag, e);
    }

    return results;
}

Main.java 的简化内容:

public static void main(String[] args)
{
    [...]

    dbInterface.excecute("DROP TABLE IF EXISTS Users");
    dbInterface.excecute("CREATE TABLE IF NOT EXISTS Users (id INTEGER, name STRING(20));");
    dbInterface.excecute("INSERT INTO Users (id, name) VALUES (1, \"Hans\");");
    dbInterface.excecute("INSERT INTO Users (id, name) VALUES (2, \"Peter\");");

    ArrayList<ResultSet> results = dbInterface.excecute("SELECT * FROM Users;");

    log(tag, ""+results.size());
    for(int i = 0; i < results.size(); i++)
    {
        ResultSet set = results.get(i);
        try
        {
            ///*
            while(set.next())
            {
                //TODO Here the information can no longer be retrieved... :(
                Main.log(tag, ""+set.getString("name")+", row:"+set.getRow());
            }
            //*/
        }
        catch (Exception e)
        {
            Main.log(tag, e);
        }
    }
}

根据我的理解,在某些情况下,SQL查询可能会创建多个ResultSet,但我只是将所有ResultSet对象添加到ArrayList中,所以我可以看到我何时在某处处理结果否则存在多少ResultSet。 虽然Stranegly,但这不起作用。 通过取消注释 DBInterface.java getResultSets()中的块,我可以确认结果集确实包含了这些信息。

但是,从ArrayList在 Main.java 中做同样的事情,由于某种原因我无法再访问这些信息。 我已经测试过更改函数而不是直接返回ResultSet对象,这确实有效,但它不能实现我的目标,因为我想返回所有可能的ResultSet对象,而不仅仅是一个。

2 个答案:

答案 0 :(得分:2)

您可以编写一个方法将ResultSet转换为list,map或class,并将ResultSet的结果存储在对象中

像这样的东西

public List resultSetToArrayList(ResultSet rs) throws SQLException {
    ResultSetMetaData md = rs.getMetaData();
    int columns = md.getColumnCount();
    List<Map<String, Object>> list = new ArrayList<>();
    while (rs.next()) {
        Map<String, Object> row = new HashMap<>(columns);
        for (int i = 1; i <= columns; ++i) {
            row.put(md.getColumnName(i), rs.getObject(i));
        }
        list.add(row);
    }

    return list;
}

请记住,在您不使用时关闭ResultSet,Statement和Connection

Closing ResultSet显式为垃圾收集器提供了尽可能早地重新收集内存的机会,因为ResultSet对象可能会占用大量内存,具体取决于查询。

答案 1 :(得分:1)

当然,使用Spring会更简单,但了解如何使用普通的旧JDBC非常好。

不是返回List对象,而是创建一个Java类来保存结果对象并返回列表中的对象。

您可以在专门设计用于返回这些对象的方法中执行此操作。

例如,如果您创建了User类,则可以执行以下操作:

private static String SQL = "SELECT * FROM Users";
...

private List<User> getUsers()
{
    List<User> results = new ArrayList<>();

    try
    {
        PreparedStatement statement = connection.prepareStatement(SQL);
        ResultSet rs = statement.executeQuery(SQL);
        while(rs.next()) {
          User aUser = new User();
           aUser.setUserid(rs.getString("USER_ID"));
           aUser.setUserName(rs.getString("USERNAME"));
           ...
         results.add(aUser);
        }
    }
    catch(SQLException e) {
     // log any exceptions ...
    }
    finally() {
     // always close your JDBC resources - google it
     // example
     try{
        if (statement != null) {
           statement.close();
           statement=null;
        }
      }
      catch(SQLException e) {
        // couldn't close statement
      }
      ...
    }

    return results;
}

如果您没有使用JDBC DataSource,那么您必须自己管理数据库连接 - 这可能很乏味并且是(应用程序错误的常见来源)。几乎所有现代Java应用程序都使用DataSource实现(并且池化DataSource是非常常见的)而不是简单的JDBC连接。