當(dāng)前位置:首頁(yè) > IT技術(shù) > 數(shù)據(jù)庫(kù) > 正文

python實(shí)現(xiàn)mysql數(shù)據(jù)庫(kù)連接池的簡(jiǎn)便方法
2022-02-14 14:08:42


原理圖python實(shí)現(xiàn)mysql數(shù)據(jù)庫(kù)連接池的簡(jiǎn)便方法_數(shù)據(jù)庫(kù)

python編程中可以使用MySQLdb進(jìn)行數(shù)據(jù)庫(kù)的連接及諸如查詢/插入/更新等操作,但是每次連接mysql數(shù)據(jù)庫(kù)請(qǐng)求時(shí),都是獨(dú)立的去請(qǐng)求訪問(wèn),相當(dāng)浪費(fèi)資源,

而且訪問(wèn)數(shù)量達(dá)到一定數(shù)量時(shí),對(duì)mysql的性能會(huì)產(chǎn)生較大的影響。

因此,實(shí)際使用中,通常會(huì)使用數(shù)據(jù)庫(kù)的連接池技術(shù),來(lái)訪問(wèn)數(shù)據(jù)庫(kù)達(dá)到資源復(fù)用的目的。

安裝數(shù)據(jù)庫(kù)連接池模塊DBUtils

pip3 install DBUtils

DBUtils是一套Python數(shù)據(jù)庫(kù)連接池包,并允許對(duì)非線程安全的數(shù)據(jù)庫(kù)接口進(jìn)行線程安全包裝。DBUtils來(lái)自Webware for Python。

DBUtils提供兩種外部接口:


  • PersistentDB :提供線程專用的數(shù)據(jù)庫(kù)連接,并自動(dòng)管理連接。
  • PooledDB :提供線程間可共享的數(shù)據(jù)庫(kù)連接,并自動(dòng)管理連接。

下面利用MySQLdb和DBUtils建立自己的mysql數(shù)據(jù)庫(kù)連接池工具包

在工程目錄下新建package命名為:dbConnecttion,并新建module命名為MySqlConn,下面是MySqlConn.py,該模塊創(chuàng)建Mysql的連接池對(duì)象,并創(chuàng)建了如查詢/插入等通用的操作方法。該部分代碼實(shí)現(xiàn)如下:

還有很多其他參數(shù)可以配置:


  • dbapi :數(shù)據(jù)庫(kù)接口
  • mincached :?jiǎn)?dòng)時(shí)開(kāi)啟的空連接數(shù)量
  • maxcached :連接池最大可用連接數(shù)量
  • maxshared :連接池最大可共享連接數(shù)量
  • maxconnections :最大允許連接數(shù)量
  • blocking :達(dá)到最大數(shù)量時(shí)是否阻塞
  • maxusage :?jiǎn)蝹€(gè)連接最大復(fù)用次數(shù)

根據(jù)自己的需要合理配置上述的資源參數(shù),以滿足自己的實(shí)際需要。

代碼:

'''
學(xué)習(xí)中遇到問(wèn)題沒(méi)人解答?小編創(chuàng)建了一個(gè)Python學(xué)習(xí)交流QQ群:778463939
尋找有志同道合的小伙伴,互幫互助,群里還有不錯(cuò)的視頻學(xué)習(xí)教程和PDF電子書(shū)!
'''
import pymysql, os, configparser
from pymysql.cursors import DictCursor
from DBUtils.PooledDB import PooledDB


class Config(object):
"""
# Config().get_content("user_information")

配置文件里面的參數(shù)
[notdbMysql]
host = 192.168.1.101
port = 3306
user = root
password = python123
"""

def __init__(self, config_filename="myProjectConfig.cnf"):
file_path = os.path.join(os.path.dirname(__file__), config_filename)
self.cf = configparser.ConfigParser()
self.cf.read(file_path)

def get_sections(self):
return self.cf.sections()

def get_options(self, section):
return self.cf.options(section)

def get_content(self, section):
result = {}
for option in self.get_options(section):
value = self.cf.get(section, option)
result[option] = int(value) if value.isdigit() else value
return result


class BasePymysqlPool(object):
def __init__(self, host, port, user, password, db_name=None):
self.db_host = host
self.db_port = int(port)
self.user = user
self.password = str(password)
self.db = db_name
self.conn = None
self.cursor = None


class MyPymysqlPool(BasePymysqlPool):
"""
MYSQL數(shù)據(jù)庫(kù)對(duì)象,負(fù)責(zé)產(chǎn)生數(shù)據(jù)庫(kù)連接 , 此類中的連接采用連接池實(shí)現(xiàn)獲取連接對(duì)象:conn = Mysql.getConn()
釋放連接對(duì)象;conn.close()或del conn
"""
# 連接池對(duì)象
__pool = None

def __init__(self, conf_name=None):
self.conf = Config().get_content(conf_name)
super(MyPymysqlPool, self).__init__(**self.conf)
# 數(shù)據(jù)庫(kù)構(gòu)造函數(shù),從連接池中取出連接,并生成操作游標(biāo)
self._conn = self.__getConn()
self._cursor = self._conn.cursor()

def __getConn(self):
"""
@summary: 靜態(tài)方法,從連接池中取出連接
@return MySQLdb.connection
"""
if MyPymysqlPool.__pool is None:
__pool = PooledDB(creator=pymysql,
mincached=1,
maxcached=20,
host=self.db_host,
port=self.db_port,
user=self.user,
passwd=self.password,
db=self.db,
use_unicode=False,
charset="utf8",
cursorclass=DictCursor)
return __pool.connection()

def getAll(self, sql, param=None):
"""
@summary: 執(zhí)行查詢,并取出所有結(jié)果集
@param sql:查詢SQL,如果有查詢條件,請(qǐng)只指定條件列表,并將條件值使用參數(shù)[param]傳遞進(jìn)來(lái)
@param param: 可選參數(shù),條件列表值(元組/列表)
@return: result list(字典對(duì)象)/boolean 查詢到的結(jié)果集
"""
if param is None:
count = self._cursor.execute(sql)
else:
count = self._cursor.execute(sql, param)
if count > 0:
result = self._cursor.fetchall()
else:
result = False
return result

def getOne(self, sql, param=None):
"""
@summary: 執(zhí)行查詢,并取出第一條
@param sql:查詢SQL,如果有查詢條件,請(qǐng)只指定條件列表,并將條件值使用參數(shù)[param]傳遞進(jìn)來(lái)
@param param: 可選參數(shù),條件列表值(元組/列表)
@return: result list/boolean 查詢到的結(jié)果集
"""
if param is None:
count = self._cursor.execute(sql)
else:
count = self._cursor.execute(sql, param)
if count > 0:
result = self._cursor.fetchone()
else:
result = False
return result

def getMany(self, sql, num, param=None):
"""
@summary: 執(zhí)行查詢,并取出num條結(jié)果
@param sql:查詢SQL,如果有查詢條件,請(qǐng)只指定條件列表,并將條件值使用參數(shù)[param]傳遞進(jìn)來(lái)
@param num:取得的結(jié)果條數(shù)
@param param: 可選參數(shù),條件列表值(元組/列表)
@return: result list/boolean 查詢到的結(jié)果集
"""
if param is None:
count = self._cursor.execute(sql)
else:
count = self._cursor.execute(sql, param)
if count > 0:
result = self._cursor.fetchmany(num)
else:
result = False
return result

def insertMany(self, sql, values):
"""
@summary: 向數(shù)據(jù)表插入多條記錄
@param sql:要插入的SQL格式
@param values:要插入的記錄數(shù)據(jù)tuple(tuple)/list[list]
@return: count 受影響的行數(shù)
"""
count = self._cursor.executemany(sql, values)
return count

def __query(self, sql, param=None):
if param is None:
count = self._cursor.execute(sql)
else:
count = self._cursor.execute(sql, param)
return count

def update(self, sql, param=None):
"""
@summary: 更新數(shù)據(jù)表記錄
@param sql: SQL格式及條件,使用(%s,%s)
@param param: 要更新的 值 tuple/list
@return: count 受影響的行數(shù)
"""
return self.__query(sql, param)

def insert(self, sql, param=None):
"""
@summary: 更新數(shù)據(jù)表記錄
@param sql: SQL格式及條件,使用(%s,%s)
@param param: 要更新的 值 tuple/list
@return: count 受影響的行數(shù)
"""
return self.__query(sql, param)

def delete(self, sql, param=None):
"""
@summary: 刪除數(shù)據(jù)表記錄
@param sql: SQL格式及條件,使用(%s,%s)
@param param: 要?jiǎng)h除的條件 值 tuple/list
@return: count 受影響的行數(shù)
"""
return self.__query(sql, param)

def begin(self):
"""
@summary: 開(kāi)啟事務(wù)
"""
self._conn.autocommit(0)

def end(self, option='commit'):
"""
@summary: 結(jié)束事務(wù)
"""
if option == 'commit':
self._conn.commit()
else:
self._conn.rollback()

def dispose(self, isEnd=1):
"""
@summary: 釋放連接池資源
"""
if isEnd == 1:
self.end('commit')
else:
self.end('rollback')
self._cursor.close()
self._conn.close()


if __name__ == '__main__':
mysql = MyPymysqlPool("notdbMysql")

sqlAll = "select * from myTest.aa;"
result = mysql.getAll(sqlAll)
print(result)

sqlAll = "select * from myTest.aa;"
result = mysql.getMany(sqlAll, 2)
print(result)

result = mysql.getOne(sqlAll)
print(result)

# mysql.insert("insert into myTest.aa set a=%s", (1))

# 釋放資源
mysql.dispose()



本文摘自 :https://blog.51cto.com/u

開(kāi)通會(huì)員,享受整站包年服務(wù)立即開(kāi)通 >