Amalan pengaturcaraan pelayan bahasa: membina bilik sembang web yang ringkas

Bahasa Go semakin digunakan dalam bidang pengaturcaraan pelayan Ciri-cirinya yang pantas, cekap dan selaras telah menjadi sebab mengapa ramai orang memilihnya. Artikel ini akan memperkenalkan anda kepada cara membina bilik sembang Web ringkas dalam bahasa Go, termasuk penghantaran mesej masa nyata, interaksi dalam talian berbilang pengguna dan mata pengetahuan lain yang berkaitan.

1. Pemilihan rangka kerja web bahasa Go

Dalam bahasa Go, terdapat banyak rangka kerja web untuk dipilih, antaranya yang lebih popular ialah Beego, Gin, Echo, dll. Artikel ini memilih untuk menggunakan rangka kerja Echo kerana ia ringan, berprestasi tinggi dan mudah digunakan serta sesuai untuk membina aplikasi Web dengan cepat. Sebelum menggunakan rangka kerja Echo, sila buka terminal dan laksanakan arahan berikut untuk mendapatkan pakej kebergantungan:

go get -u github.com/labstack/echo

2. Reka bentuk berfungsi

Sebelum membina bilik sembang web ini, kita perlu mereka bentuknya fungsi yang diperlukan Modul dan proses:

1. Daftar dan log masuk ke akaun untuk mencapai pengesahan pengguna.
2. Halaman utama bilik sembang boleh memaparkan senarai pengguna dalam talian, rekod sembang sejarah, dsb.
3. Penghantaran mesej masa nyata, menyokong interaksi dalam talian berbilang pengguna.
4. Apabila penyemak imbas ditutup, sambungan WebSocket ke pelayan diputuskan dan pengguna ditandakan di luar talian.

3. Langkah pelaksanaan

Di bawah ini kami akan melaksanakan modul dan proses berfungsi di atas langkah demi langkah:

1. Daftar dan log masuk akaun:

Dalam bahasa Go, anda boleh menggunakan pakej pihak ketiga bcrypt untuk melaksanakan penyulitan kata laluan. Mula-mula kita perlu mencipta struktur pengguna Pengguna, dan kemudian menggunakan perpustakaan gorm untuk mencipta jadual pangkalan data.

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

Kod fungsi pemprosesan untuk pendaftaran dan log masuk pengguna adalah seperti berikut:

// 用户注册
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",
    })
}

2. Laman utama bilik sembang:

Dalam laman utama bilik sembang, dalam talian pengguna perlu dipaparkan Senarai dan sejarah sembang. Senarai pengguna dalam talian boleh disimpan dalam Peta, di mana kuncinya ialah nama pengguna dan nilainya ialah maklumat yang berkaitan dengan sambungan WebSocket, seperti ID sambungan, masa sambungan, dsb.

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

Rekod sembang sejarah boleh disimpan menggunakan tatasusunan atau struktur data seperti Redis. Berikut ialah kod fungsi pemprosesan halaman utama:

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

Antaranya, room.html ialah fail templat, pelanggan dan msg ialah senarai pengguna dalam talian dan rekod sembang sejarah masing-masing.

3. Penghantaran mesej masa nyata:

Dalam rangka kerja Echo, penghantaran mesej masa nyata boleh dicapai dengan sangat mudah menggunakan WebSockets. Kita perlu menentukan fungsi pemprosesan WebSocket Selepas pelanggan berjaya menyambung ke pelayan, maklumatnya akan ditambahkan ke senarai pengguna dalam talian apabila pelanggan menghantar mesej, mesej akan dihantar kepada semua pelanggan yang disambungkan.

// 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
}

4. Apabila menutup sambungan WebSocket, putuskan sambungan daripada pelayan dan tandai pengguna di luar talian. Kita boleh menggunakan pernyataan tangguh dalam fungsi pemprosesan WebSocket untuk memadam maklumat pengguna dalam senarai pengguna dalam talian pada penghujung fungsi.

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

4 Ringkasan

Artikel ini menggunakan rangka kerja Echo dan WebSockets untuk melaksanakan ruang sembang Web yang ringkas, termasuk pengesahan pengguna, senarai pengguna dalam talian, rekod sembang sejarah dan real- mesej masa Penghantaran dan fungsi lain. Pada masa yang sama, kami juga menggunakan perpustakaan gorm dan pakej bcrypt untuk melaksanakan pengurusan pengguna dan fungsi penyulitan kata laluan.

Dalam amalan, kami juga perlu mempertimbangkan isu keselamatan seperti penapisan mesej dan penapisan perkataan sensitif. Saya harap pembaca boleh belajar melalui artikel ini cara menggunakan bahasa Go untuk membina aplikasi web, serta perpustakaan dan rangka kerja yang biasa digunakan.

Atas ialah kandungan terperinci Amalan pengaturcaraan pelayan bahasa: membina bilik sembang web yang ringkas. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!