Python多线程场景下如何让gemini_pdf返回最终输出结果

问题根源：当前gemini_pdf启动线程后直接返回，线程在后台异步执行，导致response无法拿到线程的执行结果。以下是几种可行的解决方式：

方法1：用队列传递结果+线程等待 #

通过队列存储线程执行的结果，启动线程后调用join()等待线程完成，再从队列取出结果返回：

import queue
import threading

def read_pdf(q):
    # 原PDF加载逻辑，最终将结果存入队列
    pdf_content = "加载完成的PDF内容"
    q.put(pdf_content)

def answer_from_pdf(message, q):
    # 原查询处理逻辑，最终将回答存入队列
    answer = "基于PDF生成的查询回答"
    q.put(answer)

# 全局变量，存储PDF内容
file_content = None

def gemini_pdf(message):
    global file_content
    result_queue = queue.Queue()
    
    if not file_content:
        # 启动加载PDF的线程
        load_thread = threading.Thread(target=read_pdf, args=(result_queue,))
        load_thread.start()
        load_thread.join()  # 等待线程执行完毕
        file_content = result_queue.get()
        return file_content
    else:
        # 启动处理查询的线程
        answer_thread = threading.Thread(target=answer_from_pdf, args=(message, result_queue))
        answer_thread.start()
        answer_thread.join()
        return result_queue.get()

方法2：用ThreadPoolExecutor简化结果获取 #

借助concurrent.futures.ThreadPoolExecutor，可以直接通过Future对象获取线程执行结果：

from concurrent.futures import ThreadPoolExecutor

def read_pdf():
    # 原PDF加载逻辑，直接返回结果
    return "加载完成的PDF内容"

def answer_from_pdf(message):
    # 原查询处理逻辑，直接返回结果
    return "基于PDF生成的查询回答"

# 全局变量，存储PDF内容
file_content = None

def gemini_pdf(message):
    global file_content
    with ThreadPoolExecutor() as executor:
        if not file_content:
            future = executor.submit(read_pdf)
            file_content = future.result()  # 阻塞等待线程完成并获取结果
            return file_content
        else:
            future = executor.submit(answer_from_pdf, message)
            return future.result()

方法3：改为同步调用（无需多线程场景） #

如果不需要并发执行其他任务，直接去掉线程逻辑，同步调用函数即可：

# 全局变量，存储PDF内容
file_content = None

def read_pdf():
    # 原PDF加载逻辑，直接返回结果
    return "加载完成的PDF内容"

def answer_from_pdf(message):
    # 原查询处理逻辑，直接返回结果
    return "基于PDF生成的查询回答"

def gemini_pdf(message):
    global file_content
    if not file_content:
        file_content = read_pdf()
        return file_content
    else:
        return answer_from_pdf(message)

注意事项 #

如果file_content是多线程共享变量，为避免竞态条件，需要加锁保护：

import threading

lock = threading.Lock()

def gemini_pdf(message):
    global file_content
    with lock:
        # 后续判断和执行逻辑
        ...

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