如何通过Python Sockets处理自定义异常？listener/client跨进程异常捕获

你踩了一个很常见的坑：multiprocessing的connection只是序列化传输对象，并不会自动帮你抛出异常。你在listener端发送的MyException实例，到了client端只是一个普通的对象，不会触发except块——除非你手动把它抛出来。

问题根源 #

当你在listener里执行conn.send(e)，你只是把异常对象序列化后通过socket传了过去，client端conn.recv()拿到的是这个对象的副本，但Python不会自动把接收到的异常对象当成“正在抛出的异常”处理，所以你的except MyException代码块永远不会执行。

解决方案：手动抛出接收到的异常 #

修改client端的逻辑，判断接收到的对象是否是你自定义的异常实例，如果是，就手动raise它，这样外层的异常捕获就能生效了。

修改后的client.py代码

#!/usr/bin/env python
from multiprocessing.connection import Client
from listener import MyException

def main():
    address = ('localhost', 6000)
    conn = Client(address, authkey='secret password')
    conn.send('hello')
    
    res = conn.recv()
    # 检查收到的对象是否是MyException实例，是的话手动抛出
    if isinstance(res, MyException):
        raise res
    else:
        print res
    
    conn.send('close')
    conn.close()

if __name__ == '__main__':
    try:
        main()
    except MyException as e:
        print 'Received error!', e.message

更健壮的进阶方案：用结构化数据区分响应类型

如果你的程序需要同时处理正常响应和错误，建议用结构化的字典来传输，这样能更清晰地区分不同类型的消息，避免歧义：

修改后的listener.py（部分代码）

# ... 其他代码不变
except MyException as e:
    print 'Sending {}'.format(e.message)
    # 发送标记为错误的结构化数据
    conn.send({'type': 'error', 'exception': e})
# ... 正常响应时发送：
# conn.send({'type': 'response', 'data': 'Hello world! {}'.format(rand)})

修改后的client.py（对应逻辑）

#!/usr/bin/env python
from multiprocessing.connection import Client
from listener import MyException

def main():
    address = ('localhost', 6000)
    conn = Client(address, authkey='secret password')
    conn.send('hello')
    
    res = conn.recv()
    if res.get('type') == 'error':
        raise res['exception']
    elif res.get('type') == 'response':
        print res['data']
    
    conn.send('close')
    conn.close()

if __name__ == '__main__':
    try:
        main()
    except MyException as e:
        print 'Received error!', e.message

关键注意事项 #

1. 确保异常类两端一致：自定义的MyException必须在listener和client两端都能被正确导入，并且类定义完全相同——因为pickle序列化/反序列化依赖一致的类定义。
2. 异常的可序列化性：自定义Exception默认是可以被pickle序列化的，但如果你的异常有自定义的非标准属性，需要确保这些属性也能被正确序列化（比如不要包含不可pickle的对象，比如打开的文件句柄）。

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