floraachy
/
console
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

remove old agent when endpoint exists

This commit is contained in:
liugq 2023-06-27 14:43:58 +08:00
parent b65cd42d86
commit 24063bb2ae
1 changed files with 20 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
modules/agent/api/instance.go
View File

@ -59,8 +59,7 @@ func (h *APIHandler) createInstance(w http.ResponseWriter, req *http.Request, ps
port = "8080"
}
obj.Endpoint = fmt.Sprintf("https://%s:%s", remoteIP, port)
obj.Tags = append(obj.Tags, "mtls")
obj.Tags = append(obj.Tags, "auto")
obj.Tags = append(obj.Tags, "mtls", "auto")
}
//fetch more information of agent instance
@ -101,6 +100,25 @@ func (h *APIHandler) createInstance(w http.ResponseWriter, req *http.Request, ps
log.Error(errMsg)
return
}
if token != "" {
err, result := orm.GetBy("endpoint", obj.Endpoint, oldInst)
if err != nil {
log.Error(err)
h.WriteError(w, err.Error(), http.StatusInternalServerError)
return
}
if len(result.Result) > 0 {
if m, ok := result.Result[0].(map[string]interface{}); ok {
if id, ok := m["id"].(string); ok {
oldInst.ID = id
err = orm.Delete(nil, oldInst)
if err != nil {
log.Error(err)
}
}
}
}
}
obj.Status = model.StatusOnline
err = orm.Create(nil, obj)