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

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

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

準(zhǔn)備工作

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

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

mkdir websocket-democd websocket-demo

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

go mod init websocket-demo

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

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

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

go run main.go

這個(gè)命令將會(huì)運(yùn)行我們的Websocket服務(wù)器。在瀏覽器中輸入http://localhost:8080`，并打開(kāi)控制臺(tái)，在控制臺(tái)中輸入以下代碼即可連接到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對(duì)象來(lái)建立一個(gè)與服務(wù)器的連接。當(dāng)連接建立成功后，我們打印一條"Websocket connection established"的日志，并使用ws.send()函數(shù)來(lái)向服務(wù)器發(fā)送一條消息。當(dāng)我們接收到服務(wù)器返回的消息時(shí)，我們會(huì)打印一條"Received message: "的日志，并將消息內(nèi)容輸出。最后，當(dāng)連接關(guān)閉時(shí)，我們將會(huì)打印一條"Websocket connection closed"的日志。

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

Received message: Hello, server!

總結(jié)

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

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