Python Sockets首次连接失败后切换IP重连报错问题排查

代码中的核心问题 #

1. 无效的socket地址族：socket.AF_NET是不存在的常量，IPv4网络应使用socket.AF_INET，这个错误会直接导致socket创建失败，后续连接自然无法完成。
2. 未初始化属性调用close：首次连接失败时，self.conn还未被赋值，执行self.conn.close()会触发AttributeError，中断重试流程。
3. 递归调用未传递返回值：递归调用connect()后，没有将成功建立的连接对象返回，外层调用无法获取有效连接实例。
4. SSL上下文冗余重建：每次重试都重新创建SSL上下文，实际上上下文只需初始化一次，重复创建属于不必要的开销。

修复后的完整代码 #

import socket
import ssl

class Client():
    def __init__(self):
        self.host = None
        self.error_count = 0
        # 存储主备IP列表，便于扩展更多备用地址
        self.ip_list = ['19x.x.4', '19x.x.5']
        self.port = xxxx
        self._load_certs()
        self._build_context()
        self.conn = self.connect()

    def _load_certs(self):
        self.cert = '/path/'

    def _build_context(self):
        # SSL上下文只需初始化一次
        self.context = ssl.SSLContext(ssl.PROTOCOL_TLSv1_2)
        self.context.load_verify_locations(self.cert)

    def _create_socket(self):
        # 单独抽离socket创建逻辑，每次重试生成新的socket实例
        return socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    def connect(self):
        current_ip = self.ip_list[self.error_count]
        try:
            s = self._create_socket()
            conn = self.context.wrap_socket(s, server_side=False)
            conn.connect((current_ip, self.port))
            return conn
        except Exception as ex:
            # 仅在还有备用IP时重试
            if self.error_count < len(self.ip_list) - 1:
                self.error_count += 1
                # 关闭当前失败的socket（如果已创建）
                if 'conn' in locals():
                    conn.close()
                return self.connect()
            # 所有IP尝试失败后抛出汇总错误
            raise Exception(f"所有IP连接失败，最后一次错误：{ex}") from ex

关键改进说明 #

• IP列表化管理：将主备IP存入列表，后续新增备用地址只需修改列表，无需改动重试逻辑。
• 分离socket创建与SSL上下文：SSL上下文只初始化一次，每次重试仅创建新的socket实例，避免冗余操作。
• 安全的资源释放：通过locals()检查conn是否存在，避免未初始化时调用close。
• 明确的重试边界：通过列表长度控制重试次数，避免硬编码计数，逻辑更清晰。
• 异常链保留：使用raise ... from ex保留原始异常信息，便于排查问题。

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