1. 程式人生 > >Golang使用WebSocket通訊

Golang使用WebSocket通訊

使用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包裹一個處理函式即可。