getFailureStatusCodeCategory to return error rather than empty string when unknown status code

Signed-off-by: Yijie Qin <[email protected]>

Author: qinxx108

notify/notify.go

@@ -51,8 +51,8 @@ type Peer interface {
 // to a notification pipeline.
 const MinTimeout = 10 * time.Second
 
-// defaultStatusCode is the default status code for numTotalFailedNotifications metric
-const defaultStatusCode = "5xx"
+// defaultStatusCodeCategory is the default status code category for numTotalFailedNotifications metric
+const defaultStatusCodeCategory = "5xx"
 
 // Notifier notifies about alerts under constraints of the given context. It
 // returns an error if unsuccessful and a flag whether the error is
@@ -665,13 +665,17 @@ func NewRetryStage(i Integration, groupName string, metrics *Metrics) *RetryStag
 func (r RetryStage) Exec(ctx context.Context, l log.Logger, alerts ...*types.Alert) (context.Context, []*types.Alert, error) {
 	r.metrics.numNotifications.WithLabelValues(r.integration.Name()).Inc()
 	ctx, alerts, err := r.exec(ctx, l, alerts...)
+
+	statusCodeCategory := defaultStatusCodeCategory
 	if err != nil {
 		if e, ok := errors.Cause(err).(*ErrorWithStatusCode); ok {
-			r.metrics.numTotalFailedNotifications.WithLabelValues(r.integration.Name(), getFailureStatusCodeCategory(e.StatusCode)).Inc()
-		} else {
-			r.metrics.numTotalFailedNotifications.WithLabelValues(r.integration.Name(), defaultStatusCode).Inc()
+			result, interErr := getFailureStatusCodeCategory(e.StatusCode)
+			if interErr == nil {
+				statusCodeCategory = result
+			}
 		}
 	}
+	r.metrics.numTotalFailedNotifications.WithLabelValues(r.integration.Name(), statusCodeCategory).Inc()
 	return ctx, alerts, err
 }
 

notify/util.go

@@ -167,15 +167,18 @@ func readAll(r io.Reader) string {
 	return string(bs)
 }
 
-func getFailureStatusCodeCategory(statusCode int) string {
+// getFailureStatusCodeCategory return the status code category for failure request
+// the status starts with 4 will return 4xx and starts with 5 will return 5xx
+// other than 4xx and 5xx input status will return an error.
+func getFailureStatusCodeCategory(statusCode int) (string, error) {
 	if statusCode/100 == 4 {
-		return "4xx"
+		return "4xx", nil
 	}
 	if statusCode/100 == 5 {
-		return "5xx"
+		return "5xx", nil
 	}
 
-	return ""
+	return "", fmt.Errorf("unexpected status code %v", statusCode)
 }
 
 // Retrier knows when to retry an HTTP request to a receiver. 2xx status codes