Golang使用WebSocket通信的實現

Posted on by

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

WebSocket是一種通信協議，旨在改善HTTP作為無狀態協議通信的效率問題，WebSocket是客戶端與服務器之間的全雙工連接，客戶端和服務器隻需要建立一次連接就可以使用該連接進行通信。在我們的項目中，一般客戶端是前端頁面，使用JavaScript創建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函數看起，我們使用Golang的http包創建瞭一個Server，該Server能夠處理兩個請求，一個是/，這是Web項目的根路徑，index函數作為處理方法，返回首頁index.html。

Server還處理/upper請求，該請求的處理函數是一個WebSocket的處理函數，它包裹瞭我們自己定義的upper函數，upper函數有一個參數，就是服務端創建好的WebSocket連接，upper函數邏輯很簡單，通過WebSocket連接讀取服務器的請求內容，將內容轉化為大寫後，通過WebSocket將結果返回給客戶端。

<!DOCTYPE html>
<html>
<head>
  <meta charset="utf-8"/>
  <title>Websocket</title>
</head>
<body>
  <h1>字符串轉換大寫</h1>
  <form>
    <p>
      字符串: <input id="content" type="text" placeholder="輸入要轉換的字符串">
    </p>
  </form>
  <label id="result">結果為：</label><br><br>
  <button onclick="send()">轉換</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 = "結果為：" + e.data;
    }
 
    function send() {
      var msg = document.getElementById('content').value;
      sock.send(msg);
    }
  </script>
</body>
</html>

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

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

到此這篇關於Golang使用WebSocket通信的實現的文章就介紹到這瞭,更多相關Golang WebSocket通信內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：