Update context.go
Some checks failed
Go Test / test (push) Has been cancelled

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
This commit is contained in:
WJQSERVER 2026-03-30 01:17:59 +08:00 committed by GitHub
parent d0fa14c3c5
commit fba6fedfc5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -478,8 +478,9 @@ func (c *Context) WANF(code int, obj any) {
// WANFBuf 先将 WANF 编码到 buffer, 成功后再写入状态码和响应体.
func (c *Context) WANFBuf(code int, obj any) {
data, err := wanf.Marshal(obj)
if err != nil {
var buf bytes.Buffer
encoder := wanf.NewStreamEncoder(&buf)
if err := encoder.Encode(obj); err != nil {
errMsg := fmt.Errorf("failed to encode WANF: %w", err)
c.AddError(errMsg)
c.ErrorUseHandle(http.StatusInternalServerError, errMsg)
@ -487,7 +488,7 @@ func (c *Context) WANFBuf(code int, obj any) {
}
c.Writer.Header().Set("Content-Type", "application/vnd.wjqserver.wanf; charset=utf-8")
c.Writer.WriteHeader(code)
c.Writer.Write(data)
c.Writer.Write(buf.Bytes())
}
// HTML 渲染 HTML 模板