Ep.22 Go กับ WebSocket - การสื่อสารแบบเรียลไทม์ที่ทันสมัย!

Go กับ WebSocket - การสื่อสารแบบเรียลไทม์ที่ทันสมัย!

WebSocket คืออะไร?

WebSocket คือโปรโตคอลที่ช่วยให้การสื่อสารระหว่างเซิร์ฟเวอร์และไคลเอนต์เป็นไปแบบเรียลไทม์ โดยไม่ต้องส่งคำขอใหม่ทุกครั้งเหมือน HTTP

ข้อดีของ WebSocket :

การสื่อสารสองทาง (Full Duplex) ทำให้ทั้งเซิร์ฟเวอร์และไคลเอนต์สามารถส่งข้อมูลหากันได้ตลอดเวลา
ลดความล่าช้า (Latency) ในการอัปเดตข้อมูล
เหมาะสำหรับแอปพลิเคชันที่ต้องการข้อมูลสด เช่น เกมออนไลน์ หรือระบบแชท

การใช้งาน WebSocket ใน Go

ใน Go เรามีแพ็กเกจยอดนิยมอย่าง github.com/gorilla/websocket ที่ช่วยให้การสร้าง WebSocket Server และ Client ทำได้ง่ายขึ้น

ขั้นตอนการติดตั้ง :

go get -u github.com/gorilla/websocket

การสร้าง WebSocket Server แบบง่ายๆ

ตัวอย่างโค้ด :

ในตัวอย่างนี้ :

Upgrader ใช้แปลงการเชื่อมต่อ HTTP เป็น WebSocket

ฟังก์ชัน handleConnections ใช้สำหรับรับข้อความจากไคลเอนต์และส่งกลับไป

package main

import (
    "fmt"
    "net/http"

    "github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{
    CheckOrigin: func(r *http.Request) bool {
        return true // อนุญาตการเชื่อมต่อจากทุกที่
    },
}

func handleConnections(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil) // อัปเกรดการเชื่อมต่อเป็น WebSocket
    if err != nil {
        fmt.Println("Error upgrading connection:", err)
        return
    }
    defer conn.Close()

    for {
        messageType, msg, err := conn.ReadMessage()
        if err != nil {
            fmt.Println("Error reading message:", err)
            break
        }
        fmt.Printf("Received: %s\n", msg)

        // ส่งข้อความกลับไปยังไคลเอนต์
        if err := conn.WriteMessage(messageType, msg); err != nil {
            fmt.Println("Error writing message:", err)
            break
        }
    }
}

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

    fmt.Println("WebSocket server started at :8080/ws")
    http.ListenAndServe(":8080", nil)
}

การเชื่อมต่อจากไคลเอนต์ (HTML/JavaScript)

ตัวอย่างโค้ดฝั่งไคลเอนต์ :

ในตัวอย่างนี้ :

ใช้ WebSocket API ของ JavaScript เชื่อมต่อไปยัง WebSocket Server

เมื่อเชื่อมต่อสำเร็จ จะส่งข้อความ "Hello from client!" ไปยังเซิร์ฟเวอร์ และรอรับข้อความกลับ

<!DOCTYPE html>
<html>
<head>
    <title>WebSocket Example</title>
</head>
<body>
    <h1>WebSocket Example</h1>
    <script>
        const socket = new WebSocket("ws://localhost:8080/ws");

        socket.onopen = () => {
            console.log("Connected to server");
            socket.send("Hello from client!");
        };

        socket.onmessage = (event) => {
            console.log("Received from server:", event.data);
        };

        socket.onerror = (error) => {
            console.error("WebSocket error:", error);
        };
    </script>
</body>
</html>

การจัดการหลาย Client พร้อมกัน

WebSocket Server สามารถรองรับการเชื่อมต่อจากหลายไคลเอนต์ได้โดยการเก็บการเชื่อมต่อไว้ใน map หรือ slice

ตัวอย่างการจัดการหลาย Client :

ในตัวอย่างนี้ :

clients เก็บการเชื่อมต่อของไคลเอนต์ทั้งหมด

ฟังก์ชัน broadcastMessage ใช้ส่งข้อความไปยังทุกไคลเอนต์

var clients = make(map[*websocket.Conn]bool)

func broadcastMessage(message []byte) {
    for client := range clients {
        if err := client.WriteMessage(websocket.TextMessage, message); err != nil {
            fmt.Println("Error broadcasting message:", err)
            client.Close()
            delete(clients, client)
        }
    }
}

func handleConnections(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        fmt.Println("Error upgrading connection:", err)
        return
    }
    defer conn.Close()

    clients[conn] = true

    for {
        _, msg, err := conn.ReadMessage()
        if err != nil {
            fmt.Println("Error reading message:", err)
            delete(clients, conn)
            break
        }
        broadcastMessage(msg)
    }
}

สรุปง่ายๆ

WebSocket เหมาะสำหรับการสื่อสารแบบเรียลไทม์
ใช้แพ็กเกจ gorilla/websocket เพื่อสร้าง WebSocket Server
จัดการหลายไคลเอนต์ได้ด้วยการเก็บการเชื่อมต่อใน map และส่งข้อความกลับไปยังทุกคน