Bagaimana Penulisan Semula Go Menjadikan Proksi WebSocket Kami 100x Lebih Cekap

// ────────────────────────────
// REKA BENTUK PERTAMA (naif, mutex global)
// Slice tunggal + bendera, satu mutex kasar.
// ────────────────────────────

type Conn struct {
    id      string
    ws      *websocket.Conn
    inUse   bool
    healthy bool
}

type Pool struct {
    mu       sync.Mutex
    conns    []*Conn
    maxSize  int
    dialURL  string
}

package pool

import (
	"context"
	"errors"
	"sync"
	"sync/atomic"
	"time"

	"github.com/google/uuid"
	"github.com/gorilla/websocket"
)

// ────────────────────────────
// Conn (satu sambungan WebSocket backend boleh guna semula)
// ────────────────────────────

type Conn struct {
	id       string
	ws       *websocket.Conn
	healthy  atomic.Bool   // bendera status kesihatan tanpa kunci
	lastPing atomic.Int64  // cap masa atomik untuk degupan jantung terakhir
}

// StartHealthLoop berjalan secara bebas setiap sambungan
func (c *Conn) StartHealthLoop(ctx context.Context, interval time.Duration) {
	t := time.NewTicker(interval)
	defer t.Stop()
	for {
		select {
		case <-ctx.Done():
			return
		case <-t.C:
			deadline := time.Now().Add(interval / 2)
			if err := c.ws.WriteControl(websocket.PingMessage, []byte("ping"), deadline); err != nil {
				c.healthy.Store(false)
				_ = c.Close()
				return
			}
			c.lastPing.Store(time.Now().UnixNano())
			c.healthy.Store(true)
		}
	}
}

// ────────────────────────────
// ConnPool (laluan pantas sahaja - tiada dail, tiada I/O menyekat)
// ────────────────────────────

type ConnPool struct {
	available chan *Conn   // saluran buffer untuk conns siap (tanpa kunci)
	inUse     sync.Map     // peta serentak untuk menjejaki conns aktif
	statsIn   atomic.Int64 // pembilang atomik (tidak perlu kunci)
	statsOut  atomic.Int64
}

// Acquire tidak menyekat
func (p *ConnPool) Acquire(ctx context.Context) (*Conn, error) {
	select {
	case c := <-p.available:
		p.inUse.Store(c.id, c)
		p.statsIn.Add(1)
		return c, nil
	case <-ctx.Done():
		return nil, ctx.Err()
	default:
		return nil, errors.New("tiada sambungan tersedia")
	}
}

// Release tidak menyekat
func (p *ConnPool) Release(c *Conn) {
	p.inUse.Delete(c.id)
	p.statsOut.Add(1)

	if c.IsAlive() {
		select {
		case p.available <- c:
			// dikembalikan ke kumpulan dengan selamat
		default:
			// kumpulan penuh -> buang conn basi dengan selamat
			_ = c.Close()
		}
	} else {
		// tidak sihat -> tutup segera
		_ = c.Close()
	}
}