Fix agent polling (#3378)

This commit is contained in:
Anbraten
2024-02-16 10:04:13 +01:00
committed by GitHub
parent bfff8dbc6f
commit 0e0d0188a0
5 changed files with 85 additions and 33 deletions

View File

@@ -137,6 +137,9 @@ func PatchAgent(c *gin.Context) {
}
agent.Name = in.Name
agent.NoSchedule = in.NoSchedule
if agent.NoSchedule {
server.Config.Services.Queue.KickAgentWorkers(agent.ID)
}
err = _store.AgentUpdate(agent)
if err != nil {
@@ -214,6 +217,9 @@ func DeleteAgent(c *gin.Context) {
}
}
// kick workers to remove the agent from the queue
server.Config.Services.Queue.KickAgentWorkers(agent.ID)
if err = _store.AgentDelete(agent); err != nil {
c.String(http.StatusInternalServerError, "Error deleting user. %s", err)
return