Go WebSocket

2022-05-13 17:42 更新

WebSocket是HTML5的重要特性，它實現(xiàn)了基于瀏覽器的遠程socket，它使瀏覽器和服務(wù)器可以進行全雙工通信，許多瀏覽器（Firefox、Google Chrome和Safari）都已對此做了支持。

在WebSocket出現(xiàn)之前，為了實現(xiàn)即時通信，采用的技術(shù)都是“輪詢”，即在特定的時間間隔內(nèi)，由瀏覽器對服務(wù)器發(fā)出HTTP Request，服務(wù)器在收到請求后，返回最新的數(shù)據(jù)給瀏覽器刷新，“輪詢”使得瀏覽器需要對服務(wù)器不斷發(fā)出請求，這樣會占用大量帶寬。

WebSocket采用了一些特殊的報頭，使得瀏覽器和服務(wù)器只需要做一個握手的動作，就可以在瀏覽器和服務(wù)器之間建立一條連接通道。且此連接會保持在活動狀態(tài)，你可以使用JavaScript來向連接寫入或從中接收數(shù)據(jù)，就像在使用一個常規(guī)的TCP Socket一樣。它解決了Web實時化的問題，相比傳統(tǒng)HTTP有如下好處：

一個Web客戶端只建立一個TCP連接
Websocket服務(wù)端可以推送(push)數(shù)據(jù)到web客戶端.
有更加輕量級的頭，減少數(shù)據(jù)傳送量

WebSocket URL的起始輸入是ws://或是wss://（在SSL上）。下圖展示了WebSocket的通信過程，一個帶有特定報頭的HTTP握手被發(fā)送到了服務(wù)器端，接著在服務(wù)器端或是客戶端就可以通過JavaScript來使用某種套接口（socket），這一套接口可被用來通過事件句柄異步地接收數(shù)據(jù)。

WebSocket原理

WebSocket的協(xié)議頗為簡單，在第一次handshake通過以后，連接便建立成功，其后的通訊數(shù)據(jù)都是以”\x00″開頭，以”\xFF”結(jié)尾。在客戶端，這個是透明的，WebSocket組件會自動將原始數(shù)據(jù)“掐頭去尾”。

瀏覽器發(fā)出WebSocket連接請求，然后服務(wù)器發(fā)出回應(yīng)，然后連接建立成功，這個過程通常稱為“握手” (handshaking)。請看下面的請求和反饋信息：

在請求中的"Sec-WebSocket-Key"是隨機的，對于整天跟編碼打交到的程序員，一眼就可以看出來：這個是一個經(jīng)過base64編碼后的數(shù)據(jù)。服務(wù)器端接收到這個請求之后需要把這個字符串連接上一個固定的字符串：

258EAFA5-E914-47DA-95CA-C5AB0DC85B11

即：f7cb4ezEAl6C3wRaU6JORA==連接上那一串固定字符串，生成一個這樣的字符串：

f7cb4ezEAl6C3wRaU6JORA==258EAFA5-E914-47DA-95CA-C5AB0DC85B11

對該字符串先用 sha1安全散列算法計算出二進制的值，然后用base64對其進行編碼，即可以得到握手后的字符串：

rE91AJhfC+6JdVcVXOGJEADEJdQ=

將之作為響應(yīng)頭Sec-WebSocket-Accept的值反饋給客戶端。

Go實現(xiàn)WebSocket

Go語言標準包里面沒有提供對WebSocket的支持，但是在由官方維護的go.net子包中有對這個的支持，你可以通過如下的命令獲取該包：

go get code.google.com/p/go.net/websocket

WebSocket分為客戶端和服務(wù)端，接下來我們將實現(xiàn)一個簡單的例子:用戶輸入信息，客戶端通過WebSocket將信息發(fā)送給服務(wù)器端，服務(wù)器端收到信息之后主動Push信息到客戶端，然后客戶端將輸出其收到的信息，客戶端的代碼如下：

<html>
<head></head>
<body>
    <script type="text/javascript">
        var sock = null;
        var wsuri = "ws://127.0.0.1:1234";

        window.onload = function() {

            console.log("onload");

            sock = new WebSocket(wsuri);

            sock.onopen = function() {
                console.log("connected to " + wsuri);
            }

            sock.onclose = function(e) {
                console.log("connection closed (" + e.code + ")");
            }

            sock.onmessage = function(e) {
                console.log("message received: " + e.data);
            }
        };

        function send() {
            var msg = document.getElementById('message').value;
            sock.send(msg);
        };
    </script>
    <h1>WebSocket Echo Test</h1>
    <form>
        <p>
            Message: <input id="message" type="text" value="Hello, world!">
        </p>
    </form>
    <button onclick="send();">Send Message</button>
</body>
</html>

可以看到客戶端JS，很容易的就通過WebSocket函數(shù)建立了一個與服務(wù)器的連接sock，當握手成功后，會觸發(fā)WebScoket對象的onopen事件，告訴客戶端連接已經(jīng)成功建立?？蛻舳艘还步壎怂膫€事件。

1）onopen 建立連接后觸發(fā)
2）onmessage 收到消息后觸發(fā)
3）onerror 發(fā)生錯誤時觸發(fā)
4）onclose 關(guān)閉連接時觸發(fā)

我們服務(wù)器端的實現(xiàn)如下：

package main

import (
    "code.google.com/p/go.net/websocket"
    "fmt"
    "log"
    "net/http"
)

func Echo(ws *websocket.Conn) {
    var err error

    for {
        var reply string

        if err = websocket.Message.Receive(ws, &reply); err != nil {
            fmt.Println("Can't receive")
            break
        }

        fmt.Println("Received back from client: " + reply)

        msg := "Received:  " + reply
        fmt.Println("Sending to client: " + msg)

        if err = websocket.Message.Send(ws, msg); err != nil {
            fmt.Println("Can't send")
            break
        }
    }
}

func main() {
    http.Handle("/", websocket.Handler(Echo))

    if err := http.ListenAndServe(":1234", nil); err != nil {
        log.Fatal("ListenAndServe:", err)
    }
}

當客戶端將用戶輸入的信息Send之后，服務(wù)器端通過Receive接收到了相應(yīng)信息，然后通過Send發(fā)送了應(yīng)答信息。

通過上面的例子我們看到客戶端和服務(wù)器端實現(xiàn)WebSocket非常的方便，Go的源碼net分支中已經(jīng)實現(xiàn)了這個的協(xié)議，我們可以直接拿來用，目前隨著HTML5的發(fā)展，我想未來WebSocket會是Web開發(fā)的一個重點，我們需要儲備這方面的知識。

以上內(nèi)容是否對您有幫助：

← Go Socket編程

Go REST →

寫筆記

我要補充