4、连接Level2行情WebSocket：协议基础、鉴权与心跳机制

好，咱们进入实战环节了。前面几章讲了Level2的数据结构、文件格式，但那些都是「静态」的东西。真正要拿到实时行情，你得跟交易所的服务器建立一条长连接——说白了，就是WebSocket。

我个人习惯把WebSocket比作「电话通话」，而HTTP是「发短信」。你发短信问一句，对方回一句，挂了。但做量化交易，你需要一直在线，随时接收最新数据。这时候就得用WebSocket。

4.1 WebSocket协议基础——其实没那么玄乎

WebSocket本质上是一个建立在TCP之上的全双工通信协议。它跟HTTP的握手过程很像，但一旦握手成功，就切换到了二进制帧传输模式。

我刚开始接触时也犯过嘀咕：这不就是长轮询的升级版吗？后来踩了坑才明白——长轮询每次都要带HTTP头，开销大；WebSocket握手后，头部只有2-14个字节。做高频交易，这点差距就是胜负手。

一个典型的握手请求长这样：

GET /ws/level2 HTTP/1.1
Host: api.example.com
Upgrade: websocket
Connection: Upgrade
Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==
Sec-WebSocket-Version: 13

服务器返回101状态码，表示切换协议成功。之后，你们俩就可以愉快地互发数据了。

4.2 鉴权流程——Token还是API Key？

连接Level2行情，第一关就是鉴权。交易所不会让你白嫖数据的。常见的鉴权方式有两种：Token和API Key。

鉴权方式	原理	适用场景
Token（令牌）	先通过REST接口获取一个临时令牌，再用它连接WebSocket	大多数券商、数据商的标准做法
API Key + Secret	用密钥对请求签名，服务器验证身份	部分自研系统、私有协议

我个人更推荐Token方式。为什么？因为API Key一旦泄露，别人就能用你的额度。而Token有过期时间，就算被截获，也就几分钟的有效期。

我曾经在项目中遇到过一个问题：Token过期了，但WebSocket连接没断，结果收不到数据了。后来我加了个定时器，每30分钟重新鉴权一次，问题就解决了。

Token鉴权的典型流程

1. 调用REST接口获取Token（通常需要API Key + Secret做签名）
2. 建立WebSocket连接
3. 发送鉴权消息（包含Token）
4. 服务器返回鉴权成功/失败

代码示例（Python + websockets库）：

import asyncio
import websockets
import json

async def connect_level2():
    # 第一步：获取Token（假设已有API Key）
    token = await get_token(api_key="your_key", secret="your_secret")
    
    # 第二步：建立WebSocket连接
    async with websockets.connect("wss://api.example.com/level2") as ws:
        # 第三步：发送鉴权消息
        auth_msg = {
            "action": "auth",
            "params": {
                "token": token
            }
        }
        await ws.send(json.dumps(auth_msg))
        
        # 第四步：等待鉴权结果
        response = await ws.recv()
        print(f"鉴权结果: {response}")

asyncio.run(connect_level2())

4.3 心跳机制——别让连接断了

连接建立后，你以为就万事大吉了？太天真了。网络环境复杂，防火墙、NAT设备、运营商都可能掐断长时间空闲的连接。这时候就需要心跳机制。

心跳说白了就是：客户端定时给服务器发一个「我还活着」的信号。服务器收到后回复一个「知道了」。如果连续几次没收到回复，就认为连接断了，需要重连。

我刚开始做量化时，没重视心跳。结果有一次策略跑了一整天，突然发现数据停了——连接被运营商掐了，但程序没检测到。那天的回测数据全废了。

常见的心跳策略

• 客户端主动Ping：每隔N秒发送一个Ping帧，服务器回复Pong帧
• 服务器主动Ping：部分交易所会主动发心跳，客户端只需回复
• 应用层心跳：在业务消息中夹带一个「心跳」字段

WebSocket协议本身支持Ping/Pong帧。但很多Level2行情服务商会在应用层做心跳，因为这样更灵活——可以携带时间戳、序列号等信息。

代码示例（带心跳的客户端）：

async def level2_with_heartbeat():
    async with websockets.connect("wss://api.example.com/level2") as ws:
        # 启动心跳任务
        async def heartbeat():
            while True:
                await asyncio.sleep(30)  # 每30秒发一次
                try:
                    await ws.ping()
                except:
                    print("心跳失败，准备重连")
                    break
        
        # 同时处理数据和心跳
        heartbeat_task = asyncio.create_task(heartbeat())
        
        try:
            async for message in ws:
                # 处理行情数据
                process_level2_data(message)
        finally:
            heartbeat_task.cancel()

4.4 订阅与取消订阅——只拿你需要的

鉴权通过、心跳正常，接下来就是真正的数据订阅了。Level2行情数据量巨大，你不可能全量接收。订阅机制让你只关注感兴趣的股票。

订阅消息的格式各家略有不同，但核心逻辑一致：

{
    "action": "subscribe",
    "params": {
        "code": ["600519", "000001", "300750"],
        "type": ["tick", "order", "transaction"]
    }
}

取消订阅类似：

{
    "action": "unsubscribe",
    "params": {
        "code": ["600519"],
        "type": ["tick"]
    }
}

我个人习惯在策略启动时订阅需要的股票，收盘后取消订阅。但要注意：有些交易所对订阅次数有限制，频繁订阅/取消可能会被限流。

订阅管理的最佳实践

• 批量订阅：尽量一次性订阅多个股票，减少交互次数
• 增量更新：只订阅新增的股票，不要每次都全量重订
• 订阅确认：服务器返回订阅成功/失败后，再开始处理数据

4.5 知识体系总览

下面这张图总结了本章的核心逻辑，从建立连接到数据收发的完整链路：

嗯，这张图把整个流程串起来了。你想想看，从握手到鉴权，再到心跳和订阅，每一步都有坑。但只要按这个流程走，基本不会出大问题。

好了，这一章的内容就到这里。下一章我们会深入解析Tick数据的结构，以及如何用Python高效处理每秒上千条的行情数据。

---

公众号:蓝海数据掘金营，微信deep3321