feat: support ABORTED status in ::status:: directive of exec probe

README.md

@@ -166,7 +166,7 @@ hello world
 This output is reporting latency is `123.456ms`, status is `FAILURE`, and message is `hello world`.
 
 - `::latency::`: Reports the latency of service in milliseconds.
-- `::status::`: Reports the status of service in `healthy`, `failure`, or `unknown`.
+- `::status::`: Reports the status of service in `healthy`, `failure`, `aborted`, or `unknown`.
 
 Ayd uses the last value if found multiple reports in single output.
 

probe/exec.go

@@ -58,17 +58,10 @@ func getLatencyByMessage(message string, default_ time.Duration) (replacedMessag
 }
 
 func getStatusByMessage(message string, default_ store.Status) (replacedMessage string, status store.Status) {
-	statusRe := regexp.MustCompile("(?m)^::status::((?i:healthy|failure|unknown))(?:\n|$)")
+	statusRe := regexp.MustCompile("(?m)^::status::((?i:healthy|failure|aborted|unknown))(?:\n|$)")
 
 	if m := statusRe.FindAllStringSubmatch(message, -1); m != nil {
-		switch strings.ToLower(m[len(m)-1][1]) {
-		case "healthy":
-			status = store.STATUS_HEALTHY
-		case "failure":
-			status = store.STATUS_FAILURE
-		case "unknown":
-			status = store.STATUS_UNKNOWN
-		}
+		status = store.ParseStatus(strings.ToUpper(m[len(m)-1][1]))
 		return strings.Trim(statusRe.ReplaceAllString(message, ""), "\n"), status
 	}
 

probe/exec_internal_test.go

@@ -47,6 +47,7 @@ func TestGetStatusByMessage(t *testing.T) {
 	}{
 		{"::status::healthy", store.STATUS_UNKNOWN, "", store.STATUS_HEALTHY},
 		{"::status::Failure", store.STATUS_UNKNOWN, "", store.STATUS_FAILURE},
+		{"::status::aborted", store.STATUS_UNKNOWN, "", store.STATUS_ABORTED},
 		{"::status::UNKNOWN", store.STATUS_HEALTHY, "", store.STATUS_UNKNOWN},
 		{"::status::abcdefg", store.STATUS_UNKNOWN, "::status::abcdefg", store.STATUS_UNKNOWN},
 		{"hello\n::status::FAILURE\nworld", store.STATUS_UNKNOWN, "hello\nworld", store.STATUS_FAILURE},