Mysql
 sql >> Datenbank >  >> RDS >> Mysql

Timeout für Python-MySQLDB-Abfragen

die Twisted-basierte Lösung von @nosklo ist elegant und praktikabel, aber wenn Sie die Abhängigkeit von twisted vermeiden möchten, ist die Aufgabe immer noch machbar, z. B.:

import multiprocessing

def query_with_timeout(dbc, timeout, query, *a, **k):
  conn1, conn2 = multiprocessing.Pipe(False)
  subproc = multiprocessing.Process(target=do_query,
                                    args=(dbc, query, conn2)+a, 
                                    kwargs=k)
  subproc.start()
  subproc.join(timeout)
  if conn1.poll():
    return conn1.recv()
  subproc.terminate()
  raise TimeoutError("Query %r ran for >%r" % (query, timeout))

def do_query(dbc, query, conn, *a, **k):
  cu = dbc.cursor()
  cu.execute(query, *a, **k)
  return cu.fetchall()