From 92c7b725abbd456e4a7d5f85b8c20d1eed5430d6 Mon Sep 17 00:00:00 2001
From: xucong053 <xucong053@outlook.com>
Date: Wed, 16 Mar 2022 18:32:13 +0800
Subject: [PATCH] fix: unable to sense the quit of hrp in Prometheus

---
 internal/boomer/output.go | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/internal/boomer/output.go b/internal/boomer/output.go
index dff3f975..a07fe93f 100644
--- a/internal/boomer/output.go
+++ b/internal/boomer/output.go
@@ -119,15 +119,15 @@ func (o *ConsoleOutput) OnEvent(data map[string]interface{}) {
 
 	var state string
 	switch output.State {
-	case 1:
+	case stateInit:
 		state = "initializing"
-	case 2:
+	case stateSpawning:
 		state = "spawning"
-	case 3:
+	case stateRunning:
 		state = "running"
-	case 4:
+	case stateQuitting:
 		state = "quitting"
-	case 5:
+	case stateStopped:
 		state = "stopped"
 	}
 
@@ -463,7 +463,11 @@ func (o *PrometheusPusherOutput) OnStart() {
 
 // OnStop of PrometheusPusherOutput has nothing to do.
 func (o *PrometheusPusherOutput) OnStop() {
-
+	// update runner state: stopped
+	gaugeState.Set(float64(stateStopped))
+	if err := o.pusher.Push(); err != nil {
+		log.Error().Err(err).Msg("push to Pushgateway failed")
+	}
 }
 
 // OnEvent will push metric to Prometheus Pushgataway