踩坑pymysql

2023-04-13 08:42:48 浏览数 (2)

前言记录

今天写了一个mysql数据库的用户新增模块,逻辑是根据用户 ID 查询用户数据,若不存在则创建新用户并返回用户数据,数据库是保持常连接状态,测试的时候发现无法读取最新数据,一直读取第一次的数据,然后就百度发现emmm,还真有志同道合的

相关文档: 点击跳转

解决方法我比较懒,大部分都是复制粘贴不要介意 ::(真棒)

代码如下

代码语言:javascript复制
def select(user_id):
    """
    根据用户 ID 查询用户数据,若不存在则创建新用户并返回用户数据
    """
    cursor = config.db.cursor()
    try:
        # 查询用户数据
        sql = f"SELECT * FROM `{table}` WHERE `pd_id` = {user_id}"
        cursor.execute(sql)
        results = cursor.fetchall()

        # 判断用户是否存在
        if len(results) == 1:
            # 存在则返回用户数据
            user_data = results
            print("存在")
        else:
            print("不存在")
            # 关闭游标
            cursor.close()
            cursor = config.db.cursor()
            # 不存在则创建新用户
            sql = f"INSERT IGNORE INTO {table}(pd_id) values(%s)"
            data = [user_id]
            cursor.executemany(sql, data)
            # 写入
            config.db.commit()
            # 再次查询用户数据
            sql = f"SELECT * FROM `{table}` WHERE `pd_id` = {user_id}"
            cursor.execute(sql)
            results = cursor.fetchall()
            if len(results) == 1:
                user_data = results
            else:
                raise Exception(f"Failed to create user with ID {user_id}")

        # 构造用户对象并返回
        user = User(user_data)
        cursor.close()
        return user

    except Exception as e:
        cursor.close()
        raise e

问题所在

我是直接复制的咳咳

原因在于:

  1. Mysql的存储引擎InooDB的事务隔离级别默认是 可重复读(Repeatable Read),例如A客户端事务未提交,而B客户端事务修改了数据,A客户端只能读取到小于等于当前事务版本号的数据(快照读),所以只有提交完事务后,开启新的事务中才能读取到新的数据。
  2. PyMysql模块的连接对象默认是没有自动提交事务的,需要我们用一个commite()方法才能提交,不像我们在MySQL客户端中,每次select,update,delete都帮我们自动提交事务,所以只要我们手动提交了事务,再重新select就可以查到新的数据。

解决方法

有两个解决方法,1个是每次执行完都手动提交一次,2是加个autocommit=1,我选择的是每次读取的时候就手动提交一次

自动提交示例:

con = connect(host = ‘localhost’, port = 3306, database = ‘smarthome’, user = ‘root’, password = ‘123123’, charset = ‘utf8’, autocommit = 1) 这样就不用每次查数据都手动结束事务,但是修改数据要小心,因为自动提交,没有commit也会生效。

手动提交示例:

0 人点赞