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