Flask数据库连接池

Zeng / 2024-03-06 / 原文

flask中使用mysql方式

settings文件中配置

操作数据库

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性能降低