SQLHelper ------ python实现
阅读原文时间:2023年07月10日阅读:1

1.第一种:

import pymysql
import threading
from DBUtils.PooledDB import PooledDB

class SqlHelper(object):
    def __init__(self):
        self.pool = PooledDB(
            creator=pymysql,  # 使用链接数据库的模块
            maxconnections=6,  # 连接池允许的最大连接数,0和None表示不限制连接数
            mincached=2,  # 初始化时,链接池中至少创建的链接,0表示不创建
            blocking=True,  # 连接池中如果没有可用连接后,是否阻塞等待。True,等待;False,不等待然后报错
            ping=0,
            # ping MySQL服务端,检查是否服务可用。# 如:0 = None = never, 1 = default = whenever it is requested, 2 = when a cursor is created, 4 = when a query is executed, 7 = always
            host='127.0.0.1',
            port=3306,
            user='root',
            password='000000',
            database='flask_day1',
            charset='utf8'
        )
        # 用来获取local对象
        self.local = threading.local()

    def open(self):
           """
           获取数据库连接,游标对象
           """
        conn = self.pool.connection()
        cursor = conn.cursor()
        return conn,cursor

    def close(self,cursor,conn):
        """
        关闭游标,数据库连接
        """
        cursor.close()
        conn.close()

    def __enter__(self):
        """
        with语句自动触发,执行此方法
        """
        conn,cursor = self.open()
        rv = getattr(self.local,"stack",None)
        if not rv:
            self.local.stack = [(conn,cursor)]
        else:
            rv.append([(conn,cursor)])
            self.local.stack = rv
        return cursor

    def __exit__(self, exc_type, exc_val, exc_tb):
        """
        with语句结束时执行此方法
        """
        rv = getattr(self.local,"stack",None)
        if not rv:
            return
        conn,cursor = rv[-1]
        self.close(cursor,conn)

db = SqlHelper()



# 调用文件
import threading

def task(i):
    sql = "select * from book;"
    with db as cur:
        cur.execute(sql)
        data = cur.fetchall()
        print(i,data)

for i in range(5):
    t = threading.Thread(target=task,args=(i,))
    t.start()

说明

这个是借鉴了多线程中的threading.local()方法,当每个线程都会去获取自己对应的数据,所以在每一个线程开启时就会执行task方法,这个方法中有with这方法,这个方法是去调用上下文管理器,就需要执行这个对象中的__enter__()方法和__exit__() 方法;进来时执行__enter__()这个方法,并且需要又返回值,这个返回值就是with db as cur:中的cur的值,在结束的时候需要执行__exit__() 方法;在执行__enter__()方法的时候在里边将当前这个线程的数据写成一个字典,字典的键是固定的名字stack,字典的值是一个列表套元组形式,元组中是conn,cursor(数据库连接,游标对象);将这个保存到threading.local()实例化出来的对象中self.local,在对数据库操作完之后,with也要执行__exit__()方法,在这个方法中会使用现在的self.local对象去获取存在这个对象中的那个字典,获取对应的值,然后对对游标进行关闭,数据的连接进行关闭;

2.第二种:

import pymysql
import threading
from DBUtils.PooledDB import PooledDB
import greenlet

POOL = PooledDB(
            creator=pymysql,  # 使用链接数据库的模块
            maxconnections=6,  # 连接池允许的最大连接数,0和None表示不限制连接数
            mincached=2,  # 初始化时,链接池中至少创建的链接,0表示不创建
            blocking=True,  # 连接池中如果没有可用连接后,是否阻塞等待。True,等待;False,不等待然后报错
            ping=0,
            # ping MySQL服务端,检查是否服务可用。# 如:0 = None = never, 1 = default = whenever it is requested, 2 = when a cursor is created, 4 = when a query is executed, 7 = always
            host='127.0.0.1',
            port=3306,
            user='root',
            password='000000',
            database='flask_day1',
            charset='utf8'
        )

class SqlHelper(object):
    def __init__(self):
        self.conn = None
        self.cursor = None

    def open(self):
        conn = POOL.connection()
        cursor = conn.cursor()

        return conn,cursor

    def close(self,cursor,conn):
        cursor.close()
        conn.close()

    def __enter__(self):

        self.conn,self.cursor = self.open()
        return self.cursor

    def __exit__(self, exc_type, exc_val, exc_tb):

        self.close(self.cursor,self.conn)

调用语句:

import threading

def task(i):
    sql = "select * from book;"
    with SqlHelper() as cur:
        cur.execute(sql)
        data = cur.fetchall()
        print(i,data)

for i in range(5):

    t = threading.Thread(target=task,args=(i,))
    t.start()

说明

此方法比较简单,理解起来也比较容易,首先创建一次数据库连接池,在SqlHelper类中只是调用执行就行,不用在重复实例化对象;每次执行sql语句是实例化SqlHelper方法就行,由于每次实例化的对象不一样所以就不会有数据覆盖了;