Update backend.go

simplified count calls
coupergateway · Dec 5, 2022 · de29970 · de29970
1 parent 17ccc70
commit de29970
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/handler/transport/backend.go b/handler/transport/backend.go
@@ -314,9 +314,13 @@ func (b *Backend) innerRoundTrip(req *http.Request, tc *Config, deadlineErr <-ch
 	endSeconds := time.Since(start).Seconds()
 
 	statusKey := attribute.Key("code")
+	if beresp != nil {
+		attrs = append(attrs, statusKey.Int(beresp.StatusCode))
+	}
+	defer counter.Observe(req.Context(), 1, attrs...)
+	defer duration.Record(req.Context(), endSeconds, attrs...)
+
 	if err != nil {
-		defer counter.Observe(req.Context(), 1, attrs...)
-		defer duration.Record(req.Context(), endSeconds, attrs...)
 		select {
 		case derr := <-deadlineErr:
 			if derr != nil {
@@ -331,10 +335,6 @@ func (b *Backend) innerRoundTrip(req *http.Request, tc *Config, deadlineErr <-ch
 		}
 	}
 
-	attrs = append(attrs, statusKey.Int(beresp.StatusCode))
-	defer counter.Observe(req.Context(), 1, attrs...)
-	defer duration.Record(req.Context(), endSeconds, attrs...)
-
 	return beresp, nil
 }