在接收消息并相应地处理它们的循环上工作,基本上是一个带有保活和身份验证的 websocket echo-er,我现在已经被困在保活部分了一段时间。
概念很简单,当服务器启动时,我创建了一个带有自动收报机的 goroutine,并初始化了一个 uint64 指针,每次该自动收报机滴答声(每 2 秒),我用 atomic.AddUint64(clockTicks, 1) 增加指针,然后对于每个 websocket 连接 goroutine,我使用比较和 atomic.LoadUint64(clockTicks) 检查每个刻度的变量,然后发送 ping/pong 消息。
编辑:似乎有什么东西阻塞了 for 循环,直到收到一条消息,结果:
i := atomic.LoadUint64(clockTicks)
if i != cur {
cur = i
if act != true {
fmt.Println("Quit Nao U filthy bot.")
return
} else {
fmt.Println("Keep Going.")
act = false
}
}
在此代码段中, i := atomic.LoadUint64(clockTicks) & 所有 if 块仅在发送 i 消息时运行(在 msg 上打印“继续前进。”),这不是我想要的,我希望该代码段每次运行用于迭代和“继续前进”。& "Quit nao ..." 触发每次clockTicks增加
这是重要的代码部分,我使用的是 Go 和 Gorilla 的 Websockets 库:
func Clock() {
clockTicks = new(uint64)
*clockTicks = 0
clock := time.NewTicker(authIntervals).C
for {
<-clock
atomic.AddUint64(clockTicks, 1)
}
}
var wsu = websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
CheckOrigin: OriginHandler,
}
func serveWS(w http.ResponseWriter, r *http.Request) {
if r.Method != "GET" {
http.Error(w, "Method not allowed", 405)
return
}
ws, err := wsu.Upgrade(w, r, nil)
if err != nil {
fmt.Println(err)
return
}
defer ws.Close()
cur := atomic.LoadUint64(clockTicks)
var act, val = true, false
for {
i := atomic.LoadUint64(clockTicks)
if i != cur { /* Only triggers when I receive a msg */
cur = i
if act != true {
fmt.Println("Quit Nao U filthy bot.")
return
} else {
fmt.Println("Keep Going.")
act = false
}
}
mtype, p, err := ws.ReadMessage()
if err != nil {
return
}
...
}
编辑 2:IRC 中有人建议 ws.ReadMessage 可能正在阻塞,但我不太确定(他说 ws.ReadMessage 实现中使用的 ioutil.ReadAll 正在阻塞它,他对此非常确定)
POPMUISE
相关分类