使用 x/net 庫 trowing 403 的 Websocket 伺服器實現-Golang-PHP中文網

使用 x/net 庫 trowing 403 的 Websocket 伺服器實現

WBOY

發布： 2024-02-13 17:03:08

轉載

965 人瀏覽過

使用 x/net 库 trowing 403 的 Websocket 服务器实现

php小編香蕉為您介紹一種使用x/net函式庫的方法，用於實作一個拋出403錯誤的Websocket伺服器。 Websocket是一種在客戶端和伺服器之間建立持久連接的通訊協議，而403錯誤表示伺服器拒絕了客戶端的請求。透過使用x/net函式庫，我們可以輕鬆地建立一個Websocket伺服器，並在需要的時候拋出403錯誤，從而實現對請求的拒絕。這種方法簡單有效，適用於需要對請求進行授權或限制存取的場景。

問題內容

我正在嘗試使用 x/net/websocket 標準庫實作一個 websocket 伺服器。

到目前為止我的嘗試如下：

package main

import (
    "fmt"
    "net/http"

    "golang.org/x/net/websocket"
)

type Server struct {
    baseUri     string
    connections map[string][]*websocket.Conn
}

func initServer(baseUri string) *Server {
    return &Server{
        baseUri: baseUri,
    }
}

func (server *Server) handleConnections() {
    http.Handle("/ws", websocket.Handler(server.listenConnections))
    http.ListenAndServe(":3000", nil)
}

func (server *Server) listenConnections(ws *websocket.Conn) {
    fmt.Println("New connection established")
    for {
        fmt.Println("FOO")
    }
}

func main() {
    server := initServer("/ws")
    server.handleConnections()
}

登入後複製

嘗試使用多個 ws 用戶端連線到 ws://localhost:3000/ws 時，我總是收到相同的錯誤：403-Forbidden。我什至嘗試了官方文件的範例，但仍然得到它。我錯過了一些明顯的東西嗎？像預設連接埠阻塞或類似的嗎？

提前謝謝您。

編輯：您可能需要使用不同的連接埠來重現問題。在我的範例中使用 3000 ，如果不可用，只會中斷程式的執行。

編輯2：您可以使用像websocat這樣的客戶端並執行websocat 'ws://localhost:3000/ws' 來嘗試連線到伺服器

解決方法

我放棄了，但有很好的見解：如果您像我一樣正在關注 Anthony GG 的從頭開始在 Go 上建立 Websocket 伺服器的演練，請不要。影片已經過時了，儘管它提供瞭如何創建影片的很好的直覺，但最好（並且沒有恥辱）使用 gorilla 的 websocket 庫進行學習。

package main

import (
    "fmt"
    "net/http"
    "time"

    "github.com/gorilla/mux"
    "github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
}

type Server struct {
    baseUri     string
    connections map[string][]*websocket.Conn
    router      *mux.Router
    setup       *http.Server
}

func initServer(baseUri string) *Server {
    router := mux.NewRouter()
    return &Server{
        baseUri: baseUri,
        router:  router,
        setup: &http.Server{
            Handler:      router,
            Addr:         "127.0.0.1:8000",
            WriteTimeout: 15 * time.Second,
            ReadTimeout:  15 * time.Second,
        },
    }
}

func (server *Server) handleConnections() {
    server.router.HandleFunc("/ws/{var}", server.listenConnections)
    server.setup.ListenAndServe()
}

func (server *Server) listenConnections(w http.ResponseWriter, r *http.Request) {
    connection, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        fmt.Println(err)
        return
    }
    for {
        _, message, err := connection.ReadMessage()
        if err != nil {
            break
        }

        connection.WriteMessage(websocket.TextMessage, message)
        go messageHandler(message)
    }
    fmt.Println("Out of loop")
}

func messageHandler(message []byte) {
    fmt.Println(string(message))
}

func main() {
    server := initServer("/ws")
    server.handleConnections()
}

登入後複製

我還使用 gorilla/mux 來使用路徑參數（不知道為什麼 http 處理程序不能）。請注意我如何將 http.Handle 更改為 mux.Router.HandleFunc。正如用戶 @Cerise 在評論中指出的那樣，x/net/websocket 套件不在標準庫中，但僅添加 Origin 標頭也沒有解決原始問題。

希望這能避免其他像我一樣學習 Go 的人遇到的一些麻煩。

以上是使用 x/net 庫 trowing 403 的 Websocket 伺服器實現的詳細內容。更多資訊請關注PHP中文網其他相關文章！