在多个线程中执行SQL查询(HSQLDB)

时间:2014-05-15 08:26:29

标签: java sql multithreading jdbc hsqldb

我们假设我们有一个SQL数据库(hsqldb),并希望在其上运行一些不修改内容的查询。 对于某些查询,这需要很长时间,我想在多个线程中运行查询。 所以我的问题是:实现这个的最佳方法是什么?

我没有找到任何好的样本,所以我想出了以下内容(我希望得到一些评论)。

首先,非常简短地说: 我使用线程安全集合来访问查询并将结果放入。查询在许多工作线程中执行。结果在主线程中处理,主线程检查新结果直到所有线程都完成。

现在代码:

创建查询和结果的线程安全集合:

ConcurrentLinkedQueue<String> queries = new ConcurrentLinkedQueue<String>()
ConcurrentLinkedQueue<ResultSet> sqlResults = new ConcurrentLinkedQueue<ResultSet>();

创建多个线程并启动它们(已编辑):

ExecutorService executorService = Executors.newFixedThreadPool(4);
for(int i=0; i<4; i++){
    executorService.execute(new QueryThread(sqlResults, queries));
}

在线程类QueryThread中,只要有剩余的内容,就会打开一个连接并执行查询:

private class QueryThread implements Runnable {
    private ConcurrentLinkedQueue<ResultSet> sqlResults;
    private ConcurrentLinkedQueue<String> queries;

    public QueryThread(ConcurrentLinkedQueue<ResultSet> sqlResults, ConcurrentLinkedQueue<String> queries){
        this.sqlResults = sqlResults;
        this.queries = queries;
    }

    @Override
    public void run(){
        Connection connThr = null;
        try{
            try {
                connThr = DriverManager.getConnection(dbModeSave, "sa", "");
                connThr.setAutoCommit(false);
            } catch (SQLException e1) {
                // TODO Auto-generated catch block
                e1.printStackTrace();
            }

            String currentQuery;
            do {
                currentQuery = queries.poll(); // get and remove element from remaining queries
                if (currentQuery != null) { // only continue if element was found
                    try {
                        Statement stmnt = connThr.createStatement(ResultSet.TYPE_SCROLL_INSENSITIVE,ResultSet.CONCUR_UPDATABLE);
                        try {
                            ResultSet resultSet = stmnt.executeQuery(currentQuery);
                            sqlResults.add(resultSet);
                        } catch (SQLException e) {
                            // (Do something specific)
                        } finally {
                            stmnt.close();
                        }
                    } catch (SQLException e) {
                        // (Do something specific)
                    }
                }
            } while (currentQuery != null);

        } finally {
            if (connThr != null) {
                try {
                    connThr.close();
                } catch (SQLException e) {
                    // Nothing we can do?
                }
            }
        }
    }
}

从我检查的原始线程中,如果线程全部完成,那么所有查询都被处理(编辑)。

while (!executorService.isTerminated()) {
    try {
        Thread.sleep(100);
    } catch (InterruptedException e) {
        Thread.currentThread().interrupt();
    }
    while (!sqlResults.isEmpty()) {
        ResultSet result = sqlResults.poll();
        //process result and close it in the end
    }
}

1 个答案:

答案 0 :(得分:0)

用于并行处理的Java标准解决方案是ThreadPoolExecutor。试试吧。