Go语言实现百万级WebSocket连接架构设计及服务优化

import "github.com/gorilla/websocket"

type ConnectionManager struct {
    connections map[*websocket.Conn]bool
    register   chan *websocket.Conn
    unregister chan *websocket.Conn
}

ulimit -n 1000000

sysctl -w net.ipv4.tcp_fin_timeout=30
sysctl -w net.ipv4.tcp_tw_reuse=1

func NewServer() *Server {
    return &Server{
        ConnectionManager: &ConnectionManager{
            connections: make(map[*websocket.Conn]bool),
            register:    make(chan *websocket.Conn),
            unregister:  make(chan *websocket.Conn),
        },
    }
}

func (s *Server) HandleNewConnection(w http.ResponseWriter, r *http.Request) {
    conn, err := websocket.Upgrade(w, r, nil, 1024, 1024)
    if err != nil {
        log.Println(err)
        return
    }
    s.ConnectionManager.register <- conn
}

func (manager *ConnectionManager) run() {
    for {
        select {
        case conn := <-manager.register:
            manager.connections[conn] = true
        case conn := <-manager.unregister:
            if _, ok := manager.connections[conn]; ok {
                delete(manager.connections, conn)
                conn.Close()
            }
        }
    }
}