golang框架在高并发场景中的扩展与负载均衡

package main

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

    "github.com/gin-gonic/gin"
    "github.com/gorilla/websocket"
)

var (
    mutex sync.Mutex
    count int
)

func main() {
    router := gin.Default()

    router.GET("/", func(c *gin.Context) {
        ws, err := upgrader.Upgrade(c.Writer, c.Request, nil)
        if err != nil {
            log.Fatal(err)
        }

        go handleConnection(ws)
    })

    http.ListenAndServe(":8080", router)
}

var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
}

func handleConnection(ws *websocket.Conn) {
    defer ws.Close()

    for {
        _, msg, err := ws.ReadMessage()
        if err != nil {
            if websocket.IsUnexpectedCloseError(err) {
                log.Printf("unexpected websocket closed: %v", err)
            }
            break
        }

        mutex.Lock()
        count++
        fmt.Printf("Received message: %s, count: %d\n", msg, count)
        mutex.Unlock()
    }
}