-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
7 changed files
with
460 additions
and
3 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,6 @@ | ||
| package loadbalance | ||
|
|
||
| type LoadBalance interface { | ||
| SetBackendServer(backend []string) | ||
| Run(addr string) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,72 @@ | ||
| package main | ||
|
|
||
| import ( | ||
| "github.com/gin-gonic/gin" | ||
| round "github.com/practice/Design-Patterns-practice/other/loadbalance-pattern/round_robin" | ||
| ) | ||
|
|
||
| func main() { | ||
|
|
||
| go server1() | ||
| go server2() | ||
| go server3() | ||
|
|
||
| backends := []string{ | ||
| "http://localhost:8081", | ||
| "http://localhost:8082", | ||
| "http://localhost:8083", | ||
| } | ||
|
|
||
| //rl := random.NewLoadBalancerForRandomMode() | ||
| //rl.SetBackendServer(backends) | ||
|
|
||
| rl := round.NewLoadBalancerForRoundRobinMode() | ||
| rl.SetBackendServer(backends) | ||
| rl.Run(":8080") | ||
|
|
||
| } | ||
|
|
||
| func server1() { | ||
|
|
||
| router := gin.Default() | ||
|
|
||
| // 定义路由和处理函数 | ||
| router.GET("/", func(c *gin.Context) { | ||
| c.JSON(200, gin.H{ | ||
| "message": "Hello, World!, server1", | ||
| }) | ||
| }) | ||
|
|
||
| // 启动服务器 | ||
| router.Run(":8081") | ||
| } | ||
|
|
||
| func server2() { | ||
| // 创建Gin引擎 | ||
| router := gin.Default() | ||
|
|
||
| // 定义路由和处理函数 | ||
| router.GET("/", func(c *gin.Context) { | ||
| c.JSON(200, gin.H{ | ||
| "message": "Hello, World!, server2", | ||
| }) | ||
| }) | ||
|
|
||
| // 启动服务器 | ||
| router.Run(":8082") | ||
| } | ||
|
|
||
| func server3() { | ||
| // 创建Gin引擎 | ||
| router := gin.Default() | ||
|
|
||
| // 定义路由和处理函数 | ||
| router.GET("/", func(c *gin.Context) { | ||
| c.JSON(200, gin.H{ | ||
| "message": "Hello, World!, server3", | ||
| }) | ||
| }) | ||
|
|
||
| // 启动服务器 | ||
| router.Run(":8083") | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,143 @@ | ||
| package random | ||
|
|
||
| import ( | ||
| "github.com/practice/Design-Patterns-practice/other/loadbalance-pattern/loadbalance" | ||
| "log" | ||
| "math/rand" | ||
| "net/http" | ||
| "net/http/httputil" | ||
| "net/url" | ||
| "sync" | ||
| "time" | ||
| ) | ||
|
|
||
| type randomLoadBalance struct { | ||
| healthyCheck time.Duration | ||
| proxy *httputil.ReverseProxy | ||
| backends []*Server | ||
| mu sync.Mutex | ||
| } | ||
|
|
||
| const ( | ||
| defaultHealthyCheck = time.Second * 10 | ||
| ) | ||
|
|
||
| type Server struct { | ||
| URL *url.URL | ||
| // Healthy 标示健康状态 | ||
| Healthy bool | ||
| } | ||
|
|
||
| // NewLoadBalancerForRandomMode 使用随机模式 | ||
| func NewLoadBalancerForRandomMode() loadbalance.LoadBalance { | ||
| // 创建负载均衡器实例 | ||
| lb := &randomLoadBalance{} | ||
|
|
||
| // 创建反向代理器, 也可以用 | ||
| // ServeHTTP(w http.ResponseWriter, r *http.Request) 方法实现 | ||
| lb.proxy = &httputil.ReverseProxy{ | ||
| Director: lb.director, | ||
| } | ||
|
|
||
| return lb | ||
| } | ||
|
|
||
| func (lb *randomLoadBalance) SetBackendServer(backends []string) { | ||
| // 启动负载均衡器监听的端口 | ||
| for _, u := range backends { | ||
| serverURL, err := url.Parse(u) | ||
| if err != nil { | ||
| log.Fatal(err) | ||
| } | ||
| lb.backends = append(lb.backends, &Server{ | ||
| URL: serverURL, | ||
| Healthy: true, | ||
| }) | ||
| } | ||
| } | ||
|
|
||
| func (lb *randomLoadBalance) Run(addr string) { | ||
|
|
||
| if len(lb.backends) == 0 { | ||
| panic("backend server can't be nil, use SetBackendServer method to add backend server") | ||
| } | ||
|
|
||
| go lb.healthCheck() | ||
| // 启动负载均衡器监听的端口 | ||
| err := http.ListenAndServe(addr, lb.proxy) | ||
| if err != nil { | ||
| log.Fatal(err) | ||
| } | ||
| } | ||
|
|
||
| // director 反向代理方法，选择目标服务器 | ||
| func (lb *randomLoadBalance) director(req *http.Request) { | ||
| targetServer := lb.randomHealthyServer() | ||
| // 设置目标服务器的主机 | ||
| req.URL.Scheme = targetServer.URL.Scheme | ||
| req.URL.Host = targetServer.URL.Host | ||
| req.URL.Path = singleJoiningSlash(targetServer.URL.Path, req.URL.Path) | ||
| req.Host = targetServer.URL.Host | ||
| } | ||
|
|
||
| func (lb *randomLoadBalance) healthCheck() { | ||
|
|
||
| if lb.healthyCheck == time.Duration(0) { | ||
| lb.healthyCheck = defaultHealthyCheck | ||
| } | ||
|
|
||
| ticker1 := time.NewTicker(lb.healthyCheck) | ||
|
|
||
| for { | ||
| select { | ||
| case <-ticker1.C: | ||
| lb.mu.Lock() | ||
| for _, server := range lb.backends { | ||
| // 发送健康检查请求, 修改状态 | ||
| resp, err := http.Get(server.URL.String()) | ||
| if err != nil || resp.StatusCode != http.StatusOK { | ||
| server.Healthy = false | ||
| } else { | ||
| server.Healthy = true | ||
| } | ||
| } | ||
| lb.mu.Unlock() | ||
| } | ||
| } | ||
| } | ||
|
|
||
| func (lb *randomLoadBalance) randomHealthyServer() *Server { | ||
| lb.mu.Lock() | ||
| defer lb.mu.Unlock() | ||
|
|
||
| healthyServers := make([]*Server, 0) | ||
|
|
||
| // 选择健康的服务器 | ||
| for _, server := range lb.backends { | ||
| if server.Healthy { | ||
| healthyServers = append(healthyServers, server) | ||
| } | ||
| } | ||
|
|
||
| if len(healthyServers) == 0 { | ||
| return nil | ||
| } | ||
|
|
||
| // 使用随机数选择一个健康的服务器 | ||
| rand.Seed(time.Now().UnixNano()) | ||
| index := rand.Intn(len(healthyServers)) | ||
|
|
||
| return healthyServers[index] | ||
| } | ||
|
|
||
| func singleJoiningSlash(a, b string) string { | ||
| aslash := len(a) > 0 && a[len(a)-1] == '/' | ||
| bslash := len(b) > 0 && b[0] == '/' | ||
| switch { | ||
| case aslash && bslash: | ||
| return a + b[1:] | ||
| case !aslash && !bslash: | ||
| return a + "/" + b | ||
| } | ||
| return a + b | ||
| } |
Oops, something went wrong.