使用定时线程终止查询

时间:2018-07-12 22:51:48

标签: python mysql timer execute

我想杀死一个带有时间线程的正在运行的查询。我尝试了不同的实现,但似乎没有任何效果。

现在,我调用函数executeTimedQuery,该函数从一个十秒的计时器开始,如果查询花费的时间太长,该计时器将被执行。在开始时,self.cursor.execute(self.query)开始运行,这是我想在10秒后终止的长时间运行的查询。由于某些原因,即使启动了计时器中的功能,游标仍在运行长时间运行的查询,并且不会运行KILL CONNECTION {pId};。它会等待较长的查询完成,然后才终止连接。

我在这里做错了什么,最好的解决方法是什么?我现在非常渴望解决方案。 :/

到目前为止,我有以下代码:

def killDBConnection(self, processId):
  '''
  This takes in a processId, which is the mysql cursor connection process 
  id.
  '''
  killQuery = 'KILL CONNECTION {pId};'.format(pId=processId)
  try:
    self.cursor.execute(killQuery)
  except:
    print('Testing')
  return 'temp'


def executeTimedQuery(self, ret):
  '''
  This takes in a dictionary.
  '''
  pID =  self.getProcessId()
  try:
    Timer(10.0, self.killDBConnection, [pID]).start()
    self.cursor.execute(self.query)
    ret['executable'] = True
    ret['data'] = self.cursor.fetchall()
    ret['headers'] = tuple([col[0] for col in self.cursor.description])
    t.cancel()
  except:
    print('executeTimedQuery - Last executed query: ', self.cursor._last_executed)
  return ret

1 个答案:

答案 0 :(得分:1)

我上面的示例一直失败,因为我试图在仍在处理长时间运行的查询的同一MySQL线程上执行“ KILL PID”查询。仅在长时间运行的查询完成后才运行“ KILL PID”,这不是我想要的。简而言之,杀死长时间运行的查询的方法是打开另一个连接,以“ KILL PID”击倒长时间运行的查询。现在,在执行长时间运行的查询之前,获取正确的PID是必不可少的。

最初,我编写了一个类,该类初始化了一个连接和光标,并且可以在函数中的任何位置访问。我没有写这个,而是编写了一个函数,该函数可以在调用时为我创建连接。我为长时间运行的查询初始化了一个连接,为kill查询函数初始化了一个连接。我从长时间运行的查询中获得了进程ID,然后在一段时间后使用定时线程启动“ KILL PID”。

以下是最基本的代码:

class QueryWithTimeout(object):
  def __init__(self, db, query, totalTimeout):
    self.db = db
    self.query = query
    self.totalTimeout = totalTimeout
    ...

  def setUpConnection(self):
    connection = connections[self.db]
    cursor = connection.cursor()
    return cursor, connection

  def getProcessId(self, cursor):
    processID = 'SELECT CONNECTION_ID();'
    cursor.execute(processID)
    processIDOutput = cursor.fetchall()[0]
    return processIDOutput[0]

  def killDBConnection(self, processId, ret):
    killCursor, killConnection = self.setUpConnection()
    killQuery = 'KILL {pId};'.format(pId=processId)
    try:
      killCursor.execute(killQuery)
    except:
      print('TEMP')

  def executeTimedQuery(self, pId, cursor):
    ret = {...}
    timeoutCall = Timer(self.totalTimeout, self.killDBConnection, [pId, ret])
    timeoutCall.start()
    try:
      cursor.execute(self.query)
      #timeoutCall.cancel()
    except:
      return ret
  ret[...] = ...
  return ret

  def processTimedQuery(self):
    queryCursor, queryConnection = self.setUpConnection()
    queryProcessId = self.getProcessId(queryCursor)
    return self.executeTimedQuery(queryProcessId, queryCursor)

作为旁注,我尝试了各种尝试,但所有尝试都失败了,但这是。