Golang如何使用websocket

nts host 顯示 websocket click gop In receive sam

下載websocket包

$ go get golang.org/x/net/websocket

如果下載失敗,可能是被墻了。

package golang.org/x/net/websocket: unrecognized import path "golang.org/x/net/websocket" (https fetch: Get https://golang.org/x/net/websocket?go-get=1: dial tcp 216.239.37.1:443: connectex: A connection attempt failed because the connected party did not properly respond after a period of time, or established connection failed because connected host has failed to respond.)

可以嘗試

$ go get -u github.com/golang/net/websocket

下載後記得將路徑改成golang.org/x/net/websocket

$ cd $GOPATH
$ mkdir -p golang.org/x/net
$ cp github.com/golang/net/websocket/  golang.org/x/net/ -R

實例

服務端代碼

server.go

package main

import (  

    "fmt"   

    "golang.org/x/net/websocket" 

    "html/template"              //支持模板html  

    "log"  

    "net/http"  

)

func Echo(ws *websocket.Conn) {  

    var err error  

    for {  

        var reply string  

        //websocket接受信息  

        if err = websocket.Message.Receive(ws, &reply); err != nil {  

            fmt.Println("receive failed:", err)  

            break  

        }

        fmt.Println("reveived from client: " + reply)

        msg := "received:" + reply

        fmt.Println("send to client:" + msg)

        //這裏是發送消息

        if err = websocket.Message.Send(ws, msg); err != nil {

            fmt.Println("send failed:", err)

            break

        }

    }

}

func web(w http.ResponseWriter, r *http.Request) {  

    //打印請求的方法  

    fmt.Println("method", r.Method)  

    if r.Method == "GET" { //如果請求方法為get顯示login.html,並相應給前端  

        t, _ := template.ParseFiles("websocket.html")  

        t.Execute(w, nil)  

    } else {  

        //否則走打印輸出post接受的參數username和password  

        fmt.Println(r.PostFormValue("username"))  

        fmt.Println(r.PostFormValue("password"))  

    }  

}

func main() {

    //接受websocket的路由地址

    http.Handle("/websocket", websocket.Handler(Echo))

    //html頁面

    http.HandleFunc("/web", web)

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

        log.Fatal("ListenAndServe:", err)

    }

}

客戶端代碼

websocket.html

<!DOCTYPE html>

 <html>

<head>

    <meta charset="utf-8"/>

    <title>go測試socket</title>

</head>

<body>

    <script type="text/javascript">

        var sock = null;

        var wsuri = "ws://127.0.0.1:1234/websocket";

        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>

測試

啟動服務端

$ ./server

啟動瀏覽器
輸入http://127.0.0.1:1234/web

技術分享圖片

網絡通信內容如圖所示，首先加載頁面，接著使用HTTP建立websocket連接，後續通信直接使用websocket進行。

點擊頁面中SendMessage按鈕

server端輸出：

method GET
reveived from client: Hello, world!
send to client:received:Hello, world!

瀏覽器輸出：

技術分享圖片

參考

https://github.com/astaxie/build-web-application-with-golang/blob/master/zh/08.2.md

https://github.com/ukai/go-websocket-sample/blob/master/websocket_echo_sample.go

Golang如何使用websocket

Golang如何使用websocket

下載websocket包

實例

服務端代碼

客戶端代碼

測試

參考

相關推薦