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

package main zvvq好，好zvvq

import ( zvvq

"context"

"fmt"

"net/http" zvvq

"sync"

内容来自zvvq，别采集哟

"time" 内容来自zvvq，别采集哟

) 本文来自zvvq

var ( zvvq好，好zvvq

// 记录服务器连接数

内容来自zvvq

connectionCounts = make(map[string]int) 内容来自zvvq

// 服务器列表

servers = []string{"server1", "server2", "server3"} 内容来自samhan666

// 权重 copyright zvvq

weights = []int{1, 2, 3}

// 锁

mu sync.Mutex copyright zvvq

)

内容来自samhan

// 初始化服务器列表

func init() { 内容来自zvvq，别采集哟

for _, server := range servers {

内容来自zvvq

connectionCounts[server] = 0

}

} 内容来自samhan

// 处理请求

内容来自samhan

func ServeHTTP(w http.ResponseWriter, r http.Request) {

内容来自samhan666

// 加锁 zvvq.cn

mu.Lock() zvvq好，好zvvq

defer mu.Unlock()

// 根据权重选择服务器 内容来自samhan666

server := chooseServer(r) 内容来自zvvq，别采集哟

// 递增服务器连接数

zvvq好，好zvvq

connectionCounts[server]++

// 向服务器发送请求

resp, err := http.Get(fmt.Sprintf("http://%s", server)) 本文来自zvvq

if err != nil {

http.Error(w, err.Error(), http.StatusInternalServerError)

内容来自zvvq，别采集哟

return

}

// 将响应写入 HTTP 响应 本文来自zvvq

io.Copy(w, resp.Body)

copyright zvvq

resp.Body.Close() 内容来自zvvq

}

// 选择服务器 内容来自samhan666

func chooseServer(r http.Request) string {

// 随机生成一个值 内容来自zvvq，别采集哟

r := rand.Intn(sum(weights))

// 选择服务器

zvvq好，好zvvq

for i, server := range servers { 内容来自samhan

if r < weights[i] { 内容来自samhan666

return server

}

r -= weights[i]

}

// 默认返回第一个服务器 zvvq.cn

return servers[0]

}

// 计算权重和 zvvq

func sum(arr []int) int {

sum := 0

内容来自samhan

for _, v := range arr {

zvvq好，好zvvq

sum += v

zvvq.cn

} zvvq好，好zvvq

return sum

}

本文来自zvvq

func main() {

内容来自samhan666

mux := http.NewServeMux() zvvq

mux.HandleFunc("/", ServeHTTP) zvvq.cn

http.ListenAndServe(":8080", mux)

} zvvq.cn