zlei6/go-zero
1
0
Fork 0
mirror of https://github.com/zeromicro/go-zero.git synced 2026-05-10 16:30:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

chore: add unit test for WithCodeResponseWriter (#5028)

Browse Source 
Signed-off-by: kevin <wanjunfeng@gmail.com>
This commit is contained in:
Kevin Wan
2025-07-25 21:45:47 +08:00
committed by GitHub
parent 0be63c3625
commit 25f37ca750
5 changed files with 44 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
rest/internal/response/withcoderesponsewriter.go
View File

@@ -49,6 +49,12 @@ func (w *WithCodeResponseWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) {
return nil, nil, errors.New("server doesn't support hijacking")
}
// Unwrap returns the underlying http.ResponseWriter.
// This is used by http.ResponseController to unwrap the response writer.
func (w *WithCodeResponseWriter) Unwrap() http.ResponseWriter {
return w.Writer
}
// Write writes bytes into w.
func (w *WithCodeResponseWriter) Write(bytes []byte) (int, error) {
return w.Writer.Write(bytes)
@@ -59,8 +65,3 @@ func (w *WithCodeResponseWriter) WriteHeader(code int) {
w.Writer.WriteHeader(code)
w.Code = code
}
// Unwrap returns the underlying ResponseWriter.
func (w *WithCodeResponseWriter) Unwrap() http.ResponseWriter {
return w.Writer
}