Python多线程与input()函数的阻塞陷阱
在Python多线程编程中，使用input()函数读取用户输入时，可能会遇到一个常见问题：循环调用input()的线程在接收一两次输入后就停止响应，程序继续运行，但input()似乎失效了。 这并非input()函数本身的缺陷，而是多线程环境下标准输入流（stdin）管理不当导致的。
问题根源：主线程退出与stdin
让我们分析以下代码：
立即学习“Python免费学习笔记（深入）”；
import threading
def input_thread():
while True:
i = input()
threading.Thread(target=input_thread).start()
这段代码创建了一个线程，循环等待用户输入。问题在于，主线程执行完threading.Thread(...)后立即结束。 由于Python的标准输入流stdin通常与终端关联，当主线程退出时，操作系统可能会关闭或重定向stdin，导致子线程的input()无法再正常接收输入。
解决方案：保持主线程运行
为了解决这个问题，我们需要确保主线程保持运行，直到子线程结束或需要终止。 以下几种方法可以实现：
-
使用
join()方法: 主线程等待子线程结束。这需要子线程在某个条件下结束，否则主线程会一直阻塞。
import threading
import time
def input_thread():
while True:
try:
i = input()
print(f"Received: {i}") # 处理输入
except EOFError:
break # 捕获EOFError异常，优雅退出
thread = threading.Thread(target=input_thread)
thread.start()
time.sleep(10) # 运行10秒后结束
thread.join()
print("Input thread finished.")
- 使用事件标志: 主线程设置一个事件标志，子线程根据标志决定是否继续循环。
import threading
import time
stop_event = threading.Event()
def input_thread():
while not stop_event.is_set():
try:
i = input()
print(f"Received: {i}")
except EOFError:
break
thread = threading.Thread(target=input_thread)
thread.start()
time.sleep(10)
stop_event.set()
thread.join()
print("Input thread finished.")
- 使用队列通信: 子线程将输入放入队列，主线程从队列读取输入。这是一种更健壮、更适合多线程编程的方案，避免了直接操作stdin的风险。
选择哪种方案取决于具体的应用场景。 如果需要持续监听输入，且能预知终止条件，join()或事件标志是不错的选择。 如果输入处理需要在主线程进行，或者需要更复杂的线程间交互，则队列通信是更优的选择。 总而言之，避免直接在子线程中无限循环调用input()，并确保主线程的正确管理，是解决这个问题的关键。
