go/contracts
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[feat] 优化 support/response, 并新增 NewHttpStatusResponse

Browse Source
This commit is contained in:
what 2024-10-24 15:34:47 +08:00
parent 58511825d3
commit dae5b130bb
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
support/response.go
View File

@ -35,13 +35,18 @@ var wsUpgrader = websocket.Upgrader{
} }
func (this HttpStatusResponse) Get(path ...string) req.GlobalParams { func (this HttpStatusResponse) Get(path ...string) req.GlobalParams {
if this.body == nil {
return req.NewGlobalParam("", nil)
}
return this.body.Get(path...) return this.body.Get(path...)
} }
func (this HttpStatusResponse) Send(w http.ResponseWriter, r *http.Request) { func (this HttpStatusResponse) Send(w http.ResponseWriter, r *http.Request) {
w.WriteHeader(this.status) w.WriteHeader(this.status)
if this.body != nil {
this.body.Send(w, r) this.body.Send(w, r)
} }
}
func (this RawResponse) Get(path ...string) req.GlobalParams { func (this RawResponse) Get(path ...string) req.GlobalParams {
return lo.Ternary(len(path) == 0, req.NewGlobalParam(string(this.raw), nil), req.NewGlobalParam(string(this.raw), nil).Get(strings.Join(path, "."))) return lo.Ternary(len(path) == 0, req.NewGlobalParam(string(this.raw), nil), req.NewGlobalParam(string(this.raw), nil).Get(strings.Join(path, ".")))