-
Notifications
You must be signed in to change notification settings - Fork 8
/
builder.go
82 lines (67 loc) · 1.46 KB
/
builder.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
package texts
import (
"bytes"
"fmt"
"sync"
"unsafe"
)
var (
builders = sync.Pool{
New: func() interface{} {
return &Builder{}
},
}
)
func GetBuilder() *Builder {
return builders.Get().(*Builder)
}
func PutBuilder(b *Builder) {
b.Reset()
builders.Put(b)
}
// Builder is used to efficiently build a string using Write methods. It minimizes memory copying.
// The zero value is ready to use.
type Builder bytes.Buffer
func (b *Builder) Reset() *Builder {
(*bytes.Buffer)(b).Reset()
return b
}
func (b *Builder) Truncate(n int) *Builder {
(*bytes.Buffer)(b).Truncate(n)
return b
}
func (b *Builder) Write(p []byte) (n int, err error) {
return (*bytes.Buffer)(b).Write(p)
}
func (b *Builder) WriteByte(c byte) *Builder {
(*bytes.Buffer)(b).WriteByte(c)
return b
}
func (b *Builder) WriteString(s ...string) *Builder {
buf := (*bytes.Buffer)(b)
for _, e := range s {
buf.WriteString(e)
}
return b
}
func (b *Builder) WriteStringer(s ...fmt.Stringer) *Builder {
buf := (*bytes.Buffer)(b)
for _, e := range s {
buf.WriteString(e.String())
}
return b
}
func (b *Builder) WriteFormat(format string, args ...interface{}) *Builder {
(*bytes.Buffer)(b).WriteString(fmt.Sprintf(format, args...))
return b
}
func (b *Builder) Len() int {
return (*bytes.Buffer)(b).Len()
}
func (b *Builder) Bytes() []byte {
return (*bytes.Buffer)(b).Bytes()
}
func (b *Builder) String() string {
buf := (*bytes.Buffer)(b).Bytes()
return *(*string)(unsafe.Pointer(&buf))
}