podman
190 строк · 4.5 Кб
1package jsoniter
2
3var digits []uint32
4
5func init() {
6digits = make([]uint32, 1000)
7for i := uint32(0); i < 1000; i++ {
8digits[i] = (((i / 100) + '0') << 16) + ((((i / 10) % 10) + '0') << 8) + i%10 + '0'
9if i < 10 {
10digits[i] += 2 << 24
11} else if i < 100 {
12digits[i] += 1 << 24
13}
14}
15}
16
17func writeFirstBuf(space []byte, v uint32) []byte {
18start := v >> 24
19if start == 0 {
20space = append(space, byte(v>>16), byte(v>>8))
21} else if start == 1 {
22space = append(space, byte(v>>8))
23}
24space = append(space, byte(v))
25return space
26}
27
28func writeBuf(buf []byte, v uint32) []byte {
29return append(buf, byte(v>>16), byte(v>>8), byte(v))
30}
31
32// WriteUint8 write uint8 to stream
33func (stream *Stream) WriteUint8(val uint8) {
34stream.buf = writeFirstBuf(stream.buf, digits[val])
35}
36
37// WriteInt8 write int8 to stream
38func (stream *Stream) WriteInt8(nval int8) {
39var val uint8
40if nval < 0 {
41val = uint8(-nval)
42stream.buf = append(stream.buf, '-')
43} else {
44val = uint8(nval)
45}
46stream.buf = writeFirstBuf(stream.buf, digits[val])
47}
48
49// WriteUint16 write uint16 to stream
50func (stream *Stream) WriteUint16(val uint16) {
51q1 := val / 1000
52if q1 == 0 {
53stream.buf = writeFirstBuf(stream.buf, digits[val])
54return
55}
56r1 := val - q1*1000
57stream.buf = writeFirstBuf(stream.buf, digits[q1])
58stream.buf = writeBuf(stream.buf, digits[r1])
59return
60}
61
62// WriteInt16 write int16 to stream
63func (stream *Stream) WriteInt16(nval int16) {
64var val uint16
65if nval < 0 {
66val = uint16(-nval)
67stream.buf = append(stream.buf, '-')
68} else {
69val = uint16(nval)
70}
71stream.WriteUint16(val)
72}
73
74// WriteUint32 write uint32 to stream
75func (stream *Stream) WriteUint32(val uint32) {
76q1 := val / 1000
77if q1 == 0 {
78stream.buf = writeFirstBuf(stream.buf, digits[val])
79return
80}
81r1 := val - q1*1000
82q2 := q1 / 1000
83if q2 == 0 {
84stream.buf = writeFirstBuf(stream.buf, digits[q1])
85stream.buf = writeBuf(stream.buf, digits[r1])
86return
87}
88r2 := q1 - q2*1000
89q3 := q2 / 1000
90if q3 == 0 {
91stream.buf = writeFirstBuf(stream.buf, digits[q2])
92} else {
93r3 := q2 - q3*1000
94stream.buf = append(stream.buf, byte(q3+'0'))
95stream.buf = writeBuf(stream.buf, digits[r3])
96}
97stream.buf = writeBuf(stream.buf, digits[r2])
98stream.buf = writeBuf(stream.buf, digits[r1])
99}
100
101// WriteInt32 write int32 to stream
102func (stream *Stream) WriteInt32(nval int32) {
103var val uint32
104if nval < 0 {
105val = uint32(-nval)
106stream.buf = append(stream.buf, '-')
107} else {
108val = uint32(nval)
109}
110stream.WriteUint32(val)
111}
112
113// WriteUint64 write uint64 to stream
114func (stream *Stream) WriteUint64(val uint64) {
115q1 := val / 1000
116if q1 == 0 {
117stream.buf = writeFirstBuf(stream.buf, digits[val])
118return
119}
120r1 := val - q1*1000
121q2 := q1 / 1000
122if q2 == 0 {
123stream.buf = writeFirstBuf(stream.buf, digits[q1])
124stream.buf = writeBuf(stream.buf, digits[r1])
125return
126}
127r2 := q1 - q2*1000
128q3 := q2 / 1000
129if q3 == 0 {
130stream.buf = writeFirstBuf(stream.buf, digits[q2])
131stream.buf = writeBuf(stream.buf, digits[r2])
132stream.buf = writeBuf(stream.buf, digits[r1])
133return
134}
135r3 := q2 - q3*1000
136q4 := q3 / 1000
137if q4 == 0 {
138stream.buf = writeFirstBuf(stream.buf, digits[q3])
139stream.buf = writeBuf(stream.buf, digits[r3])
140stream.buf = writeBuf(stream.buf, digits[r2])
141stream.buf = writeBuf(stream.buf, digits[r1])
142return
143}
144r4 := q3 - q4*1000
145q5 := q4 / 1000
146if q5 == 0 {
147stream.buf = writeFirstBuf(stream.buf, digits[q4])
148stream.buf = writeBuf(stream.buf, digits[r4])
149stream.buf = writeBuf(stream.buf, digits[r3])
150stream.buf = writeBuf(stream.buf, digits[r2])
151stream.buf = writeBuf(stream.buf, digits[r1])
152return
153}
154r5 := q4 - q5*1000
155q6 := q5 / 1000
156if q6 == 0 {
157stream.buf = writeFirstBuf(stream.buf, digits[q5])
158} else {
159stream.buf = writeFirstBuf(stream.buf, digits[q6])
160r6 := q5 - q6*1000
161stream.buf = writeBuf(stream.buf, digits[r6])
162}
163stream.buf = writeBuf(stream.buf, digits[r5])
164stream.buf = writeBuf(stream.buf, digits[r4])
165stream.buf = writeBuf(stream.buf, digits[r3])
166stream.buf = writeBuf(stream.buf, digits[r2])
167stream.buf = writeBuf(stream.buf, digits[r1])
168}
169
170// WriteInt64 write int64 to stream
171func (stream *Stream) WriteInt64(nval int64) {
172var val uint64
173if nval < 0 {
174val = uint64(-nval)
175stream.buf = append(stream.buf, '-')
176} else {
177val = uint64(nval)
178}
179stream.WriteUint64(val)
180}
181
182// WriteInt write int to stream
183func (stream *Stream) WriteInt(val int) {
184stream.WriteInt64(int64(val))
185}
186
187// WriteUint write uint to stream
188func (stream *Stream) WriteUint(val uint) {
189stream.WriteUint64(uint64(val))
190}
191