code: return ok before set header

nerdneilsfield · Nov 26, 2023 · f2f1f0b · f2f1f0b
1 parent 918a196
commit f2f1f0b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.go b/main.go
@@ -161,10 +161,10 @@ func LoadBalancerHandler(w http.ResponseWriter, r *http.Request) {
 		http.Error(w, "Failed to read response", http.StatusInternalServerError)
 		return
 	}
+	w.WriteHeader(http.StatusOK)
 	for k, v := range resp.Header {
 		w.Header()[k] = v
 	}
-	w.WriteHeader(http.StatusOK)
 	_, err = w.Write(body)
 	if err != nil {
 		// 处理写入错误