sqlalchemy: dừng truy vấn dài hạn

Question

Tôi có tình huống dường như thẳng thắn nhưng không thể tìm thấy giải pháp thẳng về phía trước.

Tôi đang sử dụng sqlalchemy để truy vấn postgres. Nếu thời gian chờ của khách hàng xảy ra, tôi muốn dừng/hủy các truy vấn postgres đang chạy dài từ một chuỗi khác. Chủ đề có quyền truy cập vào đối tượng Session hoặc Connection.

Tại thời điểm này tôi đã cố gắng:

session.bind.raw_connection().close() 

và

session.connection().close() 

và

session.close 

và

session.transaction.close() 

Nhưng bất kể tôi thử gì, truy vấn postgres vẫn tiếp tục cho đến khi kết thúc. Tôi biết điều này từ xem pg ở đầu trang. Điều này không dễ làm sao? Tôi đang thiếu một cái gì đó? Điều này là không thể mà không nhận được pid và gửi tín hiệu dừng trực tiếp?

Answer 1

Điều này dường như làm việc tốt, cho đến nay:

def test_close_connection(self): 
    import threading 
    from psycopg2.extensions import QueryCanceledError 
    from sqlalchemy.exc import DBAPIError 

    session = Session() 
    conn = session.connection() 
    sql = self.get_raw_sql_for_long_query() 

    seconds = 5 
    t = threading.Timer(seconds, conn.connection.cancel) 
    t.start() 

    try: 
     conn.execute(sql) 
    except DBAPIError, e: 
     if type(e.orig) == QueryCanceledError: 
      print 'Long running query was cancelled.' 
    t.cancel() 

source