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)

}