Cara menggunakan Websocket dalam golang untuk memajukan data

Dalam beberapa tahun kebelakangan ini, dengan perkembangan pesat teknologi Internet, teknologi Websocket telah menarik perhatian dan perhatian yang semakin meningkat. Websocket ialah protokol komunikasi dupleks penuh berdasarkan protokol TCP, yang membolehkan komunikasi dua hala antara aplikasi web dan pelayan. Menggunakan teknologi Websocket, sambungan masa nyata yang kritikal boleh diwujudkan sambil mengurangkan kelewatan rangkaian dan meningkatkan kecekapan penghantaran data. Artikel ini akan memperkenalkan cara menggunakan teknologi Websocket untuk memajukan data dalam golang.

1. Cara Websocket berfungsi

Proses komunikasi Websocket mempunyai empat langkah:

1 Wujudkan sambungan: klien menghantar permintaan HTTP ke pelayan dan naik taraf ke protokol Websocket.

2. Jabat Tangan: Pelayan mengesahkan permintaan dan mewujudkan sambungan Websocket.

3 Komunikasi: Seperti yang kita semua tahu, HTTP ialah protokol tanpa kewarganegaraan Setelah sambungan HTTP diwujudkan antara klien dan pelayan, mereka akan segera memutuskan sambungan. Jadi mengikut protokol ini, apabila pelanggan dan pelayan berkomunikasi, mereka mesti menghantar permintaan dan mesej respons. Walau bagaimanapun, dalam protokol Websocket, sebaik sahaja pelanggan membuat sambungan dengan pelayan, mereka boleh mengekalkan sambungan yang tahan lama tanpa menghantar permintaan HTTP dan mesej respons beberapa kali.

4. Tutup: Pelanggan atau pelayan memutuskan sambungan Websocket.

Dapat dilihat bahawa Websocket berbeza daripada protokol HTTP Protokol HTTP hanya menyokong penghantaran permintaan dan mesej respons, manakala protokol Websocket menyokong komunikasi dua hala masa nyata.

2. Menggunakan teknologi Websocket dalam golang

Dalam golang, anda boleh menggunakan perpustakaan pihak ketiga golang.org/x/net/websocket untuk melaksanakan komunikasi Websocket. golang.org/x/net/websocket ialah perpustakaan standard golang yang boleh melaksanakan protokol komunikasi Websocket dengan mudah.

1. Pemasangan

Dalam golang, memasang golang.org/x/net/websocket adalah sangat mudah. ​​Hanya masukkan arahan berikut dalam baris arahan:

go get golang.org/x/net/websocket

Selepas pemasangan selesai, kami boleh menggunakan API yang disediakan dalam pakej ini untuk melaksanakan fungsi komunikasi Websocket.

2. Pelayan Websocket

Berikut ialah contoh kod pelayan untuk Websocket dalam golang:

package main

import (
    "golang.org/x/net/websocket"
    "log"
    "net/http"
)

func echoHandler(ws *websocket.Conn) {
    if err := websocket.Message.Send(ws, "Hello, Client!"); err != nil {
        log.Fatal(err)
    }
}

func main() {
    http.Handle("/", websocket.Handler(echoHandler))
    err := http.ListenAndServe(":8080", nil)
    if err != nil {
        log.Fatal(err)
    }
}

Dalam contoh di atas, kami mentakrifkan fail yang dipanggil echoHandler fungsi, fungsi ini mengendalikan sambungan soket web dan menghantar mesej "Hello, Klien!" Akhir sekali, kami menggunakan fungsi ListenAndServe dalam pakej http untuk memulakan pelayan Websocket.

3. Pelanggan Websocket

Berikut ialah contoh kod klien untuk Websocket dalam golang:

package main

import (
    "fmt"
    "golang.org/x/net/websocket"
    "log"
)

func main() {
    origin := "http://localhost"
    url := "ws://localhost:8080/"
    ws, err := websocket.Dial(url, "", origin)
    if err != nil {
        log.Fatal(err)
    }
    var msg = make([]byte, 512)
    var n int
    if n, err = ws.Read(msg); err != nil {
        log.Fatal(err)
    }
    fmt.Printf("Received: %s.\n", msg[:n])
}

Dalam contoh ini, kami menggunakan fungsi WebSocket.Dial dengan Websocket Pelayan mewujudkan sambungan dan kemudian membaca mesej yang dikembalikan daripada pelayan melalui kaedah Baca WebSocket.

3. Laksanakan pemajuan data Websocket

Apabila berbilang pelanggan mewujudkan sambungan dengan pelayan Websocket, pelayan mesti mempunyai fungsi pemajuan data tertentu untuk memenuhi keperluan komunikasi masa nyata antara pelanggan . Berikut ialah kaedah untuk melaksanakan pemajuan data Websocket, menggunakan pakej golang.org/x/net/websocket:

package main

import (
    "golang.org/x/net/websocket"
    "log"
    "net/http"
)

// 保存所有的客户端连接
var clients []*websocket.Conn

func broadcast() {
    for {
        var data []byte
        // 从所有客户端接收数据
        for i := range clients {
            if err := websocket.Message.Receive(clients[i], &data); err != nil {
                // 如果有任何错误，则断开客户端连接
                clients[i].Close()
                log.Printf("Client %v disconnected\n", clients[i].RemoteAddr())
                // 将已关闭客户端从clients中删除
                clients = append(clients[:i], clients[i+1:]...)
                continue
            }
            // 向其他所有客户端广播数据
            for j := range clients {
                if err := websocket.Message.Send(clients[j], data); err != nil {
                    log.Printf("Unable to send message to %v: %v\n", clients[j].RemoteAddr(), err)
                }
            }
        }
    }
}

func main() {
    // 启动一个广播协程
    go broadcast()
    http.Handle("/", websocket.Handler(func(ws *websocket.Conn) {
        // 保存每个新连接
        clients = append(clients, ws)
        log.Printf("Client %v connected\n", ws.RemoteAddr())
    }))
    
    err := http.ListenAndServe(":8080", nil)
    if err != nil {
        log.Fatal(err)
    }
}

Dalam contoh ini, kami menggunakan siaran bernama goroutine untuk mengendalikan Fungsi pemajuan data. Apabila mana-mana pelanggan menyambung, majukan ia kepada pelanggan lain. Pada masa yang sama, apabila pelanggan memutuskan sambungan, coroutine siaran akan memadamkannya daripada pelanggan.

Ringkasnya, golang ialah bahasa pelaksanaan Websocket yang sangat baik dan anda boleh melaksanakan protokol komunikasi Websocket dengan mudah menggunakan pakej golang.org/x/net/websocket. Apabila menggunakan Websocket untuk mewujudkan sambungan komunikasi masa nyata, komunikasi segera antara pelanggan boleh dicapai dengan melaksanakan fungsi pemajuan data.

Atas ialah kandungan terperinci Cara menggunakan Websocket dalam golang untuk memajukan data. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!