wjqserver/touka
1
0
Fork 0
mirror of https://github.com/infinite-iroha/touka.git synced 2026-06-13 15:47:38 +08:00
Code Issues Projects Releases Packages Wiki Activity Actions 13

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

Browse source 
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
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
context.go
View file

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