gateway/balance
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

增加轮询负载策略

Browse Source
This commit is contained in:
白茶清欢 2021-04-01 20:20:04 +08:00
parent e3f273d757
commit 326bdfc7e4
6 changed files with 133 additions and 46 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

67
balance_test.go Normal file
View File

@ -0,0 +1,67 @@
// Package dispatch...
//
// Description : dispatch...
//
// Author : go_developer@163.com<张德满>
//
// Date : 2021-04-01 6:48 下午
package balance
import (
"fmt"
"testing"
"github.com/go-playground/assert/v2"
"github.com/go-developer/balance/dispatch"
"github.com/go-developer/balance/define"
)
var list = []*define.SeverNode{
{"127.0.0.1:8080", "127.0.0.1", 8080, 0, 1},
{"127.0.0.1:8081", "127.0.0.1", 8081, 0, 1},
{"127.0.0.1:8082", "127.0.0.1", 8082, 0, 1},
{"127.0.0.1:8083", "127.0.0.1", 8083, 0, 1},
}
// TestRand 测试随机选节点
//
// Author : go_developer@163.com<张德满>
//
// Date : 6:48 下午 2021/4/1
func TestRand(t *testing.T) {
result := make(map[string]int)
s := NewServer(list, dispatch.NewRand())
for i := 0; i < 10000; i++ {
node, _ := s.Get()
if _, exist := result[node]; !exist {
result[node] = 0
}
result[node]++
}
fmt.Printf("%+v", result)
}
// TestRoundRobin 轮询测试
//
// Author : go_developer@163.com<张德满>
//
// Date : 8:12 下午 2021/4/1
func TestRoundRobin(t *testing.T) {
result := make(map[string]int)
s := NewServer(list, dispatch.NewRoundRobin())
for i := 0; i < 10000; i++ {
node, _ := s.Get()
if _, exist := result[node]; !exist {
result[node] = 0
}
result[node]++
}
assert.Equal(t, 2500, result["127.0.0.1:8080"])
assert.Equal(t, 2500, result["127.0.0.1:8081"])
assert.Equal(t, 2500, result["127.0.0.1:8082"])
assert.Equal(t, 2500, result["127.0.0.1:8083"])
}

55
dispatch/round_robin.go Normal file
View File

@ -0,0 +1,55 @@
// Package dispatch ...
//
// Description : Round Robin 轮询选择服务器
//
// Author : go_developer@163.com<张德满>
//
// Date : 2021-04-01 8:04 下午
package dispatch
import (
"fmt"
"github.com/go-developer/balance/define"
"github.com/go-developer/gopkg/easylock"
)
// NewRoundRobin 轮询调度
//
// Author : go_developer@163.com<张德满>
//
// Date : 8:07 下午 2021/4/1
func NewRoundRobin() IDispatch {
return &RoundRobin{
lock: easylock.NewLock(),
nextNodeIndex: 0,
}
}
// RoundRobin 轮询选择机器
//
// Author : go_developer@163.com<张德满>
//
// Date : 8:06 下午 2021/4/1
type RoundRobin struct {
lock easylock.EasyLock
nextNodeIndex int
}
// Get ...
//
// Author : go_developer@163.com<张德满>
//
// Date : 8:05 下午 2021/4/1
func (r *RoundRobin) Get(nodeList []*define.SeverNode) (string, *define.Error) {
if len(nodeList) == 0 {
return "", define.NewError(define.ErrorTypeNodeListEmpty, "服务器可用节点为空")
}
_ = r.lock.Lock()
defer func() {
_ = r.lock.Unlock()
}()
node := fmt.Sprintf("%s:%d", nodeList[r.nextNodeIndex].Host, nodeList[r.nextNodeIndex].Port)
r.nextNodeIndex = (r.nextNodeIndex + 1) % len(nodeList)
return node, nil
}

5
go.mod
View File

@ -2,4 +2,7 @@ module github.com/go-developer/balance
go 1.16
require github.com/go-developer/gopkg v0.0.0-20210401085834-5312092ad68b
require (
github.com/go-developer/gopkg v0.0.0-20210401121730-91894a6f9659
github.com/go-playground/assert/v2 v2.0.1
)

7
go.sum
View File

@ -8,10 +8,9 @@ github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMo
github.com/fsnotify/fsnotify v1.4.9/go.mod h1:znqG4EE+3YCdAaPaxE2ZRY/06pZUdp0tY4IgpuI1SZQ=
github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI=
github.com/gin-gonic/gin v1.6.3/go.mod h1:75u5sXoLsGZoRN5Sgbi1eraJ4GU3++wFwWzhwvtwp4M=
github.com/go-developer/gopkg v0.0.0-20210326100134-08a8a8015cc5 h1:Id0MV9wihSMCncZ7OshI2ZNouijF9lrbIy+6vI6MoLM=
github.com/go-developer/gopkg v0.0.0-20210326100134-08a8a8015cc5/go.mod h1:DZcG3JkuXhqmHwZMrSJefYWNPns7nYBhA9q4ocmpG5o=
github.com/go-developer/gopkg v0.0.0-20210401085834-5312092ad68b h1:PduruFtrPw7BoxesthPBbYazz6jueMhBEFHxRA3Rd1I=
github.com/go-developer/gopkg v0.0.0-20210401085834-5312092ad68b/go.mod h1:DZcG3JkuXhqmHwZMrSJefYWNPns7nYBhA9q4ocmpG5o=
github.com/go-developer/gopkg v0.0.0-20210401121730-91894a6f9659 h1:wHI+HmfpLeG9B6EL47jLxNl4zB5dSW3ihtlWxvRrDzk=
github.com/go-developer/gopkg v0.0.0-20210401121730-91894a6f9659/go.mod h1:DZcG3JkuXhqmHwZMrSJefYWNPns7nYBhA9q4ocmpG5o=
github.com/go-playground/assert/v2 v2.0.1 h1:MsBgLAaY856+nPRTKrp3/OZK38U/wa0CcBYNjji3q3A=
github.com/go-playground/assert/v2 v2.0.1/go.mod h1:VDjEfimB/XKnb+ZQfWdccd7VUvScMdVu0Titje2rxJ4=
github.com/go-playground/locales v0.13.0/go.mod h1:taPMhCMXrRLJO55olJkUXHZBHCxTMfnGwq/HNwmWNS8=
github.com/go-playground/universal-translator v0.17.0/go.mod h1:UkSxE5sNxxRwHyU+Scu5vgOQjsIJAF8j9muTVoKLVtA=

41
rand_test.go
View File

@ -1,41 +0,0 @@
// Package dispatch...
//
// Description : dispatch...
//
// Author : go_developer@163.com<张德满>
//
// Date : 2021-04-01 6:48 下午
package balance
import (
"fmt"
"testing"
"github.com/go-developer/balance/dispatch"
"github.com/go-developer/balance/define"
)
// TestRand 测试随机选节点
//
// Author : go_developer@163.com<张德满>
//
// Date : 6:48 下午 2021/4/1
func TestRand(t *testing.T) {
result := make(map[string]int)
list := []*define.SeverNode{
{"127.0.0.1:8080", "127.0.0.1", 8080, 0, 1},
{"127.0.0.1:8081", "127.0.0.1", 8081, 0, 1},
{"127.0.0.1:8082", "127.0.0.1", 8082, 0, 1},
{"127.0.0.1:8083", "127.0.0.1", 8083, 0, 1},
}
s := NewServer(list, dispatch.NewRand())
for i := 0; i < 10000; i++ {
node, _ := s.Get()
if _, exist := result[node]; !exist {
result[node] = 0
}
result[node]++
}
fmt.Printf("%+v", result)
}

4
server.go
View File

@ -74,5 +74,9 @@ func (s *Server) Remove(nodeID string) {
//
// Date : 5:17 下午 2021/4/1
func (s *Server) Get() (string, *define.Error) {
_ = s.lock.RLock()
defer func() {
_ = s.lock.RUnlock()
}()
return s.Balance.Get(s.NodeList)
}