python - python3 nameko RPC服务器与mysql连接

Question

我尝试使用 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 服务器”。我可以只使用一个连接吗？

Answer 1

您应该使用 DependencyProvider（例如nameko-sqlalchemy）来连接到数据库。

在内部实例化 MySQL 连接__init__意味着您将在每次 RPC 方法触发时创建一个新连接，这可能意味着您的连接用完了。

Answer 2

您有挂起的数据库连接，您需要在请求结束时关闭（）数据库。

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