增加负载算法的接口约束

2021-11-22 16:52:08 +08:00
parent 30059c0341
commit ffccdeffd4
2 changed files with 32 additions and 1 deletions
--- a/balance/poll.go
+++ b/balance/poll.go
@ -8,6 +8,7 @@
 package balance

 import (
+	"net/http"
 	"sync"

 	"git.zhangdeman.cn/zhangdeman/gopkg/balance/define"
@ -48,7 +49,7 @@ type Poll struct {
 // Author : go_developer@163.com<白茶清欢>
 //
 // Date : 12:43 下午 2021/10/19
-func (p *Poll) GetServerNode() (*define.ServerNode, error) {
+func (p *Poll) GetServerNode(req *http.Request) (*define.ServerNode, error) {
 	if len(p.serverList) == 0 {
 		return nil, errors.New("server list is empty")
 	}