前言

上一篇文章 Golang 实现WebSockets中我们先介绍了什么是 WebSockets 协议。本篇文章将来介绍如何利用 Go 来实现一个 WebSockets。要基于 net/http 库编写一个简单的 WebSocket 响应服务器，我们需要：

  • 建立握手
  • 从客户端接收数据帧
  • 向客户端发送数据帧
  • 关闭握手

WebSockets 无需客户端事先请求即可从服务器传输数据，允许来回传递消息并保持连接打开，直到客户端或服务器终止它。因此，可以在客户端和服务器之间进行双向实时数据传输。 WebSocket 通信通常通过 TCP 端口号 443 完成。

WebSocket 协议规范定义了两种 URI 方案：

  • WebSocket (ws)：用于非加密连接
  • WebSocket Secure (wss)：用于加密连接

建立 HTTP 服务器

​server.go​
package main

import (
"fmt"
"net/http"
)

func main() {
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
fmt.Fprintf(w, "Welcome to the server!\n")
})

http.ListenAndServe(":8080", nil)
}
​go run server.go​​localhost:8080​
Welcome to the server!

发起握手

由于开销较小，WebSocket 可以在 Web 服务器和 Web 浏览器或客户端应用程序之间实现实时通信和快速数据传输。 WebSocket 通信发起一个握手，它使用 HTTP Upgrade() 标头从 HTTP 协议更改为 WebSocket 协议。

要建立 WebSocket 连接，需要在客户端和服务器之间进行一次握手。握手使用 Upgrade() 方法将 HTTP 服务器连接升级到 WebSocket 协议。一旦服务器停止，我们还将使用 defer 关闭连接。

​server.go​
​gorilla/websocket​
$ go get github.com/gorilla/websocket

定义一个 upgrader 结构，用来保存 WebSocket 连接的读取和写入缓冲区大小等信息：

var upgrader = websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
}
​server.go​
package main

import (
"fmt"
"net/http"

"github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
}

func main() {
http.HandleFunc("/echo", func(w http.ResponseWriter, r *http.Request) {
conn, _ := upgrader.Upgrade(w, r, nil) // error ignored for sake of simplicity

for {
// Read message from browser
msgType, msg, err := conn.ReadMessage()
if err != nil {
return
}

// Print the message to the console
fmt.Printf("%s sent: %s\n", conn.RemoteAddr(), string(msg))

// Write message back to browser
if err = conn.WriteMessage(msgType, msg); err != nil {
return
}
}
})

http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
http.ServeFile(w, r, "websockets.html")
})

http.ListenAndServe(":8080", nil)
}
​websockets.html​
<!-- websockets.html -->
<input id="input" type="text" />
<button onclick="send()">Send</button>
<pre id="output"></pre>
<script>
var input = document.getElementById("input");
var output = document.getElementById("output");
var socket = new WebSocket("ws://localhost:8080/echo");

socket.onopen = function () {
output.innerHTML += "Status: Connected\n";
};

socket.onmessage = function (e) {
output.innerHTML += "Server: " + e.data + "\n";
};

function send() {
socket.send(input.value);
input.value = "";
}
</script>
​go run server.go​
​http://localhost:8080/​
​Send​

界面如下：

回到服务器终端也能看到如下信息：

$ go run server.go
[::1]:58036 sent: Hello

至此，一个简单的 WebSockets 就完全实现了。接下来，我们看一下 WebSockets 有哪些实际运用。

WebSockets 的用例

WebSockets

目前，WebSockets 为 Android、iOS、Web 和桌面应用程序提供跨平台支持，WebSockets 常用于以下类型的应用程序：

  • 实时消息
  • 多人游戏
  • 实时比分提要
  • 协作编辑工具
  • 实时位置和方向应用
  • 使用 WebRTC 进行音频和视频聊天

总结

在本文中，我们探索了 WebSockets，并简要介绍了它们的工作原理，并仔细研究了全双工通信。为了了解 WebSocket 在 Go 中的工作原理，我们构建了一个简单的待办事项应用程序，该应用程序具有添加和删除任务的功能。最后，我们查看了使 WebSockets 有用和通用的几个附加特性，并回顾了 WebSockets 的一些实际应用。在 Go 中使用 WebSockets 相当简单明了，但这种组合会对应用程序的性能产生显着影响。