Twisted LoopingCall具有阻塞功能

时间:2014-11-06 23:41:53

标签: python multithreading twisted

我有一个应用程序需要轮询数据库以查找可能的配置更改。该应用程序是一个使用Twisted的简单xmlrpc服务器。我已经尝试使用Twisted的LoopingCall来执行轮询,但是因为LoopingCall在主线程上运行,所以对db的调用是阻塞的。因此,如果db调用因某种原因而变慢,则对xmlrpc服务器的请求必须等待。所以我尝试在一个线程中运行LoopingCall并且无法真正实现它。我的问题是,我应该在一个线程中运行它吗?如果是这样,怎么样?

from twisted.web import xmlrpc, server
from twisted.internet.threads import deferToThread
from twisted.internet import reactor, task
import platform
from time import sleep

r = reactor

class Agent(xmlrpc.XMLRPC):
    self.routine()
    xmlrpc.XMLRPC.__init__(self)

    def xmlrpc_echo(self, x):
        """
        Return arg as a simple test that the server is running
        """
        return x

    def register(self):
        """
        Register Agent with db and pick up config
        """
        sleep(3)  # simulate slow db call
        print 'registered with db'

    def routine(self):
        looping_register = task.LoopingCall(self.register)
        looping_register.start(7.0, True)

if __name__ == '__main__':
    r.listenTCP(7081, server.Site(Agent()))
    print 'Agent is running on "%s"' % platform.node()
    r.run()

1 个答案:

答案 0 :(得分:1)

您应该使用twisted.enterprise.adbapi模块。它将为所有兼容DBAPI 2.0的客户端提供非阻塞api,方法是在线程池中运行它们并返回标准Deferred给你:

from twisted.enterprise import adbapi


dbpool = adbapi.ConnectionPool('psycopg2', 'mydb', 'andrew', 'password') # replace psycopg2 with your db client name.

def getAge(user):
    return dbpool.runQuery('SELECT age FROM users WHERE name = ?', user)

def printResult(l):
    if l:
        print l[0][0], "years old"
    else:
        print "No such user"

getAge("joe").addCallback(printResult)

有关更多信息和示例,请访问官方文档:https://twistedmatrix.com/documents/14.0.0/core/howto/rdbms.html