golang是一种快速编译、高效、安全的编程语言，在web开发方面也有着广泛的应用。其中，使用golang搭建websocket是一个非常流行的应用场景。本文将介绍如何使用golang搭建一个websocket应用程序。
一、什么是Websocket
Websocket是一种协议，它允许在浏览器和服务器之间建立双向通信的连接。传统的HTTP协议只能由浏览器向服务器发出请求，并接收服务器响应。但是，使用Websocket可以让服务器主动向浏览器发送消息，实现了服务器与浏览器的全双工通信。
二、Golang如何实现Websocket
Golang自带了一个标准库net/http，这个包提供了一个http.Server结构体，它可以用来构建Web服务器。除此之外，net/http也提供了一个websocket子包，它为我们提供了构建WebSocket服务器所需的全部功能。
立即学习“go语言免费学习笔记（深入）”；
- 建立WebSocket连接
要建立WebSocket连接，客户端需要向服务器发送一个HTTP请求，并携带Upgrade头部，指示请求升级为WebSocket协议。服务器端在收到此请求后，需要检查Upgrade头部，并根据是否为WebSocket请求来进行相应的处理。
服务器端可以使用websocket.IsWebSocketUpgrade函数来检查请求是否为WebSocket请求，并使用websocket.Upgrade函数升级请求。升级成功后，服务器就可以通过*websocket.Conn类型的连接对象与客户端进行通信了。
func handler(w http.ResponseWriter, r *http.Request) {
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Println(err)
return
}
defer conn.Close()
// 使用conn进行通信
}其中，upgrader是一个websocket.Upgrader类型的变量，通过设置它的属性可以控制WebSocket连接的处理方式。Upgrade函数在升级请求时会阻塞，因此要将其放在goroutine中运行。
- 发送和接收WebSocket消息
一旦WebSocket连接建立成功，客户端和服务器都可以通过*websocket.Conn对象进行通信。使用这个对象可以发送和接收WebSocket消息。
发送消息很简单，只需要调用WriteMessage方法即可：
err := conn.WriteMessage(websocket.TextMessage, []byte("hello"))接收消息的方法有三种：
-
ReadMessage：阻塞读取，直到有消息到来 -
NextReader：阻塞等待下一个消息到来 -
NextWriter：返回一个io.WriteCloser对象，可以使用这个对象发送消息
一般使用ReadMessage方法接收WebSocket消息：
_, message, err := conn.ReadMessage()
if err != nil {
log.Println(err)
return
}
log.Printf("recv: %s", message)三、一个简单的WebSocket应用
在这个例子中，我们将创建一个简单的聊天室应用，客户端发送消息之后，服务器将消息广播给所有连接的客户端。
客户端代码：（使用JavaScript）
<!DOCTYPE html>
<html>
<head>
<title>Chatroom</title>
</head>
<body>
<input type="text" id="input-message"/>
<button onclick="sendMessage()">发送</button>
<ul id="message-list"></ul>
<script>
var ws = new WebSocket("ws://localhost:8080/ws");
ws.onmessage = function(event) {
var message = event.data;
var li = document.createElement("li");
li.appendChild(document.createTextNode(message));
document.getElementById("message-list").appendChild(li);
};
function sendMessage() {
var message = document.getElementById("input-message").value;
ws.send(message);
}
</script>
</body>
</html>服务器端代码：
package main
import (
"fmt"
"log"
"net/http"
"github.com/gorilla/websocket"
)
var upgrader = websocket.Upgrader{}
func main() {
http.HandleFunc("/ws", wsHandler)
err := http.ListenAndServe(":8080", nil)
if err != nil {
log.Fatal("ListenAndServe: ", err)
}
}
func wsHandler(w http.ResponseWriter, r *http.Request) {
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Println(err)
return
}
defer conn.Close()
for {
_, message, err := conn.ReadMessage()
if err != nil {
log.Println(err)
break
}
fmt.Printf("recv: %s
", message)
// 广播给所有客户端
for _, conn := range conns {
err := conn.WriteMessage(websocket.TextMessage, message)
if err != nil {
log.Println(err)
break
}
}
}
}在上面的代码中，我们将所有连接的websocket.Conn对象存储在conns变量中。每当有一个新消息到来时，服务器会将消息广播给所有客户端。
总结
使用Golang搭建WebSocket应用程序非常简单，只需要导入net/http和websocket包即可，直接使用标准库提供的API即可完成WebSocket应用的开发。希望此文章能够对Golang Websocket开发有所帮助。
