SQLAlchemy与多进程环境下的SSL连接错误处理指南

from sqlalchemy import create_engine

# 启用连接池的调试日志
engine = create_engine("postgresql://user:password@localhost/dbname", echo_pool="debug")

from multiprocessing import Process
from sqlalchemy import create_engine, text
from sqlalchemy.orm import sessionmaker

# 假设 db_uri 已经定义
db_uri = "postgresql://user:password@localhost/dbname"

class VMBClient:
    def upload_file(self, corp_index, filename):
        # 在子进程中创建独立的引擎和会话
        engine = create_engine(db_uri)
        Session = sessionmaker(bind=engine)
        sess = Session()
        try:
            # 执行文件上传API调用（不涉及DB）
            results = self.call_upload_file(corp_index, filename)

            # 数据库操作
            insert_command = text(f"""
                INSERT INTO corporate.vmb_items (...) VALUES (...);
            """)
            sess.execute(insert_command)

            update_command = text(f"""
                UPDATE corporate.vmb_items AS i SET child_count = (...) WHERE i.onedrive_item_id = (...);
            """)
            sess.execute(update_command)

            sess.commit()
            return results
        except Exception as e:
            sess.rollback()
            raise e
        finally:
            sess.close()
            # 确保在子进程结束时也dispose引擎
            engine.dispose()

    # 假设 call_upload_file 方法已实现
    def call_upload_file(self, *args):
        print(f"Uploading file for {args}")
        return {"id": "test_id", "parent_id": "parent_id", "name": "test_file"}


if __name__ == "__main__":
    # 主进程中初始化引擎（如果需要，否则可以省略）
    # main_engine = create_engine(db_uri)

    vmb_client = VMBClient()

    # 在派生子进程之前，确保主进程的连接池被清空
    # 如果主进程没有使用数据库，可以省略这一步。
    # 如果主进程也使用了数据库，并且在fork之后不再需要其连接，则必须调用。
    # main_engine.dispose() # 示例：如果主进程有main_engine

    # 派生子进程
    p = Process(target=vmb_client.upload_file, args=(1, "example.txt"))
    p.start()
    p.join() # 等待子进程完成
    print("Child process finished.")

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

# 在创建引擎时设置 pool_reset_on_return
# pool_reset_on_return=None 或 pool_reset_on_return=False 都可以
engine = create_engine(db_uri, pool_reset_on_return=None)
Session = sessionmaker(bind=engine)

# 后续代码与之前相同

# 示例：子进程中的 upload_file 函数
def upload_file(self, corp_index, filename):
    # 确保在子进程中创建独立的引擎
    engine = create_engine(db_uri, pool_reset_on_return=None) # 可选：根据需要调整此参数
    Session = sessionmaker(bind=engine)
    sess = Session()
    try:
        # ... 数据库操作 ...
        sess.commit()
    except Exception:
        sess.rollback()
        raise
    finally:
        sess.close()
        engine.dispose() # 关键：在子进程任务结束时释放资源