EP.102 การใช้ Goroutines และ Worker Pool สำหรับการจัดการ Concurrent Connections

ในระบบ WebSocket ที่ต้องรองรับผู้ใช้งานจำนวนมาก การจัดการ Goroutines และ Worker Pool อย่างมีประสิทธิภาพ เป็นกุญแจสำคัญที่จะทำให้ระบบของคุณ เร็ว, เสถียร, และไม่พังเมื่อโดนโหลดหนัก

บทความนี้จะพาคุณไปเรียนรู้แนวคิด พร้อมตัวอย่างโค้ด Go ที่ใช้งานได้จริง

🌀 1. ใช้ Goroutines รับมือกับการเชื่อมต่อจำนวนมาก

เมื่อมีการเชื่อมต่อจากผู้ใช้งานเข้ามา แต่ละการเชื่อมต่อควรถูกแยกไปจัดการใน Goroutine ของตัวเอง:

ป้องกันการ block ของ main thread
เพิ่มความสามารถในการรองรับ concurrent connections

✅ ตัวอย่าง: แยก Goroutine สำหรับแต่ละ WebSocket connection

package main

import (
    "fmt"
    "net/http"
    "github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{
    CheckOrigin: func(r *http.Request) bool { return true },
}

func handleConnection(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        http.Error(w, "Upgrade error", http.StatusInternalServerError)
        return
    }
    defer conn.Close()

    go func(c *websocket.Conn) {
        for {
            _, msg, err := c.ReadMessage()
            if err != nil {
                fmt.Println("Connection closed:", err)
                break
            }
            fmt.Println("Received:", string(msg))
        }
    }(conn)
}

func main() {
    http.HandleFunc("/ws", handleConnection)
    fmt.Println("WebSocket Server running on :8080")
    http.ListenAndServe(":8080", nil)
}

🧱 2. ใช้ Worker Pool เพื่อควบคุมจำนวน Goroutines

Goroutine แม้จะเบา แต่ถ้าไม่ควบคุมจำนวนก็สามารถ กิน Memory จนระบบล่มได้

การใช้ Worker Pool ช่วยให้:

ประหยัดทรัพยากร
ควบคุมจำนวน Goroutines ได้ชัดเจน
เหมาะกับการจัดการ Task จำนวนมากที่มาทีละ batch

✅ ตัวอย่าง: Worker Pool รับข้อความจาก jobs queue

type Job struct {
    Message string
}

var jobs = make(chan Job, 100)
var results = make(chan string, 100)

func worker(id int, jobs <-chan Job, results chan<- string) {
    for j := range jobs {
        fmt.Printf("Worker %d processing: %s\n", id, j.Message)
        results <- fmt.Sprintf("Done: %s", j.Message)
    }
}

func main() {
    // Start 5 workers
    for w := 1; w <= 5; w++ {
        go worker(w, jobs, results)
    }

    // Push 20 jobs
    for i := 0; i < 20; i++ {
        jobs <- Job{Message: fmt.Sprintf("Message %d", i)}
    }
    close(jobs)

    // Collect results
    for i := 0; i < 20; i++ {
        fmt.Println(<-results)
    }
}