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

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

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

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

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

下面首先看看我們的服務端代碼

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)建了一個Server，該Server能夠處理兩個請求，一個是/，這是Web項目的根路徑，index函數(shù)作為處理方法，返回首頁index.html。

Server還處理/upper請求，該請求的處理函數(shù)是一個WebSocket的處理函數(shù)，它包裹了我們自己定義的upper函數(shù)，upper函數(shù)有一個參數(shù)，就是服務端創(chuàng)建好的WebSocket連接，upper函數(shù)邏輯很簡單，通過WebSocket連接讀取服務器的請求內(nèi)容，將內(nèi)容轉(zhuǎn)化為大寫后，通過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>

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

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

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

您可能感興趣的文章: