Mark agent inactive on ack of MIGRATE action (#5260)

michalpristas · web-flow · commit a181ca4d849f · 2025-08-18T14:23:20.000+02:00
diff --git a/internal/pkg/api/handleAck.go b/internal/pkg/api/handleAck.go
@@ -34,6 +34,7 @@ import (
 const (
 	TypeUnenroll = "UNENROLL"
 	TypeUpgrade  = "UPGRADE"
+	TypeMigrate  = "MIGRATE"
 )
 
 var (
@@ -305,7 +306,7 @@ func (ack *AckT) handleAckEvents(ctx context.Context, zlog zerolog.Logger, agent
 			setResult(n, http.StatusOK)
 		}
 
-		if event.Error == nil && action.Type == TypeUnenroll {
+		if event.Error == nil && (action.Type == TypeUnenroll || action.Type == TypeMigrate) {
 			unenrollIdxs = append(unenrollIdxs, n)
 		}
 		span.End()

Original file line number	Diff line number	Diff line change
`@@ -34,6 +34,7 @@ import (`
`34`	`34`	`const (`
`35`	`35`	`TypeUnenroll = "UNENROLL"`
`36`	`36`	`TypeUpgrade = "UPGRADE"`
	`37`	`+ TypeMigrate = "MIGRATE"`
`37`	`38`	`)`
`38`	`39`
`39`	`40`	`var (`
`@@ -305,7 +306,7 @@ func (ack *AckT) handleAckEvents(ctx context.Context, zlog zerolog.Logger, agent`
`305`	`306`	`setResult(n, http.StatusOK)`
`306`	`307`	`}`
`307`	`308`
`308`		`- if event.Error == nil && action.Type == TypeUnenroll {`
	`309`	`+ if event.Error == nil && (action.Type == TypeUnenroll \|\| action.Type == TypeMigrate) {`
`309`	`310`	`unenrollIdxs = append(unenrollIdxs, n)`
`310`	`311`	`}`
`311`	`312`	`span.End()`