service测试完成

This commit is contained in:
Liujian
2021-07-22 18:26:29 +08:00
parent f2e003154f
commit 779531fef2
24 changed files with 291 additions and 507 deletions

View File

@@ -5,17 +5,17 @@ import "github.com/eolinker/goku-eosc/discovery"
//agent 从属于HTTPCheck,实现了IHealthChecker接口
type agent struct {
agentID string
*HTTPCheck
checker *HTTPCheck
}
//NewAgent 创建agent
func NewAgent(agentID string) discovery.IHealthChecker {
return &agent{agentID: agentID}
func NewAgent(agentID string, checker *HTTPCheck) discovery.IHealthChecker {
return &agent{agentID: agentID, checker: checker}
}
//AddToCheck 将节点添加进HTTPCheck的检查列表
func (a *agent) AddToCheck(node discovery.INode) error {
a.addToCheck(&checkNode{
a.checker.addToCheck(&checkNode{
node: node,
agentID: a.agentID,
})
@@ -24,6 +24,6 @@ func (a *agent) AddToCheck(node discovery.INode) error {
//Stop 停止agent并且将HTTPCheck中属于该agent的正在检查的所有节点都移除
func (a *agent) Stop() error {
a.stop(a.agentID)
a.checker.stop(a.agentID)
return nil
}

View File

@@ -77,7 +77,7 @@ func (h *HTTPCheck) doCheckLoop() {
//Agent 生成一个agent
func (h *HTTPCheck) Agent() (discovery.IHealthChecker, error) {
return NewAgent(uuid.New()), nil
return NewAgent(uuid.New(), h), nil
}
//Reset 重置HTTPCheck的配置