-
Notifications
You must be signed in to change notification settings - Fork 13
/
buf.go
46 lines (39 loc) · 880 Bytes
/
buf.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
package open_api
import (
"bytes"
"net/http"
)
type _ProxyWriterBuffer struct {
buf bytes.Buffer
statusCode int
header http.Header
}
func NewTemplateWriter() *_ProxyWriterBuffer {
b := &_ProxyWriterBuffer{}
b.Reset()
return b
}
func (t *_ProxyWriterBuffer) Reset() {
t.buf.Reset()
t.statusCode = 200
t.header = make(http.Header)
}
func (t *_ProxyWriterBuffer) WriteTo(w http.ResponseWriter) {
t.WriteHeaderTo(w)
w.WriteHeader(t.statusCode)
t.buf.WriteTo(w)
}
func (t *_ProxyWriterBuffer) WriteHeaderTo(w http.ResponseWriter) {
for k := range t.header {
w.Header().Set(k, t.header.Get(k))
}
}
func (t *_ProxyWriterBuffer) Header() http.Header {
return t.header
}
func (t *_ProxyWriterBuffer) Write(bytes []byte) (int, error) {
return t.buf.Write(bytes)
}
func (t *_ProxyWriterBuffer) WriteHeader(statusCode int) {
t.statusCode = statusCode
}