flask多线程下数据库操作（简单示例）

前言

背景：开了两个线程操作数据库插入但是获取不到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 块中，无论是否发生异常，都关闭会话，释放资源