aiohttp TCP连接器新增socket_factory参数支持自定义套接字配置

在最新版本的aiohttp库中，TCP连接器(TCPConnector)新增了一个重要的功能增强——支持通过socket_factory参数自定义套接字创建过程。这一改进使得开发者能够在TCP连接建立之前对套接字进行更精细化的配置。

背景与需求

在传统的TCP连接建立过程中，客户端与服务器之间的三次握手数据包会使用系统默认的网络参数。然而，在某些特定场景下，开发者需要对这些握手数据包进行定制化配置，例如：

1. 设置IP报头的服务类型(IP_TOS/IPv6_TCLASS)字段，用于QoS优先级标记
2. 针对IPv4和IPv6连接使用不同的套接字选项
3. 在连接建立前应用特定的网络层参数

原有的tcp_sockopts参数虽然可以设置一些套接字选项，但存在两个主要限制：一是无法在握手阶段生效，二是无法根据目标地址类型(IPv4/IPv6)进行差异化配置。

技术实现方案

新引入的socket_factory参数是一个回调函数，它允许开发者在每次创建新连接时完全控制套接字的创建和配置过程。该方案借鉴了aiohappyeyeballs库(v2.5.0)中的类似实现。

socket_factory回调函数的工作流程如下：

1. 在TCP连接初始化阶段被调用
2. 接收相关参数(如地址族等)
3. 返回一个已配置好的套接字对象
4. aiohttp使用该套接字进行后续连接操作

应用场景示例

以下是一些典型的使用场景：

设置服务类型(ToS)

def socket_factory():
    sock = socket.socket()
    sock.setsockopt(socket.IPPROTO_IP, socket.IP_TOS, 0x10)  # 设置低延迟
    return sock

connector = TCPConnector(socket_factory=socket_factory)

差异化IPv4/IPv6配置

def socket_factory(family=socket.AF_INET):
    sock = socket.socket(family=family)
    if family == socket.AF_INET6:
        sock.setsockopt(socket.IPPROTO_IPV6, socket.IPV6_TCLASS, 0x20)
    else:
        sock.setsockopt(socket.IPPROTO_IP, socket.IP_TOS, 0x20)
    return sock

向后兼容性

考虑到现有代码的兼容性，新版本仍然保留了tcp_sockopts参数，但推荐新项目使用更灵活的socket_factory方案。对于简单场景，tcp_sockopts仍然可以满足基本需求，但在需要更精细控制或差异化配置时，socket_factory是更好的选择。

性能考量

由于socket_factory在每次创建连接时都会被调用，实现时应注意：

1. 避免在回调函数中进行耗时操作
2. 考虑复用预配置的套接字模板
3. 对于大规模连接场景，评估额外开销

这一改进使得aiohttp在网络层的定制能力得到了显著提升，特别适合需要精细控制网络行为的应用场景，如实时音视频、在线游戏等对网络质量敏感的应用。