资讯详情

[pymongo]在__del__中关闭数据库连接,查询函数无法终止,程序不终止

问题

import datetime import pymongo   def get_yesterday():     """ 获得昨天的日期 """     today = datetime.date.today()     oneday = datetime.timedelta(days=1)     yesterday = today - oneday     return str(yesterday)   class MongoUtil:     def __init__(self, host, port, database, table):         self.__client = pymongo.MongoClient(host=host, port=port)         self.__table = self.__client[database][table]      def __del__(self):         print("关闭连接")         self.__client.close()      def get_top10(self, day: str):         """ # 2. 访问数据库, 取出top10 db.getCollection('tb_time_dirs').find({"day": "2021-09-09"}).sort({"visit_nums": -1}).limit(10) """         top = self.__table.find({ 
        "day": day}, { 
        "_id": 0, "ips": 0}).sort("visit_nums", -1).limit(2)
        for t in top:
            print(t)


if __name__ == '__main__':
    yester = get_yesterday()
    mu = MongoUtil("172.31.162.41", 27017, "except_dir", "tb_time_dirs")
    mu.get_top10(yester)

此时会打印出 “关闭连接” 但是

解决方法一: 手动销毁对象

import datetime
import pymongo


def get_yesterday():
    """ 获取昨天的日期 """
    today = datetime.date.today()
    oneday = datetime.timedelta(days=1)
    yesterday = today - oneday
    return str(yesterday)


class MongoUtil:
    def __init__(self, host, port, database, table):
        self.__client = pymongo.MongoClient(host=host, port=port)
        self.__table = self.__client[database][table]

    def __del__(self):
        print("关闭连接")
        self.__client.close()

    def get_top10(self, day: str):
        """ # 2. 访问数据库, 取出top10 db.getCollection('tb_time_dirs').find({"day": "2021-09-09"}).sort({"visit_nums": -1}).limit(10) """
        top = self.__table.find({ 
        "day": day}, { 
        "_id": 0, "ips": 0}).sort("visit_nums", -1).limit(2)
        for t in top:
            print(t)



if __name__ == '__main__':
    yester = get_yesterday()
    mu = MongoUtil("172.31.162.41", 27017, "except_dir", "tb_time_dirs")
    mu.get_top10(yester)
    del mu

解决方法二: 使用上下文管理器

import datetime
import pymongo


def get_yesterday():
    """ 获取昨天的日期 """
    today = datetime.date.today()
    oneday = datetime.timedelta(days=1)
    yesterday = today - oneday
    return str(yesterday)


class MongoUtil:
    def __init__(self, host, port, database, table):
        self.__client = pymongo.MongoClient(host=host, port=port)
        self.__table = self.__client[database][table]

    # def __del__(self):
    # print("__del__")
    # self.__client.close()

    def __enter__(self):
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        print('关闭数据库连接')
        self.__client.close()

    def get_top10(self, day: str):
        """ # 2. 访问数据库, 取出top10 db.getCollection('tb_time_dirs').find({"day": "2021-09-09"}).sort({"visit_nums": -1}).limit(10) """
        top = self.__table.find({ 
        "day": day}, { 
        "_id": 0, "ips": 0}).sort("visit_nums", -1).limit(2)
        for t in top:
            print(t)


if __name__ == '__main__':
    yester = get_yesterday()
    mu = MongoUtil("172.31.162.41", 27017, "except_dir", "tb_time_dirs")
    with mu:
        mu.get_top10(yester)

具体原因

不明

猜测与有关

源码的解释

    def close(self):
        """Cleanup client resources and disconnect from MongoDB. On MongoDB >= 3.6, end all server sessions created by this client by sending one or more endSessions commands. Close all sockets in the connection pools and stop the monitor threads. If this instance is used again it will be automatically re-opened and the threads restarted unless auto encryption is enabled. A client enabled with auto encryption cannot be used again after being closed; any attempt will raise :exc:`~.errors.InvalidOperation`. .. versionchanged:: 3.6 End all server sessions created by this client. """

标签: 装饰连接器

锐单商城拥有海量元器件数据手册IC替代型号,打造 电子元器件IC百科大全!

锐单商城 - 一站式电子元器件采购平台