Mysql
 sql >> Base de données >  >> RDS >> Mysql

Délai d'expiration de la requête MySQLDB en python

La solution tordue de @nosklo est élégant et réalisable, mais si vous voulez éviter la dépendance à twisted, la tâche est toujours faisable, par exemple :

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()