Flask資料庫連線池

wellplayed發表於2024-03-06
Flask資料庫

flask中使用mysql方式

settings檔案中配置

DEBUG=True
SECRET_KEY='asdfasdffasd'

MYSQL_HOST='127.0.0.1'
MYSQL_PORT=3306
MYSQL_USER='root'
MYSQL_PASSWORD='root'
MYSQL_DATABASE='cnblogs'

運算元據庫

from flask import Flask, session, jsonify

app = Flask(__name__)
app.config.from_pyfile('./settings.py')

# pymysql 操作mysql
from pymysql import Connect
import pymysql

conn = Connect(user=app.config.get('MYSQL_USER'),
               password=app.config.get('MYSQL_PASSWORD'),
               host=app.config.get('MYSQL_HOST'),
               database=app.config.get('MYSQL_DATABASE'), )
# pymysql.cursors.DictCursor 查出的資料是列表套字典形式
cursor = conn.cursor(pymysql.cursors.DictCursor)

# cursor = conn.cursor() 查出的資料是列表套列表形式


@app.route('/articles')
def articles():
    # 書寫SQL語句
    cursor.execute('select id,title,author from article limit 10')
    # 獲取SQL語句執行結果
    article_list = cursor.fetchall()
    return jsonify(article_list)


if __name__ == '__main__':
    app.run()

使用全域性conn和cursor可能會出現的問題

# 上面的  conn和cursor 都是全域性的
# 假設極端情況：同時併發兩個使用者
	-一個使用者查詢所有文章
    -一個使用者查詢所有使用者
    
# 就會出現，第一個執行緒拿著cursor執行了：
cursor.execute('select id,title,author from article limit 10')
# 然後第二個執行緒拿著 cursor 執行了：
 cursor.execute('select id,name from user limit 10')
    
# 第一個執行緒開始執行：（用的全是同一個cursor）
article_list = cursor.fetchall()

# 就會出現查詢article的cursor取出來的資料是  使用者相關資料---》出現資料錯亂

解決辦法

# 每個人：執行緒用自己的 conn和cursor，在檢視函式中，拿到連結和cursor
@app.route('/articles')
def articles():
    conn = Connect(user=app.config.get('MYSQL_USER'),
                   password=app.config.get('MYSQL_PASSWORD'),
                   host=app.config.get('MYSQL_HOST'),
                   database=app.config.get('MYSQL_DATABASE'), )
    # pymysql.cursors.DictCursor 查出的資料是列表套字典形式
    cursor = conn.cursor(pymysql.cursors.DictCursor)
    cursor.execute('select id,title,author from article limit 10')

    article_list = cursor.fetchall()
    return jsonify(article_list)


# 如果併發量過高---》就會出現連線數過多的問題-->mysql效能降低

使用資料庫連線池操作mysql

第三方資料庫連線池

### 1 安裝 pip install dbutils
###2 使用：例項化得到一個池物件---》池是單例
from dbutils.pooled_db import PooledDB
import pymysql

POOL = PooledDB(
    creator=pymysql,  # 使用連結資料庫的模組
    maxconnections=6,  # 連線池允許的最大連線數，0和None表示不限制連線數
    mincached=2,  # 初始化時，連結池中至少建立的空閒的連結，0表示不建立
    maxcached=5,  # 連結池中最多閒置的連結，0和None不限制
    maxshared=3,
    # 連結池中最多共享的連結數量，0和None表示全部共享。PS: 無用，因為pymysql和MySQLdb等模組的 threadsafety都為1，所有值無論設定為多少，_maxcached永遠為0，所以永遠是所有連結都共享。
    blocking=True,  # 連線池中如果沒有可用連線後，是否阻塞等待。True，等待；False，不等待然後報錯
    maxusage=None,  # 一個連結最多被重複使用的次數，None表示無限制
    setsession=[],  # 開始會話前執行的命令列表。如：["set datestyle to ...", "set time zone ..."]
    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='root',
    database='cnblogs',
    charset='utf8'
)

## 3 在檢視函式中匯入使用
@app.route('/article')
def article():
    conn = POOL.connection()

    cursor = conn.cursor(DictCursor)
    # 獲取10條文章
    sql = 'select id,name,url from article limit 10'
    cursor.execute(sql)
    time.sleep(1)
    # 切換
    res = cursor.fetchall()
    print(res)
    return jsonify({'code': 100, 'msg': '成功', 'result': res})

相關文章