Go實戰--golang中使用WebSocket實時聊天室(gorilla/websocket、nkovacs/go-socket.io)


websocket

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

我大概調研了一下目前在go生態下使用websocket的方案,感覺websocket這個庫是比較好用的。文檔友好,API穩定。比較推薦使用。我們的代碼也使用了這個庫。

下面我們來寫一個簡單的websocket應用 hello_robot,它的作用很簡單,就是你不管和它說什么,它都會說"hello" + 你輸入的內容。

server

server有兩個作用,第一是監聽請求並發送我們的html文件,第二是監聽用戶輸入內容並加以答復。

package main

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

我們需要一個Conn類型來表示我們的websocket連接。在客戶端,我們需要用一個upgrader對象調用Upgrade函數來獲得這個連接的指針。所以,我們首先需要創建這個Upgrader。

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

在主函數中,我們設定了兩個路由,一個是用於send我們的客戶端html文件的,一個是用來監聽用戶輸入內容的。最后,我們監聽了本地的8080端口。

func home(w http.ResponseWriter, r *http.Request) {
  http.ServeFile(w, r, "home.html")
}

home函數用於發送客戶端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函數負責處理我們的websocket連接。首先,我們調用upgrader.Upgrade(w, r, nil)獲得這個連接的指針。然后,我們進入一個循環,不停的讀取連接中傳來的信息,當有信息傳來時,對信息處理后返回給客戶端。這里有一點要注意,就是c.ReadMessage()是阻塞的,,也就是說,當有消息來時,它后面的代碼才會被執行。

客戶端

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(...)建立一個連接,並為這個連接綁定幾個函數,這里我們可以設定onopen(建立連接),onmessage(收到服務端消息),onerror(發生錯誤)。然后,我們可以給我們的send按鈕綁定單擊事件,單擊后發送數據到后台。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM