python3 nameko与mysql连接的RPC服务器

时间:2018-01-11 05:03:52

标签: python mysql python-3.x nameko

我尝试使用nameko rpc服务器从mysql读取数据。这是服务器代码。

class SendService:
    name = 'url_feature_rpc_service'
    def __init__(self):
        print('new connection')
        self.db = MySQLdb.connect(host="localhost", user="user", passwd="123456", db="today_news", charset='utf8')
        self.cursor = self.db.cursor()

    @rpc
    def get_feature(self, url):
        sql = 'select title_seg, entity_seg, title_entity_vec from article_feature where url_md5 = md5(\'{}\')'.format(url)
        self.cursor.execute(sql)
        result = self.cursor.fetchone()
        if result == None:
            return ''
        return '\t'.join(result)

这是客户端代码:

with ClusterRpcProxy(config) as cluster_rpc:
    for line in sys.stdin:
        line = line.strip()
        try:
            result = cluster_rpc.url_feature_rpc_service.get_feature(line)
        except Exception as e:
            print(e)

我的问题是每次调用rpc服务时,都会设置一个新连接。我有时会得到mysql错误(99)"无法连接mysql服务器"。我只能使用一个连接吗?

2 个答案:

答案 0 :(得分:1)

您应该使用诸如nameko-sqlalchemy之类的DependencyProvider来连接数据库。

__init__中实例化MySQL连接意味着每次RPC方法触发时都会创建一个新连接,这可能意味着你的连接已经用完了。

答案 1 :(得分:0)

您已挂起数据库连接,您需要在请求结束时关闭()数据库。

@rpc
def get_feature(self, url):
    sql = 'select title_seg, entity_seg, title_entity_vec from article_feature where url_md5 = md5(\'{}\')'.format(url)
    self.cursor.execute(sql)
    result = self.cursor.fetchone()
    self.db.close()
    if result == None:
        return ''
    return '\t'.join(result)