在现代前端越来越注重实时和交互性的时代，一种网络通信协议变得更加流行，它就是 websocket。在使用中，websocket 和 http 有一定的相似点，但是与传统的 http 请求不同，websocket 能够实现长时间保持连接。如果你正在考虑使用 websocket 来建立 web 应用程序，那么你可能需要使用一些编程语言来实现它。其中，golang 是一种非常流行的编程语言之一，让我们来了解一下如何在 golang 中实现 websocket。
一、什么是 WebSocket？
WebSocket 是一种在单个 TCP 连接上提供双向通信的网络协议。在传统的 HTTP 协议中，请求从浏览器发送到服务器，服务器进行处理后返回结果给浏览器，这个过程是一次性的，请求处理完成之后，连接就会关闭。而 WebSocket 协议则不同，当浏览器与服务器建立连接后，连接就会一直保持，直到用户或服务器决定关闭连接。这意味着，在连接保持期间，服务器可以随时向客户端发送信息，而不需要等待浏览器发出请求。
二、Golang 实现 WebSocket
Golang 是一种支持并发编程的编程语言，最初由 Google 开发，它的优势在于它的运行效率和内存占用率极低。下面我们将介绍如何使用 Golang 实现 WebSocket。
立即学习“go语言免费学习笔记（深入）”；
-
安装 Gorilla WebSocket 库
Gorilla WebSocket 是一个流行的 WebSocket 库，它提供了简单易用的 API，用于创建和处理 WebSocket 连接。在安装 Gorilla WebSocket 库之前，你需要先安装 Go 环境，Go 安装完成后，使用以下命令即可安装 Gorilla WebSocket 库：go get github.com/gorilla/websocket
- 编写代码
下面我们将使用 Go 和 Gorilla WebSocket 库来实现一个简单的聊天室。在我们的聊天室中，用户可以发送消息并查看来自其他用户的消息。以下是实现 WebSocket 聊天室的代码：
package main
import (
"fmt"
"log"
"net/http"
"github.com/gorilla/websocket"
)
var clients = make(map[*websocket.Conn]bool)
var broadcast = make(chan Message)
var upgrader = websocket.Upgrader{}
// Message struct
type Message struct {
Username string `json:"username"`
Body string `json:"body"`
}
func main() {
// Configure websocket route
http.HandleFunc("/ws", handleConnections)
// Start listening for incoming chat messages
go handleMessages()
// Start the server on localhost port 8080 and log any errors
err := http.ListenAndServe(":8080", nil)
if err != nil {
log.Fatal("ListenAndServe: ", err)
}
}
func handleConnections(w http.ResponseWriter, r *http.Request) {
// Upgrade initial GET request to a websocket
ws, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Fatal(err)
}
// Make sure we close the connection when the function returns
defer ws.Close()
// Register our new client
clients[ws] = true
for {
var msg Message
// Read in a new message as JSON and map it to a Message object
err := ws.ReadJSON(&msg)
if err != nil {
log.Printf("error: %v", err)
delete(clients, ws)
break
}
// Send the newly received message to the broadcast channel
broadcast <- msg
}
}
func handleMessages() {
for {
// Grab the next message from the broadcast channel
msg := <-broadcast
// Send it out to every client that is currently connected
for client := range clients {
err := client.WriteJSON(msg)
if err != nil {
log.Printf("error: %v", err)
client.Close()
delete(clients, client)
}
}
}
}代码的主要思想是创建一个 WebSocket 连接并添加到 clients 列表中，任何消息都会被写入到 broadcast 通道中，并在另一个 goroutine 中发送到所有客户端。每个连接通过读取和分配 Message 对象来接收消息。客户端发送消息的示例代码如下：
let socket = new WebSocket("ws://localhost:8080/ws");
socket.addEventListener("open", function() {
socket.send(JSON.stringify({
"username": "John",
"body": "Hello World!"
}));
});
socket.addEventListener("message", function(event) {
console.log("Received: " + event.data);
});在这个示例中，我们首先创建一个 WebSocket 对象并将其连接到服务器。在连接成功后，我们发送一个 JSON 作为消息体。服务端发送消息给客户端时，我们需要在客户端的 JavaScript 代码中监听 message 事件，并在收到消息时处理它。
三、总结
WebSocket 提供了一种实时通信的新方式，它为 Web 应用提供了更多互动性和用户体验。使用 Golang 和 Gorilla WebSocket 库可以轻松地实现 WebSocket 连接，并在应用程序中使用 WebSocket 协议。
本文提供了一个简单的聊天室的实现示例，希望对您有所帮助。当然，WebSocket 还可以用于许多其他类型的应用程序，因此请根据您自己的需求进行调整。
