websocket
在WebSocket API中尖坤,瀏覽器和服務器只需要做一個握手的動作盔粹,然后,瀏覽器和服務器之間就形成了一條快速通道煤篙。兩者之間就直接可以數(shù)據(jù)互相傳送。
我大概調(diào)研了一下目前在go生態(tài)下使用websocket的方案毁腿,感覺websocket這個庫是比較好用的辑奈。文檔友好,API穩(wěn)定已烤。比較推薦使用鸠窗。我們的代碼也使用了這個庫。
下面我們來寫一個簡單的websocket應用 hello_robot,它的作用很簡單胯究,就是你不管和它說什么稍计,它都會說"hello" + 你輸入的內(nèi)容。
server
server有兩個作用唐片,第一是監(jiān)聽請求并發(fā)送我們的html文件丙猬,第二是監(jiān)聽用戶輸入內(nèi)容并加以答復涨颜。
package main
import (
"log"
"net/http"
"github.com/gorilla/websocket"
)
var upgrader = websocket.Upgrader{}
我們需要一個Conn類型來表示我們的websocket連接。在客戶端茧球,我們需要用一個upgrader對象調(diào)用Upgrade函數(shù)來獲得這個連接的指針庭瑰。所以,我們首先需要創(chuàng)建這個Upgrader抢埋。
func main() {
http.HandleFunc("/echo", echo)
http.HandleFunc("/", home)
log.Fatal(http.ListenAndServe("localhost:8080", nil))
}
在主函數(shù)中弹灭,我們設定了兩個路由,一個是用于send我們的客戶端html文件的揪垄,一個是用來監(jiān)聽用戶輸入內(nèi)容的穷吮。最后,我們監(jiān)聽了本地的8080端口饥努。
func home(w http.ResponseWriter, r *http.Request) {
http.ServeFile(w, r, "home.html")
}
home函數(shù)用于發(fā)送客戶端html文件捡鱼。
func echo( w http.ResponseWriter, r *http.Request){
c, _ := upgrader.Upgrade(w, r, nil)
defer c.Close()
for {
mt, message, _ := c.ReadMessage()
c.WriteMessage(mt, append([]byte("hello "),message[:]...))
}
}
echo函數(shù)負責處理我們的websocket連接。首先酷愧,我們調(diào)用upgrader.Upgrade(w, r, nil)獲得這個連接的指針驾诈。然后,我們進入一個循環(huán)溶浴,不停的讀取連接中傳來的信息乍迄,當有信息傳來時,對信息處理后返回給客戶端士败。這里有一點要注意闯两,就是c.ReadMessage()是阻塞的,谅将,也就是說漾狼,當有消息來時,它后面的代碼才會被執(zhí)行戏自。
客戶端
ws = new WebSocket("ws://127.0.0.1:8080/echo");
ws.onopen = function(evt) {...}
ws.onmessage = function(evt) {...}
ws.onerror = function(evt) {...}
document.getElementById("send").onclick = function(evt) {
if (!ws) {return false;}
ws.send(input.value);
return;
};
客戶端(前端代碼)很簡單邦投,這里我只寫一個框架。首先擅笔,用new WebSocket(...)建立一個連接志衣,并為這個連接綁定幾個函數(shù),這里我們可以設定onopen(建立連接)猛们,onmessage(收到服務端消息),onerror(發(fā)生錯誤)念脯。然后,我們可以給我們的send按鈕綁定單擊事件弯淘,單擊后發(fā)送數(shù)據(jù)到后臺绿店。