高效处理Go语言中数千个Keep-Alive连接的策略

// main_server.go (简化概念代码)
package main

import (
    "fmt"
    "log"
    "net/http"
    "net/rpc/jsonrpc"
    "time"
)

// WorkerClient represents a client to a worker RPC server
type WorkerClient struct {
    client *jsonrpc.Client
}

func NewWorkerClient(addr string) (*WorkerClient, error) {
    conn, err := net.Dial("tcp", addr) // Or "unix" for UNIX socket
    if err != nil {
        return nil, err
    }
    return &WorkerClient{client: jsonrpc.NewClient(conn)}, nil
}

func (wc *WorkerClient) ProcessRequest(request string, reply *string) error {
    return wc.client.Call("Worker.Handle", request, reply)
}

func main() {
    // 假设我们有一个或多个 worker_server 运行在不同的地址
    workerClient, err := NewWorkerClient("localhost:8081") // Worker RPC server address
    if err != nil {
        log.Fatalf("Failed to connect to worker: %v", err)
    }

    http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
        // 这里可以有连接管理逻辑，但核心是转发请求
        requestData := r.URL.Path
        var responseData string
        err := workerClient.ProcessRequest(requestData, &responseData)
        if err != nil {
            http.Error(w, fmt.Sprintf("Error processing request: %v", err), http.StatusInternalServerError)
            return
        }
        fmt.Fprintf(w, "Processed by worker: %s", responseData)
    })

    log.Println("Main server listening on :8080")
    log.Fatal(http.ListenAndServe(":8080", nil))
}

// worker_server.go (简化概念代码)
package main

import (
    "fmt"
    "log"
    "net"
    "net/rpc"
    "net/rpc/jsonrpc"
    "time"
)

// Worker service
type Worker struct{}

func (w *Worker) Handle(request string, reply *string) error {
    // Simulate some heavy processing
    time.Sleep(100 * time.Millisecond)
    *reply = fmt.Sprintf("Handled request '%s' at %s", request, time.Now().Format(time.RFC3339))
    return nil
}

func main() {
    worker := new(Worker)
    rpc.Register(worker)

    listener, err := net.Listen("tcp", ":8081") // Or "unix" for UNIX socket
    if err != nil {
        log.Fatalf("Failed to listen: %v", err)
    }
    defer listener.Close()

    log.Println("Worker RPC server listening on :8081")
    for {
        conn, err := listener.Accept()
        if err != nil {
            log.Printf("Error accepting connection: %v", err)
            continue
        }
        go jsonrpc.ServeConn(conn)
    }
}