Comment optimiser les performances des applications Websocket en langage Go

WebSocket est devenu un protocole très populaire dans le développement d'applications Web modernes en temps réel. Lors de l'écriture d'applications à l'aide de WebSocket, nous devons prendre en compte l'optimisation de ses performances pour garantir que notre application peut répondre aux demandes des clients rapidement et avec précision. Dans cet article, nous expliquons comment optimiser les performances des applications Go WebSocket et fournissons des exemples de code concrets.

1. Utilisez la bonne bibliothèque WebSocket

Le langage Go propose plusieurs bibliothèques WebSocket populaires, telles que Gorilla WebSocket, Gobwas WebSocket et Fasthttp WebSocket. Parmi elles, la bibliothèque Gorilla WebSocket est l'une des bibliothèques les plus utilisées et offre plus de fonctionnalités que les autres bibliothèques. Lorsque vous choisissez une bibliothèque WebSocket, vous devez tenir compte de ses performances, de ses fonctionnalités et de sa facilité d'utilisation.

Dans cet article, nous utiliserons la bibliothèque Gorilla WebSocket pour faire une démonstration.

2. Utilisez les connexions WebSocket de manière appropriée

Lors de la conception d'applications WebSocket, nous devons éviter autant que possible les connexions inutiles. Chaque connexion WebSocket consomme des ressources du serveur, donc si une opération qui aurait pu être effectuée via une seule connexion entraîne plusieurs connexions car la connexion n'est pas planifiée, le serveur sera surchargé. Il est recommandé de créer des connexions lorsque cela est nécessaire et d'utiliser des connexions de longue durée aussi souvent que possible pour éviter d'avoir à établir de nouvelles connexions.

Regardons un exemple de code pour créer une connexion WebSocket à l'aide de la bibliothèque Gorilla WebSocket :

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(err)
        return
    }
    defer conn.Close()

    // use the websocket connection here
}

Dans l'exemple de code ci-dessus, nous avons créé une fonction handleWebSocket pour gérer la connexion WebSocket. Dans cette fonction, nous utilisons la fonction upgrader.Upgrade() pour mettre à niveau la connexion HTTP vers une connexion WebSocket. Notez que la fonction defer conn.Close() est utilisée ici pour garantir que la connexion WebSocket est fermée à la fin de la fonction.

3. Configurer correctement WebSocket

Lorsque le nombre de connexions atteint un certain niveau, l'équilibrage de charge de la configuration de WebSocket est très important. Pour les serveurs, WebSocket dispose de deux paramètres de configuration particulièrement importants : ReadBufferSize et WriteBufferSize. Ces deux paramètres contrôlent la taille du tampon de lecture et du tampon d'écriture de la connexion WebSocket. Un tampon trop grand peut affecter les performances de connexion, tandis qu'un tampon trop petit peut augmenter le nombre de transferts de données supplémentaires.

Lors de l'utilisation de la bibliothèque Gorilla WebSocket, nous pouvons modifier la taille du tampon en :

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

Dans l'exemple de code ci-dessus, nous définissons la taille de ReadBufferSize et WriteBufferSize à 1024 octets. Veuillez définir la taille appropriée en fonction des besoins réels.

4. Utilisation du traitement simultané

Les applications WebSocket doivent prendre en charge un grand nombre de connexions simultanées, elles doivent donc utiliser des goroutines pour gérer chaque connexion. Vous pouvez utiliser le mécanisme goroutine fourni par la bibliothèque standard du langage Go pour gérer plusieurs connexions WebSocket. Transmettez simplement les connexions WebSocket créées aux goroutines et elles géreront facilement chaque connexion.

Voici un exemple de code qui utilise le traitement simultané des connexions WebSocket :

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(err)
        return
    }

    go func(conn *websocket.Conn) {
        for {
            _, message, err := conn.ReadMessage()
            if err != nil {
                log.Println(err)
                return
            }

            log.Printf("received message: %s", message)
            // handle the message here
        }
    }(conn)
}

Dans l'exemple de code ci-dessus, nous utilisons goroutine pour gérer chaque connexion WebSocket. Dans chaque goroutine, nous recevons des messages WebSocket à l'aide de la fonction conn.ReadMessage(). Nous pouvons ensuite traiter les messages dans chaque goroutine.

5. Utilisation efficace de la mémoire

Dans chaque connexion WebSocket, le tampon créé consomme beaucoup de mémoire. Nous devons donc garantir une utilisation maximale de la mémoire. Voici quelques suggestions :

• Cache les messages à envoyer et évite les allocations de mémoire fréquentes lors de l'écriture.
• Utilisez un mécanisme de récupération de place régulier pour éviter les fuites de mémoire causées par des pointeurs non référencés.
• Évitez de créer des objets volumineux ou d'appeler des bibliothèques ou des fonctions peu performantes dans le traitement des messages WebSocket.

Par exemple, l'exemple suivant montre comment mettre en cache les messages et nettoyer le cache périodiquement :

type Connection struct {
    conn    *websocket.Conn
    send    chan []byte
}

func (c *Connection) read() {
    for {
        _, _, err := c.conn.ReadMessage()
        if err != nil {
            break
        }
    }
    c.conn.Close()
}

func (c *Connection) write() {
    ticker := time.NewTicker(10 * time.Second)
    defer func() {
        ticker.Stop()
        c.conn.Close()
    }()

    var messages [][]byte
    for {
        select {
        case message, ok := <-c.send:
            if !ok {
                c.conn.WriteMessage(websocket.CloseMessage, []byte{})
                return
            }
            messages = append(messages, message)

        case <-ticker.C:
            if err := c.conn.WriteMessage(websocket.TextMessage, bytes.Join(messages, []byte{})); err != nil {
                return
            }
            messages = nil
        }
    }
}

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(err)
        return
    }
    connection := &Connection{conn: conn, send: make(chan []byte, 256)}
    go connection.write()
    go connection.read()

    // use the connection here
}

Dans l'exemple de code ci-dessus, nous avons créé une structure de connexion avec deux champs : conn et send. Le champ d'envoi est un canal avec un tampon dans lequel tous les messages sont mis en mémoire tampon. Nous utilisons ensuite le téléscripteur pour effacer et envoyer périodiquement des messages.

Résumé :

Pour optimiser les performances des applications WebSocket en langage Go, vous devez prendre en compte les aspects suivants :

• Utilisez la bonne bibliothèque WebSocket
• Utilisez la connexion WebSocket de manière raisonnable
• Configurez WebSocket de manière raisonnable
• Utilisez le traitement simultané
• Utilisez efficacement la mémoire

Ceux-ci ci-dessus sont quelques-uns des moyens les plus efficaces pour optimiser les performances des applications WebSocket en langage Go. Bien que l'exemple de code présenté dans cet article ne soit pas exhaustif, vous devez suivre les recommandations ci-dessus pour améliorer les performances des applications lorsque vous développez des applications WebSocket.

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!