使用Golang實現(xiàn)Websocket編程

Websocket是一種在瀏覽器和服務(wù)器之間進行全雙工通信的技術(shù)。在傳統(tǒng)的HTTP通信中，瀏覽器只能向服務(wù)器發(fā)起請求，服務(wù)器在處理請求后會返回響應(yīng)。但在Websocket中，瀏覽器和服務(wù)器之間建立了一個長連接，使得數(shù)據(jù)可以在雙方之間實時傳輸。

本文將介紹如何使用Golang實現(xiàn)Websocket編程。我們將使用標(biāo)準(zhǔn)庫中的net/http和golang.org/x/net/websocket包來完成這個任務(wù)。

準(zhǔn)備工作

在開始之前，要確保你已經(jīng)安裝了Golang。如果還沒有安裝，可以參考這個鏈接來安裝：https://golang.org/doc/install

接著我們來創(chuàng)建一個新的Go模塊，首先創(chuàng)建一個文件夾：

mkdir websocket-democd websocket-demo

然后執(zhí)行下面的命令來初始化一個新的Go模塊：

go mod init websocket-demo

這個命令將會創(chuàng)建一個go.mod文件，用來記錄模塊依賴關(guān)系。

實現(xiàn)Websocket服務(wù)器

現(xiàn)在我們開始實現(xiàn)Websocket服務(wù)器。創(chuàng)建一個main.go文件，并在其中添加如下代碼：

go

package main

import (

"fmt"

"golang.org/x/net/websocket"

"net/http"

)

func main() {

http.Handle("/ws", websocket.Handler(wsHandler))

fmt.Println("Server started at http://localhost:8080")

http.ListenAndServe(":8080", nil)

}

func wsHandler(ws *websocket.Conn) {

fmt.Println("Websocket connection established")

var msg = ""

for {

err := websocket.Message.Receive(ws, &msg)

if err != nil {

fmt.Println("Connection closed")

break

}

fmt.Printf("Received message: %s\n", msg)

err = websocket.Message.Send(ws, msg)

if err != nil {

fmt.Println("Connection closed")

break

}

}

}

這段代碼中，我們首先使用http.Handle()函數(shù)來為Websocket連接監(jiān)聽路徑/ws，并將監(jiān)聽函數(shù)wsHandler作為參數(shù)傳入。wsHandler用來處理客戶端和服務(wù)器之間的通信，它接收一個websocket.Conn類型的參數(shù)，表示一個Websocket連接。在wsHandler中，我們首先打印一條"Websocket connection established"的日志，表示連接已經(jīng)建立。然后使用websocket.Message.Receive()函數(shù)來接收客戶端發(fā)送的消息，并使用websocket.Message.Send()函數(shù)將接收到的消息回傳給客戶端。如果有任何錯誤發(fā)生，我們將打印一條"Connection closed"的日志，并退出循環(huán)?，F(xiàn)在我們來運行這個程序，執(zhí)行下面的命令：

go run main.go

這個命令將會運行我們的Websocket服務(wù)器。在瀏覽器中輸入http://localhost:8080`，并打開控制臺，在控制臺中輸入以下代碼即可連接到Websocket服務(wù)器：`javascriptvar ws = new WebSocket("ws://localhost:8080/ws");ws.onopen = function() {  console.log("Websocket connection established");  ws.send("Hello, server!");};ws.onmessage = function(evt) {  console.log("Received message: " + evt.data);};ws.onclose = function() {  console.log("Websocket connection closed");};

這段代碼中，我們首先使用WebSocket對象來建立一個與服務(wù)器的連接。當(dāng)連接建立成功后，我們打印一條"Websocket connection established"的日志，并使用ws.send()函數(shù)來向服務(wù)器發(fā)送一條消息。當(dāng)我們接收到服務(wù)器返回的消息時，我們會打印一條"Received message: "的日志，并將消息內(nèi)容輸出。最后，當(dāng)連接關(guān)閉時，我們將會打印一條"Websocket connection closed"的日志。

現(xiàn)在我們來嘗試在控制臺中輸入Hello, server!，看看程序是否能夠?qū)⑾⒒貍鹘o我們。如果一切正常，我們將在控制臺中看到以下輸出：

Received message: Hello, server!

總結(jié)

在本文中，我們介紹了如何使用Golang實現(xiàn)Websocket編程。我們使用標(biāo)準(zhǔn)庫中的net/http和golang.org/x/net/websocket包來實現(xiàn)了一個簡單的Websocket服務(wù)器，并用瀏覽器發(fā)送了一個測試消息。希望這個示例可以幫助你更好地理解Websocket的工作原理，以及如何使用Golang來實現(xiàn)Websocket編程。

以上就是IT培訓(xùn)機構(gòu)千鋒教育提供的相關(guān)內(nèi)容，如果您有web前端培訓(xùn)，鴻蒙開發(fā)培訓(xùn)，python培訓(xùn)，linux培訓(xùn)，java培訓(xùn)，UI設(shè)計培訓(xùn)等需求，歡迎隨時聯(lián)系千鋒教育。