前言
背景：开了两个线程操作数据库插入但是获取不到db的信息，自己摸索的方法不一定是最佳的，有更好的可以评论或私信，感谢大佬
话不多说，直接上代码
# 模型里面的多线程新增操作
@staticmethod
def add_users_by_thread(username, password, session):
user = User(username=username, password=password)
session.add(user)
@task_blueprint.route('/add_users_by_thread', methods=['POST'])
def add_users_by_thread():
print(f"db: {db}")
print(f"db.engine: {db.engine}")
print("--------------------------------------------------")
print("Submitting tasks to thread pool")
with current_app.app_context():# 这里手动推入上下文，在线程提交之前
executor.submit(add_users_one(db))# 这里需要将db传进去，不然无法获取到session，主要是在方法里面打印db无法获取到主线程的连接信息
with current_app.app_context():
executor.submit(add_users_two(db))
print("Tasks submitted")
return jsonify({'message': 'users added!'})
# add_users_one和add_users_two是类似的操作
def add_users_one(db):
print("add_users_one started!")
try:
session_factory = sessionmaker(bind=db.engine)
Session = scoped_session(session_factory)
session = Session()
for user_data in users_to_insert:
print(f"add_users_one processing: {user_data['username']}, {user_data['password']}")
User.add_users_by_thread(user_data['username'], user_data['password'], session)
print("add_users_one done!")
time.sleep(1)
session.commit()
print("add_users_one committed successfully")
except Exception as e:
session.rollback()
print(f"Error in add_users_one: {e}")
finally:
session.close()
print("Session closed")
这里之前试了在add_users_one里面开启上下文，程序不报错，但是数据不会写入，然后将上下文在线程提交之前手动推入就有报错了，然后排查是获取不到db信息，在线程提交之后不管怎么操作都拿不到db信息，所以在提交的同时将db传递到方法里面，这样就拿到了db信息
简单解释一下add_users_one过程
- 使用 sessionmaker 创建一个会话工厂 session_factory，并将数据库引擎 db.engine 绑定到该工厂
- 使用 scoped_session 包装会话工厂，确保在多线程环境中每个线程都有自己的会话实例
- 通过 Session() 实例化一个会话对象 session
- 遍历 users_to_insert 列表中的每个用户数据 user_data
- 调用 User.add_users_by_thread 方法将用户数据插入到数据库中，传入用户名、密码和会话对象 session
- 使用 time.sleep(1) 暂停 1 秒，通常用于防止过快的数据库操作或模拟延迟
- 如果在 try 块中发生异常，捕获异常并回滚事务，以确保数据库状态不会受到未成功操作的影响
- 在 finally 块中，无论是否发生异常，都关闭会话，释放资源