From 6b5e01aacdc44e5a3dadfd0c75a0a941663a59fd Mon Sep 17 00:00:00 2001 From: googs1025 Date: Tue, 12 Mar 2024 11:58:58 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E6=96=B0=E5=A2=9E=E8=B4=9F=E8=BD=BD?= =?UTF-8?q?=E5=9D=87=E8=A1=A1=E6=A8=A1=E5=BC=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.md | 3 +- go.mod | 26 +++- go.sum | 71 ++++++++- .../loadbalance/interface.go | 6 + other/loadbalance-pattern/main.go | 72 +++++++++ other/loadbalance-pattern/random/random.go | 143 ++++++++++++++++++ .../loadbalance-pattern/round_robin/round.go | 142 +++++++++++++++++ 7 files changed, 460 insertions(+), 3 deletions(-) create mode 100644 other/loadbalance-pattern/loadbalance/interface.go create mode 100644 other/loadbalance-pattern/main.go create mode 100644 other/loadbalance-pattern/random/random.go create mode 100644 other/loadbalance-pattern/round_robin/round.go diff --git a/README.md b/README.md index aa9f09e..375e09f 100644 --- a/README.md +++ b/README.md @@ -26,4 +26,5 @@ ### Other 模式： 1. **创建资源对象模式**：模拟创建资源对象，并可扩展多个 hook 方法。[代码示例](https://github.com/StudyPlace-io/Golang-Design-Pattern-Demo/tree/main/other/create-resource-pattern) -2. **goroutine 模式**：并发简易示例。[代码示例](https://github.com/StudyPlace-io/Golang-Design-Pattern-Demo/tree/main/other/goroutine-pattern) \ No newline at end of file +2. **goroutine 模式**：并发简易示例。[代码示例](https://github.com/StudyPlace-io/Golang-Design-Pattern-Demo/tree/main/other/goroutine-pattern) +3. **反向代理模式**: 模拟http反向代理，实现轮询与随机模式。[代码示例](https://github.com/StudyPlace-io/Golang-Design-Pattern-Demo/tree/main/other/loadbalance-pattern) \ No newline at end of file diff --git a/go.mod b/go.mod index 51e2eeb..b4be7ae 100644 --- a/go.mod +++ b/go.mod @@ -3,14 +3,38 @@ module github.com/practice/Design-Patterns-practice go 1.18 require ( + github.com/gin-gonic/gin v1.9.1 gopkg.in/ini.v1 v1.67.0 gorm.io/driver/mysql v1.4.5 gorm.io/gorm v1.24.3 ) require ( + github.com/bytedance/sonic v1.9.1 // indirect + github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311 // indirect + github.com/gabriel-vasile/mimetype v1.4.2 // indirect + github.com/gin-contrib/sse v0.1.0 // indirect + github.com/go-playground/locales v0.14.1 // indirect + github.com/go-playground/universal-translator v0.18.1 // indirect + github.com/go-playground/validator/v10 v10.14.0 // indirect github.com/go-sql-driver/mysql v1.7.0 // indirect + github.com/goccy/go-json v0.10.2 // indirect github.com/jinzhu/inflection v1.0.0 // indirect github.com/jinzhu/now v1.1.5 // indirect - github.com/stretchr/testify v1.8.1 // indirect + github.com/json-iterator/go v1.1.12 // indirect + github.com/klauspost/cpuid/v2 v2.2.4 // indirect + github.com/leodido/go-urn v1.2.4 // indirect + github.com/mattn/go-isatty v0.0.19 // indirect + github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect + github.com/modern-go/reflect2 v1.0.2 // indirect + github.com/pelletier/go-toml/v2 v2.0.8 // indirect + github.com/twitchyliquid64/golang-asm v0.15.1 // indirect + github.com/ugorji/go/codec v1.2.11 // indirect + golang.org/x/arch v0.3.0 // indirect + golang.org/x/crypto v0.9.0 // indirect + golang.org/x/net v0.10.0 // indirect + golang.org/x/sys v0.8.0 // indirect + golang.org/x/text v0.9.0 // indirect + google.golang.org/protobuf v1.30.0 // indirect + gopkg.in/yaml.v3 v3.0.1 // indirect ) diff --git a/go.sum b/go.sum index 9e97578..d2bdda1 100644 --- a/go.sum +++ b/go.sum @@ -1,22 +1,90 @@ +github.com/bytedance/sonic v1.5.0/go.mod h1:ED5hyg4y6t3/9Ku1R6dU/4KyJ48DZ4jPhfY1O2AihPM= +github.com/bytedance/sonic v1.9.1 h1:6iJ6NqdoxCDr6mbY8h18oSO+cShGSMRGCEo7F2h0x8s= +github.com/bytedance/sonic v1.9.1/go.mod h1:i736AoUSYt75HyZLoJW9ERYxcy6eaN6h4BZXU064P/U= +github.com/chenzhuoyu/base64x v0.0.0-20211019084208-fb5309c8db06/go.mod h1:DH46F32mSOjUmXrMHnKwZdA8wcEefY7UVqBKYGjpdQY= +github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311 h1:qSGYFH7+jGhDF8vLC+iwCD4WpbV1EBDSzWkJODFLams= +github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311/go.mod h1:b583jCggY9gE99b6G5LEC39OIiVsWj+R97kbl5odCEk= github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/gabriel-vasile/mimetype v1.4.2 h1:w5qFW6JKBz9Y393Y4q372O9A7cUSequkh1Q7OhCmWKU= +github.com/gabriel-vasile/mimetype v1.4.2/go.mod h1:zApsH/mKG4w07erKIaJPFiX0Tsq9BFQgN3qGY5GnNgA= +github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE= +github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI= +github.com/gin-gonic/gin v1.9.1 h1:4idEAncQnU5cB7BeOkPtxjfCSye0AAm1R0RVIqJ+Jmg= +github.com/gin-gonic/gin v1.9.1/go.mod h1:hPrL7YrpYKXt5YId3A/Tnip5kqbEAP+KLuI3SUcPTeU= +github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s= +github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA= +github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY= +github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY= +github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY= +github.com/go-playground/validator/v10 v10.14.0 h1:vgvQWe3XCz3gIeFDm/HnTIbj6UGmg/+t63MyGU2n5js= +github.com/go-playground/validator/v10 v10.14.0/go.mod h1:9iXMNT7sEkjXb0I+enO7QXmzG6QCsPWY4zveKFVRSyU= github.com/go-sql-driver/mysql v1.7.0 h1:ueSltNNllEqE3qcWBTD0iQd3IpL/6U+mJxLkazJ7YPc= github.com/go-sql-driver/mysql v1.7.0/go.mod h1:OXbVy3sEdcQ2Doequ6Z5BW6fXNQTmx+9S1MCJN5yJMI= +github.com/goccy/go-json v0.10.2 h1:CrxCmQqYDkv1z7lO7Wbh2HN93uovUHgrECaO5ZrCXAU= +github.com/goccy/go-json v0.10.2/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I= +github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk= +github.com/google/go-cmp v0.5.5 h1:Khx7svrCpmxxtHBq5j2mp/xVjsi8hQMfNLvJFAlrGgU= +github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= +github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg= github.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD/E= github.com/jinzhu/inflection v1.0.0/go.mod h1:h+uFLlag+Qp1Va5pdKtLDYj+kHp5pxUVkryuEj+Srlc= github.com/jinzhu/now v1.1.4/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8= github.com/jinzhu/now v1.1.5 h1:/o9tlHleP7gOFmsnYNz3RGnqzefHA47wQpKrrdTIwXQ= github.com/jinzhu/now v1.1.5/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8= +github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM= +github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo= +github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg= +github.com/klauspost/cpuid/v2 v2.2.4 h1:acbojRNwl3o09bUq+yDCtZFc1aiwaAAxtcn8YkZXnvk= +github.com/klauspost/cpuid/v2 v2.2.4/go.mod h1:RVVoqg1df56z8g3pUjL/3lE5UfnlrJX8tyFgg4nqhuY= +github.com/leodido/go-urn v1.2.4 h1:XlAE/cm/ms7TE/VMVoduSpNBoyc2dOxHs5MZSwAN63Q= +github.com/leodido/go-urn v1.2.4/go.mod h1:7ZrI8mTSeBSHl/UaRyKQW1qZeMgak41ANeCNaVckg+4= +github.com/mattn/go-isatty v0.0.19 h1:JITubQf0MOLdlGRuRq+jtsDlekdYPia9ZFsB8h/APPA= +github.com/mattn/go-isatty v0.0.19/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y= +github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= +github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg= +github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= +github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M= +github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk= +github.com/pelletier/go-toml/v2 v2.0.8 h1:0ctb6s9mE31h0/lhu+J6OPmVeDxJn+kYnJc2jZR9tGQ= +github.com/pelletier/go-toml/v2 v2.0.8/go.mod h1:vuYfssBdrU2XDZ9bYydBu6t+6a6PYNcZljzZR9VXg+4= github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw= github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo= +github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI= +github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU= -github.com/stretchr/testify v1.8.1 h1:w7B6lhMri9wdJUVmEZPGGhZzrYTPvgJArz7wNPgYKsk= github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4= +github.com/stretchr/testify v1.8.2/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4= +github.com/stretchr/testify v1.8.3 h1:RP3t2pwF7cMEbC1dqtB6poj3niw/9gnV4Cjg5oW5gtY= +github.com/stretchr/testify v1.8.3/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo= +github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI= +github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08= +github.com/ugorji/go/codec v1.2.11 h1:BMaWp1Bb6fHwEtbplGBGJ498wD+LKlNSl25MjdZY4dU= +github.com/ugorji/go/codec v1.2.11/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZgYf6w6lg= +golang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8= +golang.org/x/arch v0.3.0 h1:02VY4/ZcO/gBOH6PUaoiptASxtXU10jazRCP865E97k= +golang.org/x/arch v0.3.0/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8= +golang.org/x/crypto v0.9.0 h1:LF6fAI+IutBocDJ2OT0Q1g8plpYljMZ4+lty+dsqw3g= +golang.org/x/crypto v0.9.0/go.mod h1:yrmDGqONDYtNj3tH8X9dzUun2m2lzPa9ngI6/RUPGR0= +golang.org/x/net v0.10.0 h1:X2//UzNDwYmtCLn7To6G58Wr6f5ahEAQgKNzv9Y951M= +golang.org/x/net v0.10.0/go.mod h1:0qNGK6F8kojg2nk9dLZ2mShWaEBan6FAoqfSigmmuDg= +golang.org/x/sys v0.0.0-20220704084225-05e143d24a9e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.8.0 h1:EBmGv8NaZBZTWvrbjNoL6HVt+IVy3QDQpJs7VRIw3tU= +golang.org/x/sys v0.8.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/text v0.9.0 h1:2sjJmO8cDvYveuX97RDLsxlyUxLl+GHoLxBiRdHllBE= +golang.org/x/text v0.9.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8= +golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4= +golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw= +google.golang.org/protobuf v1.30.0 h1:kPPoIgf3TsEvrm0PFe15JQ+570QVxYzEvvHqChK+cng= +google.golang.org/protobuf v1.30.0/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM= gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= gopkg.in/ini.v1 v1.67.0 h1:Dgnx+6+nfE+IfzjUEISNeydPJh9AXNNsWbGP9KzCsOA= gopkg.in/ini.v1 v1.67.0/go.mod h1:pNLf8WUiyNEtQjuu5G5vTm06TEv9tsIgeAvK8hOrP4k= @@ -28,3 +96,4 @@ gorm.io/driver/mysql v1.4.5/go.mod h1:SxzItlnT1cb6e1e4ZRpgJN2VYtcqJgqnHxWr4wsP8o gorm.io/gorm v1.23.8/go.mod h1:l2lP/RyAtc1ynaTjFksBde/O8v9oOGIApu2/xRitmZk= gorm.io/gorm v1.24.3 h1:WL2ifUmzR/SLp85CSURAfybcHnGZ+yLSGSxgYXlFBHg= gorm.io/gorm v1.24.3/go.mod h1:DVrVomtaYTbqs7gB/x2uVvqnXzv0nqjB396B8cG4dBA= +rsc.io/pdf v0.1.1/go.mod h1:n8OzWcQ6Sp37PL01nO98y4iUCRdTGarVfzxY20ICaU4= diff --git a/other/loadbalance-pattern/loadbalance/interface.go b/other/loadbalance-pattern/loadbalance/interface.go new file mode 100644 index 0000000..ba91452 --- /dev/null +++ b/other/loadbalance-pattern/loadbalance/interface.go @@ -0,0 +1,6 @@ +package loadbalance + +type LoadBalance interface { + SetBackendServer(backend []string) + Run(addr string) +} diff --git a/other/loadbalance-pattern/main.go b/other/loadbalance-pattern/main.go new file mode 100644 index 0000000..467695e --- /dev/null +++ b/other/loadbalance-pattern/main.go @@ -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") +} diff --git a/other/loadbalance-pattern/random/random.go b/other/loadbalance-pattern/random/random.go new file mode 100644 index 0000000..d271161 --- /dev/null +++ b/other/loadbalance-pattern/random/random.go @@ -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 +} diff --git a/other/loadbalance-pattern/round_robin/round.go b/other/loadbalance-pattern/round_robin/round.go new file mode 100644 index 0000000..efbc646 --- /dev/null +++ b/other/loadbalance-pattern/round_robin/round.go @@ -0,0 +1,142 @@ +package round + +import ( + "github.com/practice/Design-Patterns-practice/other/loadbalance-pattern/loadbalance" + "log" + "net/http" + "net/http/httputil" + "net/url" + "sync" + "time" +) + +type roundLoadBalance struct { + healthyCheck time.Duration + proxy *httputil.ReverseProxy + backends []*Server + healthyBackends []*Server + mu sync.Mutex +} + +const ( + defaultHealthyCheck = time.Second * 10 +) + +type Server struct { + URL *url.URL + // Healthy 标示健康状态 + Healthy bool +} + +// NewLoadBalancerForRoundRobinMode 使用轮询模式 +func NewLoadBalancerForRoundRobinMode() loadbalance.LoadBalance { + // 创建负载均衡器实例 + lb := &roundLoadBalance{} + + // 创建反向代理器, 也可以用 + // ServeHTTP(w http.ResponseWriter, r *http.Request) 方法实现 + lb.proxy = &httputil.ReverseProxy{ + Director: lb.director, + } + + return lb +} + +func (lb *roundLoadBalance) 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, + }) + } + lb.healthyBackends = lb.backends +} + +func (lb *roundLoadBalance) 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 *roundLoadBalance) 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 *roundLoadBalance) 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 *roundLoadBalance) randomHealthyServer() *Server { + lb.mu.Lock() + defer lb.mu.Unlock() + + healthyServers := make([]*Server, 0) + + // 选择健康的服务器 + for _, server := range lb.healthyBackends { + if server.Healthy { + healthyServers = append(healthyServers, server) + } + } + + if len(healthyServers) == 0 { + return nil + } + + server := healthyServers[0] + lb.healthyBackends = append(lb.healthyBackends[1:], server) + + return server +} + +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 +}