Python：上下文管理器内启动线程的行为及上下文延续方法咨询

你的流程假设是否正确？ #

完全正确。上下文管理器的__exit__方法会在代码块执行完毕后立即触发，对于socket.socket来说，__exit__会直接关闭套接字。你启动的是守护线程，主线程退出上下文块后会立刻关闭套接字，此时守护线程仍在循环调用sendall，必然会因套接字已关闭抛出异常（比如OSError: [Errno 9] Bad file descriptor）。

如何让线程延续上下文资源的使用？ #

核心思路是让上下文管理器的生命周期覆盖线程的运行周期，或者手动控制资源的关闭时机，确保线程完成工作后再释放资源。以下是两种可行方案：

方案1：等待线程结束后再退出上下文块 #

通过事件标志控制线程退出，让主线程等待线程完成后再触发上下文管理器的资源释放：

import socket
import threading
import time
import signal

# 控制线程退出的信号标志
stop_event = threading.Event()

def stream(message, socket_):
    while not stop_event.is_set():
        try:
            socket_.sendall(message)
            time.sleep(1)
        except OSError as e:
            print(f"发送失败: {e}")
            break

address = "目标IP地址"
port = 目标端口号

with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as sock:
    sock.connect((address, port))
    thread = threading.Thread(target=stream, args=("hello world!", sock), daemon=True)
    thread.start()
    
    # 主线程等待外部中断（比如Ctrl+C）
    try:
        signal.pause()
    except KeyboardInterrupt:
        print("收到中断信号，停止线程")
        stop_event.set()
        thread.join()  # 等待线程完全退出

方案2：手动管理资源，脱离上下文管理器自动关闭 #

如果需要线程长期运行，可以手动创建套接字，在所有线程完成后再手动关闭资源：

import socket
import threading
import time
import signal

stop_event = threading.Event()

def stream(message, socket_):
    while not stop_event.is_set():
        try:
            socket_.sendall(message)
            time.sleep(1)
        except OSError as e:
            print(f"发送失败: {e}")
            break

address = "目标IP地址"
port = 目标端口号

sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
try:
    sock.connect((address, port))
    thread = threading.Thread(target=stream, args=("hello world!", sock))
    thread.start()
    
    try:
        signal.pause()
    except KeyboardInterrupt:
        print("收到中断信号，停止线程")
        stop_event.set()
        thread.join()
finally:
    sock.close()  # 确保所有线程结束后再关闭套接字

关键注意点 #

• 守护线程会随主线程强制终止，若希望线程完成当前任务再退出，不要设置daemon=True，或确保主线程通过join()等待线程结束。
• 多线程操作套接字时，虽然sendall多数场景下线程安全，但复杂业务场景建议通过锁机制保证操作的原子性。

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