2023.11.1使用SELECT ... FOR UPDATE语句实现数据库的锁机制

2023.11.1使用SELECT … FOR UPDATE语句实现数据库的锁机制

为了防止两个不同的用户同时对同一个数据修改，可以使用数据库的锁机制来实现。

数据库的锁机制分为行级锁和表级锁。行级锁是指锁住一行数据，其他用户不能修改这一行数据，但是可以修改其他行的数据。表级锁是指锁住整张表，其他用户不能修改任何数据。

如果需要防止两个不同的用户同时对同一个数据修改，可以使用行级锁。具体实现方式如下：

在执行SQL语句之前，使用SELECT … FOR UPDATE语句锁住要修改的数据行。然后建立数据库连接，再执行SQL语句，最后关闭数据库连接。

例如，要修改id为1的用户的信息，可以使用以下SQL语句：

SELECT * FROM users WHERE id=1 FOR UPDATE;

这个语句会锁住id为1的用户数据行，直到事务结束或者显示地进行COMMIT或ROLLBACK，其他用户就不能修改这一行数据了。如果有其他用户尝试修改该行数据，会被阻塞直到当前事务结束。

需要注意的是，使用行级锁可能会导致性能问题和死锁问题。因此，在实际应用中，需要根据具体情况选择合适的锁机制，并进行性能测试和死锁检测。

示例代码：

from flask import Flask, request, jsonify
import pymysql

app = Flask(__name__)

# 建立数据库连接
db = pymysql.connect(host='域名', user='用户名', password='密码', db='数据库名', charset='utf8mb4')
cursor = db.cursor()

@app.route('/update', methods=['POST'])
def update_user_info():
    # 获取请求参数
    user_id = request.json.get('user_id')
    new_info = request.json.get('new_info')

    try:
        # 开始事务
        db.begin()

        # 锁定要修改的数据行
        cursor.execute("SELECT * FROM users WHERE id=%s FOR UPDATE", user_id)

        # 修改数据
        cursor.execute("UPDATE users SET info=%s WHERE id=%s", (new_info, user_id))

        # 提交事务
        db.commit()

        return jsonify({'code': 0, 'msg': '更新成功'})

    except Exception as e:
        # 回滚事务
        db.rollback()
        return jsonify({'code': -1, 'msg': '更新失败：{}'.format(str(e))})

    finally:
        # 关闭数据库连接
        cursor.close()
        db.close()

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

可以使用try…except…finally语句块来确保无论如何都会关闭数据库连接。在实际应用中，也需要注意及时关闭数据库连接，避免占用过多的资源。