draincordon

phuhung273 · phuhung273 · commit 6e910d08b274 · 2024-10-29T21:29:53.000+07:00
diff --git a/pkg/interruptionevent/draincordon/handler.go b/pkg/interruptionevent/draincordon/handler.go
@@ -114,6 +114,7 @@ func (h *Handler) HandleEvent(drainEvent *monitor.InterruptionEvent) error {
 		h.commonHandler.InterruptionEventStore.CancelInterruptionEvent(drainEvent.EventID)
 	} else {
 		h.commonHandler.InterruptionEventStore.MarkAllAsProcessed(nodeName)
+		h.commonHandler.Metrics.NodesDec(nodeName)
 	}
 
 	if (err == nil || (!nodeFound && h.commonHandler.NthConfig.DeleteSqsMsgIfNodeNotFound)) && drainEvent.PostDrainTask != nil {

Original file line number	Diff line number	Diff line change
`@@ -114,6 +114,7 @@ func (h Handler) HandleEvent(drainEvent monitor.InterruptionEvent) error {`
`114`	`114`	`h.commonHandler.InterruptionEventStore.CancelInterruptionEvent(drainEvent.EventID)`
`115`	`115`	`} else {`
`116`	`116`	`h.commonHandler.InterruptionEventStore.MarkAllAsProcessed(nodeName)`
	`117`	`+ h.commonHandler.Metrics.NodesDec(nodeName)`
`117`	`118`	`}`
`118`	`119`
`119`	`120`	`if (err == nil \|\| (!nodeFound && h.commonHandler.NthConfig.DeleteSqsMsgIfNodeNotFound)) && drainEvent.PostDrainTask != nil {`