-
-
Notifications
You must be signed in to change notification settings - Fork 69
/
response.go
101 lines (76 loc) · 2.62 KB
/
response.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
package atreugo
import (
jsoniter "github.com/json-iterator/go"
"github.com/valyala/bytebufferpool"
"github.com/valyala/fasthttp"
)
func (ctx *RequestCtx) newResponse(contentType string, statusCode ...int) {
ctx.SetContentType(contentType)
if len(statusCode) > 0 {
ctx.SetStatusCode(statusCode[0])
} else {
ctx.SetStatusCode(fasthttp.StatusOK)
}
}
// JSONResponse return response with body in json format
func (ctx *RequestCtx) JSONResponse(body interface{}, statusCode ...int) error {
ctx.newResponse("application/json", statusCode...)
data, err := jsoniter.Marshal(body)
if err != nil {
return err
}
ctx.SetBody(data)
return nil
}
// HTTPResponse return response with body in html format
func (ctx *RequestCtx) HTTPResponse(body string, statusCode ...int) error {
ctx.newResponse("text/html; charset=utf-8", statusCode...)
ctx.SetBodyString(body)
return nil
}
// HTTPResponseBytes return response with body in html format
func (ctx *RequestCtx) HTTPResponseBytes(body []byte, statusCode ...int) error {
ctx.newResponse("text/html; charset=utf-8", statusCode...)
ctx.SetBody(body)
return nil
}
// TextResponse return response with body in text format
func (ctx *RequestCtx) TextResponse(body string, statusCode ...int) error {
ctx.newResponse("text/plain; charset=utf-8", statusCode...)
ctx.SetBodyString(body)
return nil
}
// TextResponseBytes return response with body in text format
func (ctx *RequestCtx) TextResponseBytes(body []byte, statusCode ...int) error {
ctx.newResponse("text/plain; charset=utf-8", statusCode...)
ctx.SetBody(body)
return nil
}
// RawResponse returns response without encoding the body.
func (ctx *RequestCtx) RawResponse(body string, statusCode ...int) error {
ctx.newResponse("application/octet-stream", statusCode...)
ctx.SetBodyString(body)
return nil
}
// RawResponseBytes returns response without encoding the body.
func (ctx *RequestCtx) RawResponseBytes(body []byte, statusCode ...int) error {
ctx.newResponse("application/octet-stream", statusCode...)
ctx.SetBody(body)
return nil
}
// FileResponse return a streaming response with file data.
func (ctx *RequestCtx) FileResponse(fileName, filePath, mimeType string) error {
fasthttp.ServeFile(ctx.RequestCtx, filePath)
buff := bytebufferpool.Get()
buff.SetString("attachment; filename=")
buff.WriteString(fileName)
ctx.Response.Header.Set("Content-Disposition", buff.String())
ctx.SetContentType(mimeType)
bytebufferpool.Put(buff)
return nil
}
// RedirectResponse redirect request to an especific url
func (ctx *RequestCtx) RedirectResponse(url string, statusCode int) error {
ctx.Redirect(url, statusCode)
return nil
}