快捷導(dǎo)航

Golang?WebSocket創(chuàng)建單獨(dú)會話詳細(xì)實(shí)例

更新時(shí)間：2023年11月16日 11:11:31 作者：一只會寫程序的貓

這篇文章主要給大家介紹了關(guān)于Golang?WebSocket創(chuàng)建單獨(dú)會話的相關(guān)資料,WebSocket 協(xié)議主要為了解決基于 HTTP/1.x 的 Web 應(yīng)用無法實(shí)現(xiàn)服務(wù)端向客戶端主動推送的問題,文中通過代碼介紹的非常詳細(xì),需要的朋友可以參考下

引言

在互聯(lián)網(wǎng)應(yīng)用程序中，實(shí)時(shí)通信是一種非常重要的功能。WebSocket 是一種基于 TCP 的協(xié)議，它允許客戶端和服務(wù)器之間進(jìn)行雙向通信。Golang 是一種高性能的編程語言，它提供了對 WebSocket 的原生支持，使得在 Golang 中創(chuàng)建 WebSocket 會話變得非常簡單。本文將介紹如何使用 Golang 創(chuàng)建單獨(dú)的 WebSocket 會話，以實(shí)現(xiàn)實(shí)時(shí)通信功能。

WebSocket 簡介

WebSocket 是一種在單個(gè) TCP 連接上進(jìn)行全雙工通信的協(xié)議。它與傳統(tǒng)的 HTTP 協(xié)議不同，HTTP 是一種無狀態(tài)的協(xié)議，每個(gè)請求都需要建立一個(gè)新的連接。而 WebSocket 在客戶端和服務(wù)器之間建立一條持久的連接，可以實(shí)現(xiàn)實(shí)時(shí)的雙向通信。

WebSocket 協(xié)議的握手是通過 HTTP 請求完成的，握手后，客戶端和服務(wù)器之間的連接將保持打開狀態(tài)，可以發(fā)送和接收任意數(shù)據(jù)。WebSocket 使用一種類似于事件的機(jī)制，當(dāng)有新消息到達(dá)時(shí)，服務(wù)器可以主動推送給客戶端，而不需要客戶端主動發(fā)送請求。

Golang 中的 WebSocket 支持

Golang 提供了 net/http 包來處理 HTTP 請求和響應(yīng)，同時(shí)也提供了 gorilla/websocket 庫來實(shí)現(xiàn) WebSocket 協(xié)議的支持。gorilla/websocket 是一個(gè)非常流行的第三方庫，它提供了對 WebSocket 協(xié)議的高級封裝，使得在 Golang 中創(chuàng)建 WebSocket 會話變得更加簡單。

在開始之前，我們首先需要安裝 gorilla/websocket 庫?？梢允褂靡韵旅顏戆惭b：

go get github.com/gorilla/websocket

安裝完成后，我們就可以開始創(chuàng)建 WebSocket 會話了。

創(chuàng)建 WebSocket 服務(wù)器

首先，我們需要?jiǎng)?chuàng)建一個(gè) WebSocket 服務(wù)器，用于接收來自客戶端的連接請求，并處理 WebSocket 會話。以下是一個(gè)簡單的 WebSocket 服務(wù)器示例：

package main
import (
    "log"
    "net/http"
    "github.com/gorilla/websocket"
)
// 創(chuàng)建一個(gè) upgrader 對象，用于升級 HTTP 連接為 WebSocket 連接
var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
}
func main() {
    http.HandleFunc("/ws", handleWebSocket)
    log.Fatal(http.ListenAndServe(":8080", nil))
}
func handleWebSocket(w http.ResponseWriter, r *http.Request) {
    // 將 HTTP 連接升級為 WebSocket 連接
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println("Upgrade failed:", err)
        return
    }
    defer conn.Close()
    // 處理 WebSocket 會話
    for {
        // 讀取客戶端發(fā)送的消息
        messageType, message, err := conn.ReadMessage()
        if err != nil {
            log.Println("Read message failed:", err)
            break
        }
        // 處理客戶端發(fā)送的消息
        log.Printf("Received message: %s", message)
        // 回復(fù)客戶端消息
        err = conn.WriteMessage(messageType, message)
        if err != nil {
            log.Println("Write message failed:", err)
            break
        }
    }
}

在上面的示例中，我們首先創(chuàng)建了一個(gè) upgrader 對象，它用于將 HTTP 連接升級為 WebSocket 連接。然后，我們定義了一個(gè) handleWebSocket 函數(shù)，用于處理 WebSocket 會話。在該函數(shù)中，我們首先將 HTTP 連接升級為 WebSocket 連接，然后進(jìn)入一個(gè)無限循環(huán)，不斷讀取客戶端發(fā)送的消息，并給客戶端回復(fù)相同的消息。

最后，我們使用 http.HandleFunc 函數(shù)將 /ws 路徑映射到 handleWebSocket 函數(shù)，然后調(diào)用 http.ListenAndServe 函數(shù)來啟動 WebSocket 服務(wù)器。

創(chuàng)建 WebSocket 客戶端

接下來，我們需要?jiǎng)?chuàng)建一個(gè) WebSocket 客戶端，用于向服務(wù)器發(fā)送 WebSocket 請求，并處理服務(wù)器推送的消息。以下是一個(gè)簡單的 WebSocket 客戶端示例：

package main
import (
    "log"
    "os"
    "os/signal"
    "time"
    "github.com/gorilla/websocket"
)
func main() {
    // 創(chuàng)建一個(gè) dialer 對象，用于建立 WebSocket 連接
    dialer := websocket.Dialer{
        HandshakeTimeout: 10 * time.Second,
    }
    // 建立 WebSocket 連接
    conn, _, err := dialer.Dial("ws://localhost:8080/ws", nil)
    if err != nil {
        log.Fatal("Dial failed:", err)
    }
    defer conn.Close()
    // 啟動一個(gè) goroutine 來接收服務(wù)器推送的消息
    go func() {
        for {
            _, message, err := conn.ReadMessage()
            if err != nil {
                log.Println("Read message failed:", err)
                break
            }
            log.Printf("Received message: %s", message)
        }
    }()
    // 向服務(wù)器發(fā)送消息
    message := []byte("Hello, WebSocket!")
    err = conn.WriteMessage(websocket.TextMessage, message)
    if err != nil {
        log.Println("Write message failed:", err)
        return
    }
    // 等待用戶按下 Ctrl+C 終止程序
    interrupt := make(chan os.Signal, 1)
    signal.Notify(interrupt, os.Interrupt)
    <-interrupt
}

在上面的示例中，我們首先創(chuàng)建了一個(gè) dialer 對象，它用于建立 WebSocket 連接。然后，我們使用 dialer.Dial 函數(shù)建立 WebSocket 連接，并指定服務(wù)器的地址為 ws://localhost:8080/ws。然后，我們使用 conn.WriteMessage 函數(shù)向服務(wù)器發(fā)送消息，并使用一個(gè) goroutine 來接收服務(wù)器推送的消息。

最后，我們使用 signal.Notify 函數(shù)來注冊一個(gè)信號監(jiān)聽器，當(dāng)用戶按下 Ctrl+C 時(shí)，程序會接收到一個(gè)中斷信號，然后程序退出。

創(chuàng)建單獨(dú)會話

在實(shí)際應(yīng)用中，我們可能需要為每個(gè)客戶端創(chuàng)建一個(gè)單獨(dú)的會話，以便管理和跟蹤客戶端的狀態(tài)。在 Golang 中，可以通過為每個(gè) WebSocket 連接創(chuàng)建一個(gè)獨(dú)立的 goroutine 來實(shí)現(xiàn)這一點(diǎn)。以下是一個(gè)示例：

package main
import (
    "log"
    "net/http"
    "github.com/gorilla/websocket"
)
var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
}
func main() {
    http.HandleFunc("/ws", handleWebSocket)
    log.Fatal(http.ListenAndServe(":8080", nil))
}
func handleWebSocket(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println("Upgrade failed:", err)
        return
    }
    defer conn.Close()
    // 創(chuàng)建一個(gè)獨(dú)立的會話
    session := NewSession()
    // 處理會話
    session.Handle(conn)
}
type Session struct {
    conn *websocket.Conn
}
func NewSession() *Session {
    return &Session{}
}
func (s *Session) Handle(conn *websocket.Conn) {
    s.conn = conn
    go s.readLoop()
    go s.writeLoop()
}
func (s *Session) readLoop() {
    for {
        messageType, message, err := s.conn.ReadMessage()
        if err != nil {
            log.Println("Read message failed:", err)
            break
        }
        log.Printf("Received message: %s", message)
    }
}
func (s *Session) writeLoop() {
    for {
        select {
        // 從消息隊(duì)列中獲取消息并發(fā)送給客戶端
        case message := <-s.messageQueue:
            err := s.conn.WriteMessage(websocket.TextMessage, message)
            if err != nil {
                log.Println("Write message failed:", err)
            }
        }
    }
}

在上面的示例中，我們首先定義了一個(gè) Session 結(jié)構(gòu)體，它包含一個(gè) WebSocket 連接。然后，我們創(chuàng)建了一個(gè) NewSession 函數(shù)，用于創(chuàng)建一個(gè)新的會話對象。會話對象有兩個(gè)重要的方法：Handle 方法用

場景

Golang的WebSocket可以用于創(chuàng)建單獨(dú)的會話，適用于許多場景。以下是一個(gè)使用場景的介紹：

場景：在線聊天室

在一個(gè)在線聊天室中，用戶可以通過WebSocket與其他用戶進(jìn)行實(shí)時(shí)的文字交流。每個(gè)用戶都可以創(chuàng)建一個(gè)單獨(dú)的會話，與其他用戶進(jìn)行私聊或在群組中發(fā)送消息。

使用場景描述：

用戶進(jìn)入聊天室，并在前端頁面上輸入昵稱和聊天內(nèi)容。
前端頁面通過WebSocket與后端的Golang服務(wù)器建立連接。
后端服務(wù)器使用Golang的WebSocket包處理客戶端的連接請求。
當(dāng)用戶發(fā)送消息時(shí)，前端頁面將消息通過WebSocket發(fā)送至后端服務(wù)器。
后端服務(wù)器接收到消息后，將其廣播給所有在線的用戶，或者根據(jù)需要僅發(fā)送給特定的用戶。
每個(gè)連接的客戶端都可以接收到其他用戶發(fā)送的消息，并在前端頁面上展示出來。

此場景中，Golang的WebSocket實(shí)現(xiàn)了用戶之間的實(shí)時(shí)通信，并保持了每個(gè)用戶的會話獨(dú)立性。它可以處理并發(fā)連接，使得多個(gè)用戶能夠同時(shí)進(jìn)行聊天，而不會相互干擾。

值得注意的是，Golang的WebSocket還可以通過添加必要的安全性和認(rèn)證機(jī)制來確保聊天室的安全性。例如，可以使用SSL/TLS加密連接，或者使用令牌進(jìn)行用戶身份驗(yàn)證。這些安全性措施可以確保用戶的聊天內(nèi)容和個(gè)人信息得到保護(hù)。

總結(jié)

到此這篇關(guān)于Golang WebSocket創(chuàng)建單獨(dú)會話的文章就介紹到這了,更多相關(guān)Go WebSocket創(chuàng)建單獨(dú)會話內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: