Pratique de programmation de serveur de langage Go : création d'une salle de discussion Web simple-Golang-php.cn

Pratique de programmation de serveur de langage Go : création d'une salle de discussion Web simple

WBOY

Libérer： 2023-06-18 08:27:17

original

1813 Les gens l'ont consulté

Le langage

Go est de plus en plus utilisé dans le domaine de la programmation serveur. Ses fonctionnalités rapides, efficaces et concurrentes sont devenues les raisons pour lesquelles de nombreuses personnes le choisissent. Cet article vous expliquera comment créer une salle de discussion Web simple en langage Go, y compris la transmission de messages en temps réel, l'interaction en ligne multi-utilisateurs et d'autres points de connaissances connexes.

1. Sélection du framework Web du langage Go

Dans le langage Go, il existe de nombreux frameworks Web parmi lesquels choisir, parmi lesquels les plus populaires sont Beego, Gin, Echo, etc. . Cet article choisit d'utiliser le framework Echo car il est léger, performant et facile à utiliser, et convient à la création rapide d'applications Web. Avant d'utiliser le framework Echo, veuillez ouvrir un terminal et exécuter la commande suivante pour obtenir le package de dépendances :

go get -u github.com/labstack/echo

Copier après la connexion

2. Conception fonctionnelle

Avant de construire ce salon de discussion Web, nous besoin de le concevoir Modules fonctionnels et processus requis :

Inscrivez-vous et connectez-vous au compte pour obtenir l'authentification de l'utilisateur.
La page d'accueil du salon de discussion peut afficher la liste des utilisateurs en ligne, l'historique des enregistrements de discussion, etc.
Transmission de messages en temps réel, prenant en charge l'interaction en ligne multi-utilisateurs.
Lorsque le navigateur est fermé, la connexion WebSocket au serveur est déconnectée et l'utilisateur est marqué comme hors ligne.

3. Étapes de mise en œuvre

Ci-dessous, nous mettrons en œuvre les modules fonctionnels et les processus ci-dessus étape par étape :

Inscrivez-vous et connectez-vous au compte :

En langage Go, vous pouvez utiliser le package tiers bcrypt pour implémenter le cryptage des mots de passe. Nous devons d'abord créer la structure utilisateur User, puis utiliser la bibliothèque gorm pour créer la table de base de données.

type User struct {
    gorm.Model
    Username string `gorm:"unique;not null"`
    Password string `gorm:"not null"`
}

Copier après la connexion

Le code de fonction de traitement pour l'enregistrement et la connexion des utilisateurs est le suivant :

// 用户注册
func Register(c echo.Context) error {
    username := c.FormValue("username")
    password := c.FormValue("password")
    confirmPwd := c.FormValue("confirm_pwd")
    u := &models.User{
        Username: username,
        Password: util.HashPwd(password),
    }
    if password != confirmPwd {
        return c.JSON(http.StatusOK, map[string]interface{}{
            "success": false,
            "message": "Passwords do not match",
        })
    }
    err := db.Create(&u).Error
    if err != nil {
        return c.JSON(http.StatusOK, map[string]interface{}{
            "success": false,
            "message": "Username already exists",
        })
    }
    return c.JSON(http.StatusOK, map[string]interface{}{
        "success": true,
        "message": "Register success",
    })
}

// 用户登录
func Login(c echo.Context) error {
    username := c.FormValue("username")
    password := c.FormValue("password")
    user := &models.User{}
    err := db.Where("username = ?", username).First(&user).Error
    if err != nil {
        return c.JSON(http.StatusOK, map[string]interface{}{
            "success": false,
            "message": "Username or password is incorrect",
        })
    }
    if util.CheckPwd(password, user.Password) {
        return c.JSON(http.StatusOK, map[string]interface{}{
            "success": true,
            "message": "Login success",
        })
    }
    return c.JSON(http.StatusOK, map[string]interface{}{
        "success": false,
        "message": "Username or password is incorrect",
    })
}

Copier après la connexion

Page d'accueil du salon de discussion :

#🎜🎜 # discute Sur la page d'accueil de la salle, la liste des utilisateurs en ligne et l'historique des enregistrements de discussion doivent être affichés. La liste des utilisateurs en ligne peut être enregistrée dans une carte, où la clé est le nom d'utilisateur et la valeur est des informations liées à la connexion WebSocket, telles que l'ID de connexion, l'heure de connexion, etc.

// 在线用户列表
var clients = make(map[string]*Client)

Copier après la connexion

Les enregistrements de discussion historiques peuvent être stockés à l'aide de tableaux ou de structures de données de type Redis. Voici le code de la fonction de traitement de la page d'accueil :

// 聊天室首页
func Room(c echo.Context) error {
    return c.Render(http.StatusOK, "room.html", map[string]interface{}{
        "clients": clients,
        "msgs":    msgs,
    })
}

Copier après la connexion

Parmi eux, room.html est le fichier modèle, les clients et les msgs sont respectivement la liste des utilisateurs en ligne et les enregistrements de discussion historiques.

Dans le framework Echo, la transmission de messages en temps réel peut être réalisée très simplement à l'aide de WebSockets. Nous devons définir une fonction de traitement WebSocket. Une fois que le client s'est connecté avec succès au serveur, ses informations seront ajoutées à la liste des utilisateurs en ligne ; lorsque le client envoie un message, le message sera envoyé à tous les clients connectés.

// WebSocket处理函数
func wsHandler(c echo.Context) error {
    // 建立WebSocket连接
    ws, err := upgrader.Upgrade(c.Response(), c.Request(), nil)
    if err != nil {
        return err
    }
    defer ws.Close()

    // 获取用户名
    username, ok := c.QueryParams()["username"]
    if !ok {
        return errors.New("username is required")
    }

    // 添加用户
    client := &Client{
        WS:        ws,
        Username: username[0],
        ConnTime:  time.Now(),
    }
    clients[client.Username] = client

    // 遍历用户列表，发送欢迎消息
    for _, c := range clients {
        msg := &Message{
            Type: WelcomeMsg,
            Data: fmt.Sprintf("User '%s' joined the chat room", client.Username),
        }
        err = c.Send(msg)
        if err != nil {
            log.Println(err)
            continue
        }
    }

    // 接收消息，广播给所有用户
    for {
        msg := &Message{}
        err = client.Receive(msg)
        if err != nil {
            if err == io.EOF {
                delete(clients, client.Username)
                for _, c := range clients {
                    msg = &Message{
                        Type: LeaveMsg,
                        Data: fmt.Sprintf("User '%s' left the chat room", client.Username),
                    }
                    err = c.Send(msg)
                    if err != nil {
                        log.Println(err)
                        continue
                    }
                }
                break
            }
            log.Println(err)
            continue
        }
        for _, c := range clients {
            if c.Username == client.Username {
                continue
            }
            err = c.Send(msg)
            if err != nil {
                log.Println(err)
                continue
            }
        }
        msgs = append(msgs, *msg)
    }

    return nil
}

Copier après la connexion

defer func() {
    delete(clients, client.Username)
}()

Copier après la connexion

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!