Python – 如何关闭 Flask 应用中的线程，结合 atexit 和 signal 模块

在 Flask 应用中，有时候需要在应用关闭时优雅地关闭正在运行的线程，以避免资源泄露或其他潜在问题。下面介绍一种结合 atexit 和 signal 模块的方法，可以在不同的关闭场景下都实现优雅的线程关闭。

使用 atexit 模块

atexit 模块允许你注册在 Python 解释器退出时执行的函数。我们可以利用这个特性来确保在 Flask 应用正常退出时关闭相关的线程。

import atexit
the_threads = []  # 假设这是存储所有线程的列表

def close_running_threads():
    for thread in the_threads:
        thread.join()
    print("Threads complete, ready to finish")

atexit.register(close_running_threads)

当 Python 解释器退出时，close_running_threads() 函数将会被调用，从而关闭所有正在运行的线程。

使用 signal 模块

除了正常退出外，我们还需要处理非正常关闭情况，比如用户使用 Ctrl+C 强制关闭服务器的情况。这时，我们可以使用 signal 模块来捕捉中断信号，并在接收到信号时执行清理工作。

import signal
import sys

def signal_handler(signal, frame):
    print('You pressed Ctrl+C! Exiting gracefully...')
    close_running_threads()
    sys.exit(0)

signal.signal(signal.SIGINT, signal_handler)

当接收到中断信号（Ctrl+C）时，signal_handler() 函数将会被调用，关闭所有正在运行的线程，并退出程序。

通过结合 atexit 和 signal 模块，可以确保在不同的关闭场景下，都能够优雅地关闭 Flask 应用中的线程。