queue.get(block=True) while start_background_task (flask-socketio) 正在运行并执行 queue.put()

Question

我有一个与使用永不结束的后台任务队列相关的问题（连续运行以获取实时数据。

我想要达到的目标：

1. 通过 flask-socketio (eventlet) 启动服务器，
2. monkey_patch(),
3. 使用 start_background_task，从另一个实时抓取数据的文件运行 function，
4. 当这个后台任务（无限期地）运行时，通过 queue.put() 将传入数据存储在队列中，
5. 总是在这个任务运行时，从主程序观察队列中的新数据并处理它们，这里意味着 socketio.emit()。

什么有效：如果在后台任务文件中，while 循环结束（例如，while count < 100：），我的程序运行良好。 在这种情况下，我可以从主文件访问队列并发出数据。

什么不起作用：如果这个 while 循环现在是 while True:，程序在某处阻塞，我无法从主程序访问队列，因为它似乎在等待后台任务返回或停止。

所以我想我在这里遗漏了一些东西......所以如果你们可以帮助我，或者给我一些线索，那就太棒了。

这里是代码的一些相关部分：

主程序

from threading import Thread
from threading import Lock
from queue import Queue

from get_raw_program import get_raw_data

from flask import Flask, send_from_directory, Response, jsonify, request, abort
from flask_socketio import SocketIO
import eventlet
eventlet.patcher.monkey_patch(select=True, socket=True)

app = Flask(__name__, static_folder=static_folder, static_url_path='')
app.config['SECRET_KEY'] = 'secret_key'

socketio = SocketIO(app, binary=True, async_mode="eventlet", logger=True, engineio_logger=True)
thread = None
thread_lock = Lock()
data_queue = Queue()


[...]

@socketio.on('WebSocket_On')
def grab_raw_data(test):
    global thread

    with thread_lock:
        if thread is None:
            socketio.emit('raw_data', {'msg': 'Thread is None:'})
            socketio.emit('raw_data', {'msg': 'Starting Thread... '})
            thread = socketio.start_background_task(target=get_raw_data(data_queue, test['mode']))

        while True:
            if not data_queue.empty():
                data = data_queue.get(block=True, timeout=0.05)
                socketio.emit('raw_data', {'msg': data})
                socketio.sleep(0.0001)

get_raw_program.py （有效，可以从 main.py 访问队列）

def get_raw_data(data_queue, test):
    count = 0
    while count < 100:
        data.put(b'\xe5\xce\x04\x00\xfe\xd2\x04\x00')
        time.sleep(0.001)
        count += 1

get_raw_program.py （不起作用，无法从 main.py 访问队列）

def get_raw_data(data_queue, test):
    count = 0
    while True:
        data.put(b'\xe5\xce\x04\x00\xfe\xd2\x04\x00')
        time.sleep(0.001)
        count += 1

Answer 1

我尝试使用常规 Thread 而不是 start_background_task，它运行良好。 再次感谢您的帮助，非常感谢:-)