处理一个接收消息并相应地处理它们的循环,基本上是一个具有保持活动和身份验证的 websocket echo-er,我已经在保持活动部分停留了一段时间。
这个概念很简单,当服务器启动时,我用一个ticker创建一个goroutine,并初始化一个uint64指针,每次ticker滴答(每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 ..." 以触发每次时钟刻度增加
这是重要的代码部分,我正在使用 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 正在阻塞它,他对此非常确定)