首頁  >  文章  >  後端開發  >  Go WebSocket 如何與其他協定配合使用?

Go WebSocket 如何與其他協定配合使用?

WBOY
WBOY原創
2024-06-05 12:58:561103瀏覽

Go WebSocket 可與其他協定配合使用,包括:HTTP/HTTPS:WebSocket 通常在 HTTP/HTTPS 之上運行,透過握手程序協商 WebSocket 協定。 gRPC:gRPC 是一種 RPC 框架,可與 WebSocket 集成,以便在客戶端和伺服器之間進行低延遲、高吞吐量的 RPC 呼叫。

Go WebSocket 如何与其他协议配合使用?

Go WebSocket 如何與其他協定配合使用

WebSocket是一種用於在客戶端和伺服器之間進行全雙工通訊的協定。它通常用於建立即時應用程序,例如聊天、訊息和遊戲。 WebSocket可以在廣泛的場景中與其他協定配合使用,例如HTTP、HTTPS、gRPC和WebSockets。

使用HTTP和HTTPS

WebSocket通常在HTTP或HTTPS之上運作。當客戶端建立與伺服器的WebSocket連線時,它會先傳送一個HTTP請求,其中包含一個名為"Upgrade"的特殊頭,其值設定為"websocket"。伺服器回應此請求,其中包含一個"Upgrade"頭,其值也設定為"websocket"。此握手過程使用戶端和伺服器能夠協商使用WebSocket協議,並在HTTP/HTTPS之上建立全雙工通訊通道。

import (
    "fmt"
    "log"
    "net/http"

    "github.com/gorilla/websocket"
)

const (
    port = ":8080"
)

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

func main() {
    http.HandleFunc("/", indexHandler)
    http.HandleFunc("/ws", websocketHandler)

    log.Printf("Listening on port %s", port)
    if err := http.ListenAndServe(port, nil); err != nil {
        log.Fatal(err)
    }
}

func indexHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintf(w, "Welcome to the WebSocket page.")
}

func websocketHandler(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Fatal(err)
    }
    defer conn.Close()

    for {
        mt, message, err := conn.ReadMessage()
        if err != nil {
            log.Println(err)
            conn.Close()
            break
        }

        if err := conn.WriteMessage(mt, message); err != nil {
            log.Println(err)
            conn.Close()
            break
        }
    }
}

使用gRPC

gRPC是一種高效能遠端過程呼叫(RPC)框架,可以與WebSocket一起使用。客戶端使用gRPC函數呼叫伺服器端方法,伺服器端方法使用WebSocket傳送回應。此整合允許應用程式以低延遲、高吞吐量的方式處理RPC呼叫。

import (
    "context"
    "fmt"
    "log"
    "net/http"

    "github.com/grpc-ecosystem/grpc-gateway/v2/runtime"
    "github.com/gorilla/websocket"
    "google.golang.org/protobuf/proto"
)

const (
    port  = ":8080"
    grpcAddr = ":50051"
)

var upgrader = websocket.Upgrader{}
var mux *runtime.ServeMux

func main() {
    mux = runtime.NewServeMux()

    grpcConn, err := grpc.DialContext(
        context.Background(),
        grpcAddr,
        grpc.WithInsecure(),
        )
    if err != nil {
        log.Fatal(err)
    }

    // 注册 gRPC 服务到网关。
    if err := helloworldpb.RegisterGreeterHandler(context.Background(), mux, grpcConn); err != nil {
        log.Fatal(err)
    }

    http.HandleFunc("/", indexHandler)
    http.HandleFunc("/grpc-ws", websocketHandler)

    log.Printf("Listening on port %s", port)
    if err := http.ListenAndServe(port, nil); err != nil {
        log.Fatal(err)
    }
}

func indexHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintf(w, "Welcome to the gRPC-WebSocket page.")
}

func websocketHandler(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Fatal(err)
    }
    defer conn.Close()

    for {
        mt, message, err := conn.ReadMessage()
        if err != nil {
            log.Println(err)
            conn.Close()
            break
        }

        req := &helloworldpb.HelloRequest{}
        if err := proto.Unmarshal(message, req); err != nil {
            log.Println(err)
            conn.Close()
            break
        }

        ctx := context.Background()
        resp, err := helloworldpb.NewGreeterClient(grpcConn).SayHello(ctx, req)
        if err != nil {
            log.Println(err)
            conn.Close()
            break
        }

        message, err := proto.Marshal(resp)
        if err != nil {
            log.Println(err)
            conn.Close()
            break
        }

        if err := conn.WriteMessage(mt, message); err != nil {
            log.Println(err)
            conn.Close()
            break
        }
    }
}

以上是Go WebSocket 如何與其他協定配合使用?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn