validate hosts unique (#371)

* support tt automation by job

* format

* import order

* use map to avoid repetition

Co-authored-by: dujiashu <dujiashu@didiglobal.com>
This commit is contained in:
dujiashu 2020-11-01 11:07:17 +08:00 committed by GitHub
parent 6f999f6a87
commit bec893d662
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 5 deletions

View File

@ -12,9 +12,9 @@ import (
"github.com/toolkits/pkg/net/httplib"
"github.com/toolkits/pkg/slice"
"github.com/didi/nightingale/src/common/address"
"github.com/didi/nightingale/src/models"
"github.com/didi/nightingale/src/modules/job/config"
"github.com/didi/nightingale/src/common/address"
)
type taskForm struct {
@ -658,11 +658,16 @@ func taskRunForTT(c *gin.Context) {
dangerous(task.Save(arr, "start"))
go func() {
var arr2Map = map[string]int{}
for _, a := range arr {
arr2Map[a] = 1
}
for {
var (
restHosts []string
restHosts = map[string]int{}
)
for _, h := range arr {
for h, _ := range arr2Map {
th, err := models.TaskHostGet(task.Id, h)
if err == nil {
if th.Status == "killed" {
@ -702,13 +707,14 @@ func taskRunForTT(c *gin.Context) {
logger.Errorf("send callback to ticket, err: %v", err)
}
} else {
restHosts = append(restHosts, h)
restHosts[h] = 1
}
} else {
logger.Errorf("get task_host err: %v", err)
}
}
arr = restHosts
arr2Map = restHosts
time.Sleep(time.Second)
}
}()