用 Go 编写一个简单的 WebSocket 推送服务

本文中代码可以在 github.com/alfred-zhong/wserver 获取。

背景

最近拿到需求要在网页上展示报警信息。以往报警信息都是通过短信，微信和 App 推送给用户的，现在要让登录用户在网页端也能实时接收到报警推送。

依稀记得以前工作的时候遇到过类似的需求。因为以前的浏览器标准比较陈旧，并且那时用 Java 较多，所以那时候解决这个问题就用了 Comet4J。具体的原理就是长轮询，长链接。但现在毕竟 html5 流行开来了，IE 都被 Edge 接替了，再用以前这种技术就显得过时。

很早以前就听过 WebSocket 的大名，但因为那时很多用户的浏览器还不支持，所以对这个技术也就是浅尝辄止，没有太深入研究过。现在趁着项目需要，就来稍微深入了解一下。

websocket 简介

以往浏览器要获取服务端数据，都是通过发送 HTTP 请求，然后等待服务端回应的。也就是说浏览器端一直是整个请求的发起者，只有它主动，才能获取到数据。而要让浏览器一侧能够获取到服务端的实时数据，就需要不停地向服务端发起请求。虽然大多数情况下并没有获取到实际数据，但这大大增加了网络压力，对于服务端来说压力也直线上升。

后来我们学会了使用长连接 + 长轮询的方式。换句话说，也就是延长 HTTP 请求的存在时间，尽量保持 HTTP 连接。虽然这在一定程度上降低了不少压力，但仍然需要不停地进行轮询，也做不到真正的实时性。（借用一张图）

随着 HTML5 的到来，WebSocket 在 2011 年被定为标准（详情请参见 RFC 6455）。

借用 《Go Web 编程》的话。WebSocket 采用了一些特殊的报头，使得浏览器和服务器只需要做一个握手的动作，就可以在浏览器和服务器之间建立一条连接通道。且此连接会保持在活动状态，你可以使用 JavaScript 来向连接写入或从中接收数据，就像在使用一个常规的 TCP Socket 一样。它解决了 Web 实时化的问题。

由于 WebSocket 是全双工通信，所以当建立了 WebSocket 连接之后，接下来的通信就类似于传统的 TCP 通信了。客户端和服务端可以相互发送数据，不再有实时性的问题。

开发包的选择

在 Go 官方的 SDK 中，并不包含对 WebSocket 的支持，所以必须使用第三方库。

要使用 Golang 开发 WebSocket，选择基本就在 x/net/websocket 和 gorilla/websocket 之间。《Go Web 编程》一书中的例子使用了 x/net/websocket 作为开发包，而且貌似它也更加官方且正式。而实际根据我在网上查询得到的反馈看来，并非如此。x/net/websocket 貌似 Bug 较多，且较为不稳定，问题解决也并不及时。相比之下，gorilla/websocket 则更加优秀。

还有对于 Gorilla web toolkit 组织的贡献，必须予以感谢。