Rumah > Artikel > pembangunan bahagian belakang > Golang membina soket web

Golang membina soket web

WBOYasal: 2023-05-22 19:21:071742semak imbas

Golang ialah bahasa pengaturcaraan yang kompilasi pantas, cekap dan selamat yang juga digunakan secara meluas dalam pembangunan web. Antaranya, menggunakan Golang untuk membina Websocket adalah senario aplikasi yang sangat popular. Artikel ini akan memperkenalkan cara menggunakan Golang untuk membina aplikasi Websocket.

1. Apakah itu Websocket

Websocket ialah protokol yang membenarkan penubuhan sambungan komunikasi dua hala antara penyemak imbas dan pelayan. Protokol HTTP tradisional hanya boleh menghantar permintaan kepada pelayan oleh penyemak imbas dan menerima respons daripada pelayan. Walau bagaimanapun, menggunakan Websocket membolehkan pelayan menghantar mesej secara aktif kepada penyemak imbas, merealisasikan komunikasi dupleks penuh antara pelayan dan penyemak imbas.

2. Cara Golang melaksanakan Websocket

Golang disertakan dengan perpustakaan standard net/http Pakej ini menyediakan struktur http.Server, yang boleh digunakan untuk membina pelayan Web. Di samping itu, net/http juga menyediakan websocket subpakej, yang memberikan kami semua fungsi yang diperlukan untuk membina pelayan WebSocket.

Mewujudkan sambungan WebSocket

Untuk mewujudkan sambungan WebSocket, pelanggan perlu menghantar permintaan HTTP ke pelayan dengan pengepala Naik Taraf yang menunjukkan bahawa permintaan itu dinaik taraf kepada protokol WebSocket. Selepas menerima permintaan ini, pelayan perlu menyemak pengepala Naik Taraf dan memprosesnya dengan sewajarnya bergantung pada sama ada ia adalah permintaan WebSocket.

Pihak pelayan boleh menggunakan fungsi websocket.IsWebSocketUpgrade untuk menyemak sama ada permintaan itu adalah permintaan WebSocket dan menggunakan fungsi websocket.Upgrade untuk meningkatkan permintaan. Selepas peningkatan berjaya, pelayan boleh berkomunikasi dengan klien melalui objek sambungan jenis *websocket.Conn.

func handler(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println(err)
        return
    }
    defer conn.Close()
    // 使用conn进行通信
}

Antaranya, upgrader ialah pembolehubah jenis websocket.Upgrader Dengan menetapkan sifatnya, anda boleh mengawal kaedah pemprosesan sambungan WebSocket. UpgradeFungsi akan menyekat apabila menaik taraf permintaan, jadi ia harus dijalankan dalam goroutine.

Menghantar dan menerima mesej WebSocket

Setelah sambungan WebSocket berjaya diwujudkan, kedua-dua pelanggan dan pelayan boleh berkomunikasi melalui objek *websocket.Conn. Gunakan objek ini untuk menghantar dan menerima mesej WebSocket.

Menghantar mesej adalah sangat mudah, cuma panggil kaedah WriteMessage:

err := conn.WriteMessage(websocket.TextMessage, []byte("hello"))

Terdapat tiga cara untuk menerima mesej:

ReadMessage: menyekat baca Ambil sehingga mesej sampai
NextReader: Sekat dan tunggu mesej seterusnya sampai
NextWriter: Kembalikan objek io.WriteCloser, yang boleh digunakan untuk menghantar mesej

Umum menggunakan kaedah ReadMessage untuk menerima mesej WebSocket:

_, message, err := conn.ReadMessage()
if err != nil {
    log.Println(err)
    return
}
log.Printf("recv: %s", message)

3. aplikasi untuk klien Selepas klien menghantar mesej, pelayan menyiarkan mesej kepada semua klien yang disambungkan.

Kod sisi pelanggan: (menggunakan JavaScript)

<!DOCTYPE html>
<html>
<head>
    <title>Chatroom</title>
</head>
<body>
    <input type="text" id="input-message"/>
    <button onclick="sendMessage()">发送</button>
    <ul id="message-list"></ul>

    <script>
        var ws = new WebSocket("ws://localhost:8080/ws");
        ws.onmessage = function(event) {
            var message = event.data;
            var li = document.createElement("li");
            li.appendChild(document.createTextNode(message));
            document.getElementById("message-list").appendChild(li);
        };

        function sendMessage() {
            var message = document.getElementById("input-message").value;
            ws.send(message);
        }
    </script>
</body>
</html>

Kod sisi pelayan:

package main

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

    "github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{}

func main() {
    http.HandleFunc("/ws", wsHandler)
    err := http.ListenAndServe(":8080", nil)
    if err != nil {
        log.Fatal("ListenAndServe: ", err)
    }
}

func wsHandler(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println(err)
        return
    }
    defer conn.Close()

    for {
        _, message, err := conn.ReadMessage()
        if err != nil {
            log.Println(err)
            break
        }
        fmt.Printf("recv: %s
", message)

        // 广播给所有客户端
        for _, conn := range conns {
            err := conn.WriteMessage(websocket.TextMessage, message)
            if err != nil {
                log.Println(err)
                break
            }
        }
    }
}

Dalam kod di atas, kami menyimpan semua objek

yang disambungkan dalam

Dalam pembolehubah. Setiap kali mesej baru tiba, pelayan menyiarkan mesej itu kepada semua pelanggan. websocket.ConnconnsRingkasan

Sangat mudah untuk membina aplikasi WebSocket menggunakan Golang Anda hanya perlu mengimport pakej

dan

Anda boleh terus menggunakan API yang disediakan oleh standard perpustakaan untuk melengkapkan pembangunan aplikasi WebSocket. Saya harap artikel ini dapat membantu pembangunan Golang Websocket. net/http

Atas ialah kandungan terperinci Golang membina soket web. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!

Kenyataan：

Kandungan artikel ini disumbangkan secara sukarela oleh netizen, dan hak cipta adalah milik pengarang asal. Laman web ini tidak memikul tanggungjawab undang-undang yang sepadan. Jika anda menemui sebarang kandungan yang disyaki plagiarisme atau pelanggaran, sila hubungi admin@php.cn

Artikel sebelumnya：Senarai terbalik golangArtikel seterusnya：Senarai terbalik golang

Artikel berkaitan

Lihat lagi