websocket介绍
The WebSocket Protocol RFC6455，这个是WebSocket的RFC文档，所以内容非常全面（当然只涉及协议，不涉及具体实现），不过内容太多，如果是初次了解，可以挑自己感兴趣的看看。
《WebSocket 是什么原理？为什么可以实现持久连接？》，这个是知乎上的一篇文章，对于WebSocket讲的通俗易懂，可以了解一下。
在介绍WebSocket之前，我们先来了解一下为什么会出现WebSocket。我们知道HTTP协议（底层使用的是TCP协议）是无状态的，即一次Request，一个Response就结束了。实际中的场景就是客户端（比如浏览器）向服务器发送一次请求，然后服务器返回一个响应，一次交互就结束了，底层的TCP连接也会断掉，下次请求时，重新再创建新的连接。而且这种通信是被动式的，就是说服务器端不能主动向客户端发响应，只能是客户端一个Request，服务的一个Response这种模式（当然最新的协议里面，可能可以将多个Request合并一次发给服务端，但模型仍旧是这种模式）。
如果你曾经使用TCP协议写过通信程序，应该非常熟悉那种模式：客户端和服务端（有时都没有清晰的界限）通过三步握手建立连接后，就可以相互随便发送数据，除非网络异常或者主动关闭，否则该TCP连接将一直存在。而WebSocket的出现就是为了在应用层实现类似传输层的TCP协议，当然它底层和HTTP一样使用的是TCP协议。这样我们就明白一些了，WebSocket不像HTTP协议，一次交互后就结束，它建立后会一直存在（除非主动断开或者网络异常等），而且客户端和服务端可以任意向对方发送数据，不再像以前那么“傻”了。也就是说，HTTP协议是一次性的，“单工的”；而WebSocket是真正意义上的长连接，且是全双工的。当然，上述提及的需求HTTP通过poll和轮循等方式也可以实现，但弊端非常多：
- 服务器端需要在底层为每个HTTP连接维护一个TCP连接，比如一个用于发送消息，一个用于接收消息等。
- 资源浪费，每次的HTTP请求中都需要携带消息头。
- 客户端还必须通过一些手段知道哪些响应对应发出去的哪些请求。
go websocket
Go官方的标准包里面提供了一个WebSocket的包 golang.org/x/net/websocket，但也说的很明确，这个包里面并没有实现WebSocket协议规定的一些特性，而推荐使用github.com/gorilla/websocket这个包。
代码示例，代码主要实现服务端向客户端写入数据，你可以通过channel Broadcast插入数据。
web页面的测试，可以在百度上选择一些在线的，如http://coolaf.com/tool/chattest
package controllers import (
"github.com/astaxie/beego"
"github.com/gorilla/websocket"
"net/http"
) const (
readBufferSize = 1024
writeBufferSize = 1024
) type Client struct {
conn *websocket.Conn
messages chan []byte
} var (
upgrader = websocket.Upgrader{
ReadBufferSize: readBufferSize,
WriteBufferSize: writeBufferSize,
CheckOrigin: func(r *http.Request) bool {
return true
},
}
clients map[*Client]bool // 存储所有的链接
addClients chan *Client // 新链接进来的chan
delClients chan *Client // 删除退出的chan
Broadcast chan []byte // 广播聊天的chan
) func (c *Client) writePump() {
defer func() {
delClients <- c
c.conn.Close()
}()
for {
select {
case message := <- c.messages:
err := c.conn.WriteMessage(websocket.TextMessage, message)
if err != nil {
return
}
}
}
} func manager() {
clients = make(map[*Client]bool)
addClients = make(chan *Client)
delClients = make(chan *Client) for {
select {
case message := <- Broadcast:
for client := range clients {
select {
case client.messages <- message:
default:
close(client.messages)
delete(clients, client)
}
}
case client := <- addClients:
clients[client] = true
case client := <- delClients:
if _, ok := clients[client]; ok {
close(client.messages)
delete(clients, client)
}
}
}
} type WebSocketController struct {
beego.Controller
} func (this *WebSocketController) Ws() {
ws, err := upgrader.Upgrade(this.Ctx.ResponseWriter, this.Ctx.Request, nil)
if err != nil {
return
}
client := &Client{conn: ws, messages: make(chan []byte, 256)}
addClients <- client
go client.writePump()
} func init() {
Broadcast = make(chan []byte)
go manager()
}
参考文章：
http://time-track.cn/websocket-and-golang.html
https://www.cnblogs.com/cornor/p/6178507.html
https://github.com/gorilla/websocket/tree/master/examples/chat
https://github.com/astaxie/build-web-application-with-golang/blob/master/zh/08.2.md