ws的connect的时候，如何绑定参数

[1426919587]

比如ws的连接url的时候
ws://your-server.com/ws?jwttoken=sjdkkl.jskj.sjkjk'
这个在连接的时候，我通过http的req的
query := req.URL.Query()
token := query.Get("token")
获取到token，然后绑定到当前的连接

if !validateToken(userId, token) {
zlog.Warn("token 无效，关闭连接")
conn.Stop()
return
}

//然后绑定在当前的conn上
conn.SetProperty("userId", userId)

但是我看源码里，是没有实现这种逻辑的，因为http升级之后，才会生成conn，生成conn之后，就没有前面的req的信息，
请问这个要怎么实现

[aceld]

@1426919587

你说的“Upgrade 之后才生成 conn，所以拿不到 *http.Request”这个点，在 最新 master 里已经考虑到了：服务端 WebSocket 的握手阶段框架会把 *http.Request 传进来，并且提供了官方的鉴权 Hook。

1）在 Upgrade 前读取 query 并做鉴权（推荐）

可以用 SetWebsocketAuth：

s.SetWebsocketAuth(func(r *http.Request) error {
    q := r.URL.Query()
    token := q.Get("jwttoken") // 或 token

    // 这里校验 token，不通过返回 error，框架会直接拒绝握手(401)
    // if !validateToken(userId, token) { return errors.New("invalid token") }

    return nil
})

对应位置：znet/server.go 的 ListenWebsocketConn() 里会在 Upgrade() 之前调用这个 websocketAuth，所以这里拿得到 r.URL.Query()。

2）鉴权通过后，如何把 userId 绑定到 conn 上？

两种方式：

• 简单方式：在 OnConnStart 里从“你在握手阶段塞的信息/上下文”取出来，然后再 conn.SetProperty("userId", userId)。（WsConnection 内部会保留一份 context，用来传递握手阶段信息）
• 更规范方式：在 SetWebsocketAuth 里把解析出的 userId/claims 放进 r.Context()，后续连接对象会继承这个 context；你就能在 OnConnStart 里通过 conn.Context() 取出来再 SetProperty。

总结

• token/query 的读取与校验要放在 握手（Upgrade 前） 完成；
• 绑定到连接属性则放在 OnConnStart（或后续业务）阶段完成；
• 不需要自己维护全局 map 去兜底。

另外补充：校验失败时不需要 conn.Stop()（此时 conn 还没建立），直接在 SetWebsocketAuth 里面 return error 即可，框架会返回 401 并拒绝升级。

如需我补一个 “SetWebsocketAuth + OnConnStart 结合 context 完整示例”，你再说下你希望把 userId/claims 放在 context 的什么 key 里即可。