golang+websocket

websocket

在WebSocket API中,瀏覽器和服務(wù)器只需要做一個(gè)握手的動(dòng)作,然后,瀏覽器和服務(wù)器之間就形成了一條快速通道。兩者之間就直接可以數(shù)據(jù)互相傳送。

我大概調(diào)研了一下目前在go生態(tài)下使用websocket的方案,感覺websocket這個(gè)庫是比較好用的。文檔友好,API穩(wěn)定。比較推薦使用。我們的代碼也使用了這個(gè)庫。

下面我們來寫一個(gè)簡單的websocket應(yīng)用 hello_robot,它的作用很簡單,就是你不管和它說什么,它都會(huì)說"hello" + 你輸入的內(nèi)容。

server

server有兩個(gè)作用,第一是監(jiān)聽請(qǐng)求并發(fā)送我們的html文件,第二是監(jiān)聽用戶輸入內(nèi)容并加以答復(fù)。

package main

import (
  "log"
  "net/http"
  "github.com/gorilla/websocket"
)
var upgrader = websocket.Upgrader{}

我們需要一個(gè)Conn類型來表示我們的websocket連接。在客戶端,我們需要用一個(gè)upgrader對(duì)象調(diào)用Upgrade函數(shù)來獲得這個(gè)連接的指針。所以,我們首先需要?jiǎng)?chuàng)建這個(gè)Upgrader。

func main() {
    http.HandleFunc("/echo", echo)
    http.HandleFunc("/", home)
    log.Fatal(http.ListenAndServe("localhost:8080", nil))
}

在主函數(shù)中,我們?cè)O(shè)定了兩個(gè)路由,一個(gè)是用于send我們的客戶端html文件的,一個(gè)是用來監(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ù)負(fù)責(zé)處理我們的websocket連接。首先,我們調(diào)用upgrader.Upgrade(w, r, nil)獲得這個(gè)連接的指針。然后,我們進(jìn)入一個(gè)循環(huán),不停的讀取連接中傳來的信息,當(dāng)有信息傳來時(shí),對(duì)信息處理后返回給客戶端。這里有一點(diǎn)要注意,就是c.ReadMessage()是阻塞的,,也就是說,當(dāng)有消息來時(shí),它后面的代碼才會(huì)被執(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;
};

客戶端(前端代碼)很簡單,這里我只寫一個(gè)框架。首先,用new WebSocket(...)建立一個(gè)連接,并為這個(gè)連接綁定幾個(gè)函數(shù),這里我們可以設(shè)定onopen(建立連接),onmessage(收到服務(wù)端消息),onerror(發(fā)生錯(cuò)誤)。然后,我們可以給我們的send按鈕綁定單擊事件,單擊后發(fā)送數(shù)據(jù)到后臺(tái)。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

  • WebSocket簡介 談到Web實(shí)時(shí)推送,就不得不說WebSocket。在WebSocket出現(xiàn)之前,很多網(wǎng)站為...
    吧啦啦小湯圓閱讀 8,313評(píng)論 15 75
  • Spring Cloud為開發(fā)人員提供了快速構(gòu)建分布式系統(tǒng)中一些常見模式的工具(例如配置管理,服務(wù)發(fā)現(xiàn),斷路器,智...
    卡卡羅2017閱讀 136,506評(píng)論 19 139
  • 死磕自己,服務(wù)大眾! 今天熟男很認(rèn)真很認(rèn)真的閱讀了一本關(guān)未來十年葡萄酒將出現(xiàn)的變化。內(nèi)容很多,有我認(rèn)同接受,也有我...
    張錦全閱讀 399評(píng)論 1 1
  • 為什么我會(huì)這樣? 為什么? 為什么我不能寫字? 為什么我不能奔跑? 為什么我不能有玩耍的自由 為什么我會(huì)被命運(yùn)擺布...
    寫作星閱讀 332評(píng)論 2 2

友情鏈接更多精彩內(nèi)容