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

update code

Browse Source
This commit is contained in:
白茶清欢 2025-05-23 22:22:16 +08:00
parent 640e022a40
commit e1597328d2
3 changed files with 15 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
balance_test.go
View File

@ -13,7 +13,7 @@ import (
"github.com/go-playground/assert/v2" "github.com/go-playground/assert/v2"
"git.zhangdeman.cn/gateway/balance/dispatch" "git.zhangdeman.cn/gateway/balance/implement"
"git.zhangdeman.cn/gateway/balance/define" "git.zhangdeman.cn/gateway/balance/define"
) )
@ -33,7 +33,7 @@ var list = []*define.SeverNode{
func TestRand(t *testing.T) { func TestRand(t *testing.T) {
result := make(map[string]int) result := make(map[string]int)
s := NewServer(list, dispatch.NewRand()) s := NewServer(list, implement.NewRand())
for i := 0; i < 10000; i++ { for i := 0; i < 10000; i++ {
node, _ := s.Get() node, _ := s.Get()
if _, exist := result[node]; !exist { if _, exist := result[node]; !exist {
@ -52,7 +52,7 @@ func TestRand(t *testing.T) {
func TestRoundRobin(t *testing.T) { func TestRoundRobin(t *testing.T) {
result := make(map[string]int) result := make(map[string]int)
s := NewServer(list, dispatch.NewRoundRobin()) s := NewServer(list, implement.NewRoundRobin())
for i := 0; i < 10000; i++ { for i := 0; i < 10000; i++ {
node, _ := s.Get() node, _ := s.Get()
if _, exist := result[node]; !exist { if _, exist := result[node]; !exist {

10
dispatch/rand.go → implement/rand.go
View File

@ -5,10 +5,12 @@
// Author : go_developer@163.com<张德满> // Author : go_developer@163.com<张德满>
// //
// Date : 2021-04-01 5:58 下午 // Date : 2021-04-01 5:58 下午
package dispatch package implement
import ( import (
"fmt" "fmt"
"git.zhangdeman.cn/gateway/balance/abstract"
"git.zhangdeman.cn/zhangdeman/exception"
"math/rand" "math/rand"
"git.zhangdeman.cn/gateway/balance/define" "git.zhangdeman.cn/gateway/balance/define"
@ -19,7 +21,7 @@ import (
// Author : go_developer@163.com<张德满> // Author : go_developer@163.com<张德满>
// //
// Date : 6:51 下午 2021/4/1 // Date : 6:51 下午 2021/4/1
func NewRand() IDispatch { func NewRand() abstract.IBalance {
return &Rand{} return &Rand{}
} }
@ -36,9 +38,9 @@ type Rand struct {
// Author : go_developer@163.com<张德满> // Author : go_developer@163.com<张德满>
// //
// Date : 6:01 下午 2021/4/1 // Date : 6:01 下午 2021/4/1
func (r Rand) Get(nodeList []*define.SeverNode) (string, *define.Error) { func (r Rand) Get(nodeList []*define.SeverNode) (string, exception.IException) {
if len(nodeList) == 0 { if len(nodeList) == 0 {
return "", define.NewError(define.ErrorTypeNodeListEmpty, "服务器可用节点为空") return "", exception.New(define.ErrorTypeNodeListEmpty, nil, "服务器可用节点为空")
} }
node := nodeList[rand.Intn(len(nodeList))] node := nodeList[rand.Intn(len(nodeList))]
return fmt.Sprintf("%s:%d", node.Host, node.Port), nil return fmt.Sprintf("%s:%d", node.Host, node.Port), nil

10
dispatch/round_robin.go → implement/round_robin.go
View File

@ -5,10 +5,12 @@
// Author : go_developer@163.com<张德满> // Author : go_developer@163.com<张德满>
// //
// Date : 2021-04-01 8:04 下午 // Date : 2021-04-01 8:04 下午
package dispatch package implement
import ( import (
"fmt" "fmt"
"git.zhangdeman.cn/gateway/balance/abstract"
"git.zhangdeman.cn/zhangdeman/exception"
"git.zhangdeman.cn/gateway/balance/define" "git.zhangdeman.cn/gateway/balance/define"
"git.zhangdeman.cn/zhangdeman/easylock" "git.zhangdeman.cn/zhangdeman/easylock"
@ -19,7 +21,7 @@ import (
// Author : go_developer@163.com<张德满> // Author : go_developer@163.com<张德满>
// //
// Date : 8:07 下午 2021/4/1 // Date : 8:07 下午 2021/4/1
func NewRoundRobin() IDispatch { func NewRoundRobin() abstract.IBalance {
return &RoundRobin{ return &RoundRobin{
lock: easylock.NewLock(), lock: easylock.NewLock(),
nextNodeIndex: 0, nextNodeIndex: 0,
@ -41,9 +43,9 @@ type RoundRobin struct {
// Author : go_developer@163.com<张德满> // Author : go_developer@163.com<张德满>
// //
// Date : 8:05 下午 2021/4/1 // Date : 8:05 下午 2021/4/1
func (r *RoundRobin) Get(nodeList []*define.SeverNode) (string, *define.Error) { func (r *RoundRobin) Get(nodeList []*define.SeverNode) (string, exception.IException) {
if len(nodeList) == 0 { if len(nodeList) == 0 {
return "", define.NewError(define.ErrorTypeNodeListEmpty, "服务器可用节点为空") return "", exception.New(define.ErrorTypeNodeListEmpty, nil, "服务器可用节点为空")
} }
_ = r.lock.Lock() _ = r.lock.Lock()
defer func() { defer func() {