Làm thế nào để đóng kết nối SQLAlchemy trong MySQL

Question

Đây là một mẫu mã tôi muốn chạy:

for i in range(1,2000): 
    db = create_engine('mysql://root@localhost/test_database') 
    conn = db.connect() 
    #some simple data operations 
    conn.close() 
    db.dispose() 

Có cách nào chạy này mà không nhận được "Quá nhiều kết nối" lỗi từ MySQL? Tôi đã biết tôi có thể xử lý kết nối bằng cách khác hoặc có một hồ bơi kết nối. Tôi chỉ muốn hiểu làm thế nào để đúng cách đóng một kết nối từ sqlalchemy. Cảm ơn bạn trước!

Answer 1

Dưới đây là làm thế nào để viết mã một cách chính xác:

db = create_engine('mysql://root@localhost/test_database') 
for i in range(1,2000): 
    conn = db.connect() 
    #some simple data operations 
    conn.close() 
db.dispose() 

Đó là, Engine là một nhà máy cho các kết nối cũng như một hồ bơi các kết nối, không phải là kết nối riêng của mình. Khi bạn nói conn.close(), kết nối là được trả về hồ bơi kết nối trong Engine, không thực sự bị đóng.

Nếu bạn muốn kết nối được thực sự đóng cửa, đó là, không gộp lại, vô hiệu hóa tổng hợp qua NullPool:

from sqlalchemy.pool import NullPool 
db = create_engine('mysql://root@localhost/test_database', poolclass=NullPool) 

Với Engine cấu hình trên, mỗi cuộc gọi đến conn.close() sẽ đóng kết nối DBAPI tiềm ẩn.

Nếu OTOH bạn thực sự muốn kết nối với cơ sở dữ liệu khác nhau trên mỗi cuộc gọi, có nghĩa là, hardcoded bạn "localhost/test_database" chỉ là một ví dụ và bạn thực sự có rất nhiều cơ sở dữ liệu khác nhau, sau đó phương pháp sử dụng dispose() là tốt; nó sẽ đóng tất cả các kết nối không được kiểm tra từ hồ bơi.

Trong tất cả các trường hợp trên, điều quan trọng là đối tượng Connection bị đóng qua close(). Nếu bạn đang sử dụng bất kỳ loại thực thi "không kết nối" nào, tức là engine.execute() hoặc statement.execute(), đối tượng ResultProxy được trả về từ lệnh thực thi đó phải được đọc đầy đủ hoặc được đóng một cách rõ ràng qua close(). Một số Connection hoặc ResultProxy vẫn mở sẽ cấm các phương pháp NullPool hoặc dispose() đóng mọi kết nối cuối cùng.

Answer 2

Cố gắng tìm ra giải pháp để ngắt kết nối khỏi cơ sở dữ liệu cho một vấn đề không liên quan (phải ngắt kết nối trước khi tắt).

Bạn cần phải invalidate kết nối từ Kết nối với Hồ bơi.

Trong ví dụ của bạn:

for i in range(1,2000): 
    db = create_engine('mysql://root@localhost/test_database') 
    conn = db.connect() 
    # some simple data operations 
    # session.close() if needed 
    conn.invalidate() 
    db.dispose()