Golang使用WebSocket通信的實(shí)現(xiàn)

更新時(shí)間：2021年02月21日 10:29:08 作者：有理叔

這篇文章主要介紹了Golang使用WebSocket通信的實(shí)現(xiàn)，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧

使用Golang能夠創(chuàng)建WebSocket通信，只需要使用golang.org/x/net/websocket包即可，該包有可能被墻，無(wú)法使用go get下載，但是我們可以從golang中國(guó)提供的地址下載，然后將包放在對(duì)應(yīng)的路徑下。

WebSocket是一種通信協(xié)議，旨在改善HTTP作為無(wú)狀態(tài)協(xié)議通信的效率問(wèn)題，WebSocket是客戶端與服務(wù)器之間的全雙工連接，客戶端和服務(wù)器只需要建立一次連接就可以使用該連接進(jìn)行通信。在我們的項(xiàng)目中，一般客戶端是前端頁(yè)面，使用JavaScript創(chuàng)建WebSocket與后端的WebSocket服務(wù)端進(jìn)行通信。

下面首先看看我們的服務(wù)端代碼

package main
 
import (
 "fmt"
 "html/template"
 "net/http"
 "os"
 "strings"
 
 "golang.org/x/net/websocket"
)
 
func upper(ws *websocket.Conn) {
 var err error
 for {
 var reply string
 
 if err = websocket.Message.Receive(ws, &reply); err != nil {
  fmt.Println(err)
  continue
 }
 
 if err = websocket.Message.Send(ws, strings.ToUpper(reply)); err != nil {
  fmt.Println(err)
  continue
 }
 }
}
 
func index(w http.ResponseWriter, r *http.Request) {
 if r.Method != "GET" {
 return
 }
 
 t, _ := template.ParseFiles("index.html")
 t.Execute(w, nil)
}
 
func main() {
 http.Handle("/upper", websocket.Handler(upper))
 http.HandleFunc("/", index)
 
 if err := http.ListenAndServe(":9999", nil); err != nil {
 fmt.Println(err)
 os.Exit(1)
 }
}

我們從main函數(shù)看起，我們使用Golang的http包創(chuàng)建了一個(gè)Server，該Server能夠處理兩個(gè)請(qǐng)求，一個(gè)是/，這是Web項(xiàng)目的根路徑，index函數(shù)作為處理方法，返回首頁(yè)index.html。

Server還處理/upper請(qǐng)求，該請(qǐng)求的處理函數(shù)是一個(gè)WebSocket的處理函數(shù)，它包裹了我們自己定義的upper函數(shù)，upper函數(shù)有一個(gè)參數(shù)，就是服務(wù)端創(chuàng)建好的WebSocket連接，upper函數(shù)邏輯很簡(jiǎn)單，通過(guò)WebSocket連接讀取服務(wù)器的請(qǐng)求內(nèi)容，將內(nèi)容轉(zhuǎn)化為大寫后，通過(guò)WebSocket將結(jié)果返回給客戶端。

<!DOCTYPE html>
<html>
<head>
  <meta charset="utf-8"/>
  <title>Websocket</title>
</head>
<body>
  <h1>字符串轉(zhuǎn)換大寫</h1>
  <form>
    <p>
      字符串: <input id="content" type="text" placeholder="輸入要轉(zhuǎn)換的字符串">
    </p>
  </form>
  <label id="result">結(jié)果為：</label><br><br>
  <button onclick="send()">轉(zhuǎn)換</button>
 
  <script type="text/javascript">
    var sock = null;
    var wsuri = "ws://127.0.0.1:9999/upper";
    sock = new WebSocket(wsuri);
    sock.onmessage = function(e) {
      var result = document.getElementById('result');
      result.innerHTML = "結(jié)果為：" + e.data;
    }
 
    function send() {
      var msg = document.getElementById('content').value;
      sock.send(msg);
    }
  </script>
</body>
</html>

前端頁(yè)面我們主要看JavaScript代碼部分，首先建立一個(gè)全局的WebSocket，之后點(diǎn)擊轉(zhuǎn)換按鈕時(shí)，就會(huì)將input的內(nèi)容通過(guò)WebSocket發(fā)送出去。WebSocket的onmessage函數(shù)用來(lái)處理服務(wù)端的響應(yīng)，還有onerror，onclose等方法，分別是WebSocket的錯(cuò)誤回調(diào)和關(guān)閉回調(diào)，本文并沒(méi)有使用，一般用來(lái)處理資源回收和報(bào)錯(cuò)。

Golang可以很方便的使用WebSocket，只是使用websocket.Handler包裹一個(gè)處理函數(shù)即可。

到此這篇關(guān)于Golang使用WebSocket通信的實(shí)現(xiàn)的文章就介紹到這了,更多相關(guān)Golang WebSocket通信內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: