python 运用 代码 实现 mysql 查询数据 快速导入进 redis 数据库

mysql 查询数据 快速导入进 redis 数据库

class RedisDB(object):
    # 打开数据库
    def __init__(self):
        self.db_conn = redis.StrictRedis(host='localhost', port=6379, db=4)
        self.mysql_db = DataBase()

    # 插入数据库数据
    def insert_db(self):
 
        demo = self.mysql_db.select_from_db()  # mysql 查询语句
        try:
            p = self.db_conn.pipeline()
            for record in demo:
                name = record[0]
                p.sadd('id:', name)
            replies = p.execute()
            print(replies)
        except Exception as e:
            logging.warning('插入数据 操作失败>>>>', e)

    def insert(self, data):
        self.db_conn.sadd('id:', data)

    # 关闭数据库
    def close(self):
        self.db_conn.connection_pool.disconnect()