golang框架中如何进行负载均衡之负载均衡与性能优化

package main

import (
    "context"
    "fmt"
    "net/http"
    "sync"
    "time"
)

var (
    // 记录服务器连接数
    connectionCounts = make(map[string]int)
    // 服务器列表
    servers = []string{"server1", "server2", "server3"}
    // 权重
    weights = []int{1, 2, 3}
    // 锁
    mu sync.Mutex
)

// 初始化服务器列表
func init() {
    for _, server := range servers {
        connectionCounts[server] = 0
    }
}

// 处理请求
func ServeHTTP(w http.ResponseWriter, r *http.Request) {
    // 加锁
    mu.Lock()
    defer mu.Unlock()

    // 根据权重选择服务器
    server := chooseServer(r)

    // 递增服务器连接数
    connectionCounts[server]++

    // 向服务器发送请求
    resp, err := http.Get(fmt.Sprintf("http://%s", server))
    if err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
        return
    }

    // 将响应写入 HTTP 响应
    io.Copy(w, resp.Body)
    resp.Body.Close()
}

// 选择服务器
func chooseServer(r *http.Request) string {
    // 随机生成一个值
    r := rand.Intn(sum(weights))

    // 选择服务器
    for i, server := range servers {
        if r < weights[i] {
            return server
        }
        r -= weights[i]
    }

    // 默认返回第一个服务器
    return servers[0]
}

// 计算权重和
func sum(arr []int) int {
    sum := 0
    for _, v := range arr {
        sum += v
    }
    return sum
}

func main() {
    mux := http.NewServeMux()
    mux.HandleFunc("/", ServeHTTP)
    http.ListenAndServe(":8080", mux)
}