91 lines
1.9 KiB
Go
91 lines
1.9 KiB
Go
package discover
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"apigo.cc/go/log"
|
|
)
|
|
|
|
// AppClient 用于管理单个请求的重试和负载均衡状态
|
|
type AppClient struct {
|
|
excludes map[string]bool
|
|
tryTimes int
|
|
Logger *log.Logger
|
|
App string
|
|
Method string
|
|
Path string
|
|
Data *map[string]any
|
|
Headers *map[string]string
|
|
}
|
|
|
|
func (ac *AppClient) logError(error string, extra ...any) {
|
|
if ac.Logger == nil {
|
|
ac.Logger = log.DefaultLogger
|
|
}
|
|
ac.Logger.Error("Discover Client: "+error, extra...)
|
|
}
|
|
|
|
func (ac *AppClient) Next(app string, request *http.Request) *NodeInfo {
|
|
return ac.NextWithNode(app, "", request)
|
|
}
|
|
|
|
func (ac *AppClient) CheckApp(app string) bool {
|
|
nodes := getAppNodes(app)
|
|
if nodes == nil {
|
|
if !addApp(app, "", true) {
|
|
ac.logError("app not found", "app", app, "calls", Config.Calls)
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
func (ac *AppClient) NextWithNode(app, withNode string, request *http.Request) *NodeInfo {
|
|
if ac.excludes == nil {
|
|
ac.excludes = make(map[string]bool)
|
|
}
|
|
|
|
allNodes := getAppNodes(app)
|
|
if len(allNodes) == 0 {
|
|
ac.logError("node not found", "app", app)
|
|
return nil
|
|
}
|
|
|
|
ac.tryTimes++
|
|
if withNode != "" {
|
|
ac.excludes[withNode] = true
|
|
return allNodes[withNode]
|
|
}
|
|
|
|
readyNodes := make([]*NodeInfo, 0)
|
|
for _, node := range allNodes {
|
|
if ac.excludes[node.Addr] || node.FailedTimes >= Config.CallRetryTimes {
|
|
continue
|
|
}
|
|
readyNodes = append(readyNodes, node)
|
|
}
|
|
|
|
if len(readyNodes) == 0 {
|
|
// 如果没有可用节点，尝试已经失败但未被本次请求排除的节点
|
|
for _, node := range allNodes {
|
|
if !ac.excludes[node.Addr] {
|
|
readyNodes = append(readyNodes, node)
|
|
}
|
|
}
|
|
}
|
|
|
|
var node *NodeInfo
|
|
if len(readyNodes) > 0 {
|
|
node = settedLoadBalancer.Next(ac, readyNodes, request)
|
|
if node != nil {
|
|
ac.excludes[node.Addr] = true
|
|
}
|
|
}
|
|
|
|
if node == nil {
|
|
ac.logError("no available node", "app", app, "tryTimes", ac.tryTimes)
|
|
}
|
|
|
|
return node
|
|
}
|