Maison  >  Article  >  développement back-end  >  Comment développer une salle de discussion Websocket en utilisant le langage Go

Comment développer une salle de discussion Websocket en utilisant le langage Go

PHPz
PHPzoriginal
2023-12-14 13:46:50895parcourir

Comment développer une salle de discussion Websocket en utilisant le langage Go

Comment développer une salle de discussion Websocket en utilisant le langage Go

Websocket est un protocole de communication en temps réel qui permet une communication bidirectionnelle entre le serveur et le client en établissant une connexion une fois. Websocket est un très bon choix lors du développement de salles de discussion car il permet l'échange de messages en temps réel et offre des performances efficaces. Cet article expliquera comment développer une salle de discussion Websocket simple à l'aide du langage Go et fournira quelques exemples de code spécifiques.

1. Préparation

1. Installez l'environnement linguistique Go

Avant de commencer, vous devez d'abord installer l'environnement linguistique Go. Le site officiel du langage Go (https://golang.org/) propose des packages d'installation pour chaque plateforme, qui peuvent être téléchargés et installés selon votre propre système d'exploitation.

2. Installez la bibliothèque websocket

En langage Go, utilisez le package "github.com/gorilla/websocket" pour fournir les fonctions liées à WebSocket. Le package peut être installé via la commande suivante :

go get github.com/gorilla/websocket

2 Créer un serveur de salle de discussion

Tout d'abord, nous devons créer un serveur simple pour gérer l'établissement des connexions et des messages Websocket. Envoyez et recevez.

package main

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

    "github.com/gorilla/websocket"
)

var clients = make(map[*websocket.Conn]bool) // 存储连接的客户端
var broadcast = make(chan Message)           // 消息广播通道

// 定义消息结构体
type Message struct {
    Username string `json:"username"`
    Message  string `json:"message"`
}

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

    go handleMessages()

    // 启动服务器
    log.Println("服务器启动,监听端口:8080")
    err := http.ListenAndServe(":8080", nil)
    if err != nil {
        log.Fatal("ListenAndServe: ", err)
    }
}

func handleConnections(w http.ResponseWriter, r *http.Request) {
    // 升级HTTP连接为Websocket连接
    ws, err := websocket.Upgrade(w, r, nil, 1024, 1024)
    if err != nil {
        log.Fatal(err)
    }

    // 关闭连接
    defer ws.Close()

    // 将客户端连接添加到clients映射中
    clients[ws] = true

    for {
        var msg Message
        // 读取客户端发送的消息
        err := ws.ReadJSON(&msg)
        if err != nil {
            log.Printf("读取消息错误:%v", err)
            delete(clients, ws)
            break
        }

        // 将接收到的消息放入广播通道中
        broadcast <- msg
    }
}

func handleMessages() {
    for {
        // 从广播通道中读取消息
        msg := <-broadcast

        // 遍历所有连接的客户端,将消息发送给每个客户端
        for client := range clients {
            err := client.WriteJSON(msg)
            if err != nil {
                log.Printf("发送消息错误:%v", err)
                client.Close()
                delete(clients, client)
            }
        }
    }
}

Le code ci-dessus crée un simple serveur de salle de discussion Websocket. Dans la fonction handleConnections, nous mettons le message reçu dans le canal de diffusion. La fonction handleMessages lit les messages du canal de diffusion et les envoie à tous les clients connectés. handleConnections函数中,我们将接收到的消息放入广播通道中。handleMessages函数从广播通道中读取消息,并将消息发送给所有连接的客户端。

三、创建前端界面

接下来,我们需要创建一个前端界面,用于用户输入用户名和消息,并实时显示聊天内容。

<!doctype html>
<html>

<head>
  <title>Websocket 聊天室</title>
  
  <style>
    #message-box {
      height: 500px;
      overflow-y: scroll;
    }
  </style>
</head>

<body>
  <h1>Websocket 聊天室</h1>

  <div id="message-box"></div>

  <form id="message-form" onsubmit="sendMessage(event)">
    <input type="text" id="username" placeholder="请输入用户名" required>
    <br />
    <input type="text" id="message" placeholder="请输入消息" required>
    <br />
    <button type="submit">发送</button>
  </form>

  <script>
    const socket = new WebSocket("ws://localhost:8080/ws");

    socket.onmessage = function (event) {
      const data = JSON.parse(event.data);
      const messageBox = document.getElementById("message-box");
      const messageElement = document.createElement("p");
      messageElement.innerText = data.username + ": " + data.message;
      messageBox.appendChild(messageElement);
    };

    function sendMessage(event) {
      event.preventDefault();

      const username = document.getElementById("username").value;
      const message = document.getElementById("message").value;

      const data = {
        username: username,
        message: message
      };

      socket.send(JSON.stringify(data));
      document.getElementById("message").value = "";
    }
  </script>
</body>

</html>

以上代码创建了一个简单的HTML界面,包含一个用户名输入框、一个消息输入框和一个发送按钮。通过socket.onmessage处理接收到的消息,并将消息显示在消息盒子中。通过socket.send发送输入的消息。

四、运行程序

在终端中进入项目目录,执行以下命令启动服务器:

go run main.go

然后,在浏览器中打开index.html

3. Créer une interface frontale

Ensuite, nous devons créer une interface frontale permettant aux utilisateurs de saisir leurs noms d'utilisateur et leurs messages, et d'afficher le contenu du chat en temps réel. 🎜rrreee🎜Le code ci-dessus crée une interface HTML simple, comprenant une zone de saisie du nom d'utilisateur, une zone de saisie du message et un bouton d'envoi. Traitez le message reçu via socket.onmessage et affichez le message dans la boîte de message. Envoyez le message d'entrée via socket.send. 🎜🎜4. Exécutez le programme 🎜🎜 Entrez le répertoire du projet dans le terminal et exécutez la commande suivante pour démarrer le serveur : 🎜rrreee🎜 Ensuite, ouvrez la page index.html dans le navigateur, saisissez le nom d'utilisateur et message, puis cliquez sur Envoyer. C'est tout. 🎜🎜Grâce aux étapes ci-dessus, nous avons développé avec succès un salon de discussion Websocket simple utilisant le langage Go. En développement actuel, davantage de fonctions peuvent être ajoutées, telles que les enregistrements de chat, les messages privés, etc. J'espère que cet article pourra vous aider à comprendre et à apprendre le langage Go pour développer des salles de discussion 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!

Déclaration:
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