Maison > développement back-end > Golang > le corps du texte

Comment configurer le serveur Go WebSocket ?

WBOY
Libérer: 2024-05-31 22:47:59
original
815 Les gens l'ont consulté

Comment créer un serveur Go WebSocket : installez la bibliothèque gorilla/websocket. Créez un serveur HTTP pour gérer les demandes de prise de contact WebSocket. Mettez à niveau les requêtes HTTP vers les connexions WebSocket. Gérez les messages WebSocket côté serveur, y compris l'envoi et la réception de données. Le serveur peut être étendu pour permettre aux clients de s'abonner à des canaux spécifiques et de recevoir des messages uniquement de ces canaux.

Go WebSocket 服务器如何搭建?

Créez un serveur Go WebSocket

WebSocket est un protocole de communication en duplex intégral qui permet aux clients et aux serveurs de transférer des données dans les deux sens via une seule connexion TCP. Le langage Go offre une prise en charge puissante pour la création de serveurs WebSocket. Cet article explique comment utiliser Go pour créer un serveur WebSocket, avec des exemples pratiques.

Démarrage

Tout d'abord, vous avez besoin d'un environnement de développement Go, qui comprend le compilateur Go et un éditeur de code. Vous devez également installer la bibliothèque gorilla/websocket, qui fournit une implémentation WebSocket dans Go. gorilla/websocket 库,它提供了 Go 语言的 WebSocket 实现。

go get -u github.com/gorilla/websocket
Copier après la connexion

创建服务器

WebSocket 服务器本质上是一个传统的 HTTP 服务器,但它处理的是 WebSocket 握手请求。下面是如何在 Go 中创建 WebSocket 服务器:

package main

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

    "github.com/gorilla/websocket"
)

func main() {
    // 升级 HTTP 请求到 WebSocket 连接
    upgrader := websocket.Upgrader{
        ReadBufferSize:  1024,
        WriteBufferSize: 1024,
    }

    http.HandleFunc("/ws", func(w http.ResponseWriter, r *http.Request) {
        conn, err := upgrader.Upgrade(w, r, nil)
        if err != nil {
            log.Fatal(err)
        }

        // websocket.Conn 是一个双向的连接
        for {
            // 从客户端接收消息
            messageType, p, err := conn.ReadMessage()
            if err != nil {
                log.Fatal(err)
            }

            // 将消息返回客户端
            if err := conn.WriteMessage(messageType, p); err != nil {
                log.Fatal(err)
            }
        }
    })

    http.ListenAndServe(":8080", nil)
}
Copier après la connexion

实战案例

现在,我们来扩展这个服务器,允许客户端订阅特定的频道并仅接收来自这些频道的消息。

package main

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

    "github.com/gorilla/websocket"
)

type channel struct {
    sync.Mutex
    clients map[*websocket.Conn]bool
}

var channels = struct {
    sync.RWMutex
    m map[string]*channel
}{m: make(map[string]*channel)}

func NewChannel(name string) *channel {
    channels.Lock()
    defer channels.Unlock()

    if _, ok := channels.m[name]; !ok {
        channels.m[name] = &channel{clients: make(map[*websocket.Conn]bool)}
    }
    return channels.m[name]
}

func main() {
    upgrader := websocket.Upgrader{
        ReadBufferSize:  1024,
        WriteBufferSize: 1024,
    }

    http.HandleFunc("/ws", func(w http.ResponseWriter, r *http.Request) {
        conn, err := upgrader.Upgrade(w, r, nil)
        if err != nil {
            log.Fatal(err)
        }

        // 订阅和取消订阅频道
        go func() {
            for {
                // 从客户端接收消息
                messageType, p, err := conn.ReadMessage()
                if err != nil {
                    log.Fatal(err)
                }

                // 如果消息是一种订阅请求,则将连接添加到相应频道
                if messageType == websocket.TextMessage && string(p)[:4] == "join" {
                    channelName := string(p)[5:]
                    channel := NewChannel(channelName)

                    // 锁定频道的客户端列表
                    channel.Lock()
                    channel.clients[conn] = true
                    // 解锁频道的客户端列表
                    channel.Unlock()
                }

                // 如果消息是一种取消订阅请求,则将连接从相应频道中移除
                if messageType == websocket.TextMessage && string(p)[:7] == "leave" {
                    channelName := string(p)[8:]
                    channel := NewChannel(channelName)

                    channel.Lock()
                    delete(channel.clients, conn)
                    channel.Unlock()
                }
            }
        }()

        // 广播消息到客户端
        go func() {
            for {
                // 分别广播来自每个频道的消息
                channels.RLock()
                for _, channel := range channels.m {
                    channel.RLock()
                    for client := range channel.clients {
                        if err := client.WriteMessage(websocket.TextMessage, "hello"); err != nil {
                            // 如果写入失败,则从频道中移除连接
                            channel.Lock()
                            delete(channel.clients, client)
                            channel.Unlock()
                        }
                    }
                    channel.RUnlock()
                }
                channels.RUnlock()
            }
        }()
    })

    http.ListenAndServe(":8080", nil)
}
Copier après la connexion

现在,你可以启动服务器并使用 WebSocket 客户端连接到 /wsrrreee

Créer un serveur🎜🎜Un serveur WebSocket est essentiellement un serveur HTTP traditionnel, mais il gère les demandes de prise de contact WebSocket. Voici comment créer un serveur WebSocket dans Go : 🎜rrreee🎜 Exemple pratique 🎜🎜 Maintenant, étendons ce serveur pour permettre aux clients de s'abonner à des chaînes spécifiques et de recevoir des messages uniquement de ces chaînes. 🎜rrreee🎜Vous pouvez maintenant démarrer le serveur et utiliser un client WebSocket pour vous connecter au point de terminaison /ws, vous abonner à la chaîne et recevoir des messages de la chaîne. 🎜

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!

Étiquettes associées:
go
source:php.cn
Déclaration de ce site Web
Le contenu de cet article est volontairement contribué par les internautes et les droits d'auteur appartiennent à l'auteur original. Ce site n'assume aucune responsabilité légale correspondante. Si vous trouvez un contenu suspecté de plagiat ou de contrefaçon, veuillez contacter admin@php.cn
Tutoriels populaires
Plus>
Derniers téléchargements
Plus>
effets Web
Code source du site Web
Matériel du site Web
Modèle frontal