83 lines
1.7 KiB
Go
83 lines
1.7 KiB
Go
// Package balance ...
|
|
//
|
|
// Description : 接口定义
|
|
//
|
|
// Author : go_developer@163.com<张德满>
|
|
//
|
|
// Date : 2021-04-01 2:43 下午
|
|
package balance
|
|
|
|
import (
|
|
"git.zhangdeman.cn/gateway/balance/define"
|
|
"git.zhangdeman.cn/gateway/balance/dispatch"
|
|
"git.zhangdeman.cn/zhangdeman/easylock"
|
|
)
|
|
|
|
// NewServer ...
|
|
//
|
|
// Author : go_developer@163.com<张德满>
|
|
//
|
|
// Date : 6:51 下午 2021/4/1
|
|
func NewServer(nodeList []*define.SeverNode, d dispatch.IDispatch) *Server {
|
|
return &Server{
|
|
lock: easylock.NewLock(),
|
|
NodeList: nodeList,
|
|
Balance: d,
|
|
}
|
|
}
|
|
|
|
// Server server 的具体配置
|
|
//
|
|
// Author : go_developer@163.com<张德满>
|
|
//
|
|
// Date : 2:59 下午 2021/4/1
|
|
type Server struct {
|
|
lock easylock.EasyLock
|
|
NodeList []*define.SeverNode
|
|
Balance dispatch.IDispatch
|
|
}
|
|
|
|
// Add 添加一个Server
|
|
//
|
|
// Author : go_developer@163.com<张德满>
|
|
//
|
|
// Date : 3:00 下午 2021/4/1
|
|
func (s *Server) Add(node *define.SeverNode) {
|
|
_ = s.lock.Lock()
|
|
defer func() {
|
|
_ = s.lock.Unlock()
|
|
}()
|
|
s.NodeList = append(s.NodeList, node)
|
|
}
|
|
|
|
// Remove 移除一个server
|
|
//
|
|
// Author : go_developer@163.com<张德满>
|
|
//
|
|
// Date : 5:09 下午 2021/4/1
|
|
func (s *Server) Remove(nodeID string) {
|
|
_ = s.lock.Lock()
|
|
defer func() {
|
|
_ = s.lock.Unlock()
|
|
}()
|
|
for nodeIndex, item := range s.NodeList {
|
|
if item.ID == nodeID {
|
|
s.NodeList = append(s.NodeList[0:nodeIndex], s.NodeList[nodeIndex:]...)
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
// Get 按照指定策略获取一台机器
|
|
//
|
|
// Author : go_developer@163.com<张德满>
|
|
//
|
|
// 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)
|
|
}
|