fix response status code panic (#5117)

kama910 · web-flow · commit a63bbb07fe4f · 2023-02-02T02:51:01.000Z
Signed-off-by: Kama Huang &lt;kamatogo13@gmail.com&gt;
diff --git a/pkg/frontend/transport/handler.go b/pkg/frontend/transport/handler.go
@@ -147,9 +147,11 @@ func (f *Handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 		f.reportSlowQuery(r, queryString, queryResponseTime)
 	}
 	if f.cfg.QueryStatsEnabled {
-		statusCode := resp.StatusCode
+		var statusCode int
 		if err != nil {
 			statusCode = getStatusCodeFromError(err)
+		} else if resp != nil {
+			statusCode = resp.StatusCode
 		}
 
 		f.reportQueryStats(r, queryString, queryResponseTime, stats, err, statusCode)

Original file line number	Diff line number	Diff line change
`@@ -147,9 +147,11 @@ func (f Handler) ServeHTTP(w http.ResponseWriter, r http.Request) {`
`147`	`147`	`f.reportSlowQuery(r, queryString, queryResponseTime)`
`148`	`148`	`}`
`149`	`149`	`if f.cfg.QueryStatsEnabled {`
`150`		`- statusCode := resp.StatusCode`
	`150`	`+ var statusCode int`
`151`	`151`	`if err != nil {`
`152`	`152`	`statusCode = getStatusCodeFromError(err)`
	`153`	`+ } else if resp != nil {`
	`154`	`+ statusCode = resp.StatusCode`
`153`	`155`	`}`
`154`	`156`
`155`	`157`	`f.reportQueryStats(r, queryString, queryResponseTime, stats, err, statusCode)`