一个简单的几乎功能性的例子,给你一个想法
var c = make(chan *websocket.Conn, 5) //5 is an arbitrary buffer size
var c2 = make(chan []byte, 5)
func EchoServer(ws *websocket.Conn) {
buff := make([]byte, 256)
c <- ws
for size, e := ws.Read(buff); e == nil; size, e = ws.Read(buff) {
c2 <- buff[0:size]
}
ws.Close()
}
func main() {
go func() {
var somekindofstorage
for {
select {
case newC := <-c:
somekindofstorage.Add(newC)
case msg := <-c2:
for _, v := range somekindofstorage {
if _, e := v.Write(msg); e != nil { //assuming the client disconnected on write errors
somekindofstorage.Remove(v)
}
}
}
}
}()
http.Handle("/echo", websocket.Handler(EchoServer));
http.ListenAndServe(":12345", nil);
}
这将启动一个监听两个通道的 goroutine,一个用于添加新连接,另一个用于向所有活动连接发送消息。somekindofstorage
可以是地图或矢量。
编辑:
或者,您可以将所有连接存储在全局映射中并写入每个 from EchoServer
。但是地图的设计不是为了同时访问。