chore: avoid nested WithCodeResponseWriter (#3406)

2023-07-11 23:59:43 +08:00
parent e8c1e6e09b
commit 13cdbdc98b
10 changed files with 81 additions and 39 deletions
--- a/rest/handler/sheddinghandler.go
+++ b/rest/handler/sheddinghandler.go
@@ -41,7 +41,7 @@ func SheddingHandler(shedder load.Shedder, metrics *stat.Metrics) func(http.Hand
 				return
 			}

-			cw := &response.WithCodeResponseWriter{Writer: w}
+			cw := response.NewWithCodeResponseWriter(w)
 			defer func() {
 				if cw.Code == http.StatusServiceUnavailable {
 					promise.Fail()