在Flask中启动与停止thread线程的实现方案咨询

现有代码失效核心原因 #

Python本身没有提供强制终止运行中线程的原生API，所有优雅退出的逻辑都依赖线程内部主动响应终止信号，你当前代码的问题主要有三点：

• 线程实例、退出事件没有跨请求持久化：每次POST请求生成的Thread和threading.Event都是临时变量，和之前已启动的运行中线程没有关联，对临时事件调用set()完全不会作用到目标线程。
• 任务函数无退出判断逻辑：原task直接调用time.sleep()进入阻塞，全程没有检查终止信号的逻辑，就算传入了正确的事件也无法响应退出操作。
• 仅设置daemon=True仅会在Flask主进程退出时自动杀掉线程，无法在进程运行过程中主动触发终止。

修正后可运行实现 #

from flask import (
    Flask, jsonify, render_template, request
)
import threading
import time
import random

app = Flask(__name__)

# 全局持久化存储：结果对象、退出事件、运行中线程实例
thread_return = {'seconds': 0}
exit_event = threading.Event()
running_thread = None

def task(thread_return, exit_event):
    random_seconds=random.random()
    print(f'Start task {threading.current_thread().name} ')
    # 将长阻塞sleep拆分为小段轮询，每次检查退出信号
    total_sleep = random_seconds * 15
    slept_time = 0
    check_step = 0.1
    while slept_time < total_sleep:
        # 收到退出信号则提前终止
        if exit_event.is_set():
            print(f'Task {threading.current_thread().name} terminated early')
            return
        time.sleep(check_step)
        slept_time += check_step
    print(f'completed {threading.current_thread().name} ')
    thread_return['seconds']= random_seconds*5

        
@app.route("/", methods=["GET", "POST"])
def index():
    global running_thread, exit_event
    if request.method == "GET":
        return render_template("index-ajax.html")

    val = request.json.get("c_check")
    if val==1:
        print("val1 is ", val)
        # 避免重复启动线程
        if running_thread and running_thread.is_alive():
            return jsonify({"data": {"val": val, "msg": "thread is already running"}})
        # 启动新线程前重置退出事件
        exit_event.clear()
        running_thread = threading.Thread(target=task, args=(thread_return, exit_event), daemon=True)
        running_thread.start()
    elif val==0:
        print('trying to kill thread')
        if running_thread and running_thread.is_alive():
            # 触发退出信号
            exit_event.set()
            # 可选：最多等待1秒等待线程优雅退出，避免阻塞请求
            running_thread.join(timeout=1)
            print("thread terminated")
        else:
            print("no running thread to kill")
        print("val0 is ", val)
    
    return jsonify({"data": {"val": val}})
    
if __name__ == "__main__":
    # 测试时关闭debug模式，避免双进程导致全局变量重复初始化
    app.run(host="127.0.0.1", port=5000, debug=False)

注意事项 #

• 如果需要支持同时运行多个线程，可以用字典存储线程ID与对应退出事件、结果对象的映射关系，终止时传入目标线程ID匹配操作即可。
• 实际业务中的长耗时任务建议拆分为多段短执行逻辑，每段执行完成后检查一次退出事件，保证终止操作的响应速度，避免出现长时间无检查的阻塞逻辑。
• 生产环境如果使用多进程部署Flask，全局变量不会在进程间共享，需要用Redis等外部存储做任务状态、终止信号的同步，或者改用Celery等异步任务框架管理后台任务。

内容的提问来源于stack exchange，提问作者vt2424253