python如何查询数据库_如何使用python并行查询数据库

我有两个用于查询数据库的函数.假设有两个单独的查询,如何并行运行这些查询以查询相同的数据库,还等待两个结果返回,然后继续执行其余的代码？

def query1(param1, param2):

result = None

logging.info("Connecting to database...")

try:

conn = connect(host=host, port=port, database=db)

curs = conn.cursor()

curs.execute(query)

result = curs

curs.close()

conn.close()

except Exception as e:

logging.error("Unable to access database %s" % str(e))

return result

def query2(param1, param2):

result = None

logging.info("Connecting to database...")

try:

conn = connect(host=host, port=port, database=db)

curs = conn.cursor()

curs.execute(query)

result = curs

curs.close()

conn.close()

except Exception as e:

logging.error("Unable to access database %s" % str(e))

return result

最佳答案

这是一个多线程代码,可以完成您要完成的任务：

from threading import Thread, Lock

class DatabaseWorker(Thread):

__lock = Lock()

def __init__(self, db, query, result_queue):

Thread.__init__(self)

self.db = db

self.query = query

self.result_queue = result_queue

def run(self):

result = None

logging.info("Connecting to database...")

try:

conn = connect(host=host, port=port, database=self.db)

curs = conn.cursor()

curs.execute(self.query)

result = curs

curs.close()

conn.close()

except Exception as e:

logging.error("Unable to access database %s" % str(e))

self.result_queue.append(result)

delay = 1

result_queue = []

worker1 = DatabaseWorker("db1", "select something from sometable",

result_queue)

worker2 = DatabaseWorker("db1", "select something from othertable",

result_queue)

worker1.start()

worker2.start()

# Wait for the job to be done

while len(result_queue) < 2:

sleep(delay)

job_done = True

worker1.join()

worker2.join()