From 896182417f7c3144e646308cfb7000a09dfbe1cb Mon Sep 17 00:00:00 2001
From: wjqserver <114663932+WJQSERVER@users.noreply.github.com>
Date: Wed, 11 Jun 2025 11:24:54 +0800
Subject: [PATCH] fix header writer after status issue

---
 context.go | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/context.go b/context.go
index dd0f1a3..d4b9a5b 100644
--- a/context.go
+++ b/context.go
@@ -602,11 +602,6 @@ func (c *Context) GetReqQueryString() string {
 // SetBodyStream 设置响应体为一个 io.Reader，并指定内容长度
 // 如果 contentSize 为 -1，则表示内容长度未知，将使用 Transfer-Encoding: chunked
 func (c *Context) SetBodyStream(reader io.Reader, contentSize int) {
-	// 确保在写入数据前设置状态码
-	if !c.Writer.Written() {
-		c.Writer.WriteHeader(http.StatusOK) // 默认 200 OK
-	}
-
 	// 如果指定了内容长度且大于等于 0，则设置 Content-Length 头部
 	if contentSize >= 0 {
 		c.Writer.Header().Set("Content-Length", fmt.Sprintf("%d", contentSize))
@@ -615,6 +610,11 @@ func (c *Context) SetBodyStream(reader io.Reader, contentSize int) {
 		c.Writer.Header().Del("Content-Length")
 	}
 
+	// 确保在写入数据前设置状态码
+	if !c.Writer.Written() {
+		c.Writer.WriteHeader(http.StatusOK) // 默认 200 OK
+	}
+
 	// 将 reader 的内容直接复制到 ResponseWriter
 	// ResponseWriter 实现了 io.Writer 接口
 	_, err := copyb.Copy(c.Writer, reader)