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

package main 内容来自samhan

import (

"context"

内容来自samhan

"fmt"

内容来自zvvq，别采集哟

"net/http" zvvq

"sync" 内容来自samhan

"time" zvvq

) zvvq

var ( 内容来自zvvq，别采集哟

// 记录服务器连接数 内容来自zvvq

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

// 服务器列表

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

// 权重 内容来自zvvq，别采集哟

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

copyright zvvq

// 锁 zvvq

mu sync.Mutex 内容来自zvvq

)

// 初始化服务器列表

func init() {

zvvq好，好zvvq

for _, server := range servers {

connectionCounts[server] = 0 内容来自samhan666

} 内容来自samhan

}

copyright zvvq

// 处理请求 zvvq

func ServeHTTP(w http.ResponseWriter, r http.Request) { 内容来自zvvq

// 加锁

mu.Lock() 内容来自zvvq

defer mu.Unlock() zvvq好，好zvvq

// 根据权重选择服务器

zvvq

server := chooseServer(r) zvvq好，好zvvq

// 递增服务器连接数 zvvq

connectionCounts[server]++ 本文来自zvvq

// 向服务器发送请求 zvvq好，好zvvq

resp, err := http.Get(fmt.Sprintf("http://%s", server)) 内容来自samhan666

if err != nil { 内容来自zvvq

http.Error(w, err.Error(), http.StatusInternalServerError) zvvq好，好zvvq

return

内容来自samhan

}

zvvq

// 将响应写入 HTTP 响应 zvvq.cn

io.Copy(w, resp.Body) zvvq好，好zvvq

resp.Body.Close()

内容来自samhan666

} 内容来自zvvq，别采集哟

// 选择服务器

内容来自samhan

func chooseServer(r http.Request) string { zvvq

// 随机生成一个值

内容来自samhan

r := rand.Intn(sum(weights)) 内容来自zvvq

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

for i, server := range servers {

if r < weights[i] { zvvq.cn

return server copyright zvvq

} zvvq

r -= weights[i]

} zvvq好，好zvvq

// 默认返回第一个服务器 内容来自samhan666

return servers[0] 内容来自samhan666

} zvvq.cn

// 计算权重和 内容来自zvvq，别采集哟

func sum(arr []int) int { copyright zvvq

sum := 0

内容来自zvvq

for _, v := range arr {

sum += v 内容来自zvvq，别采集哟

}

return sum copyright zvvq

}

内容来自samhan

func main() {

copyright zvvq

mux := http.NewServeMux()

本文来自zvvq

mux.HandleFunc("/", ServeHTTP)

zvvq好，好zvvq

http.ListenAndServe(":8080", mux)

copyright zvvq

} zvvq.cn