go/json-iterator
1
0
Fork 0
mirror of https://github.com/json-iterator/go.git synced 2025-06-24 23:16:47 +02:00
Code Issues Releases Activity

fix write string

Browse Source
This commit is contained in:
Tao Wen
2017-01-22 00:04:08 +08:00
parent 102cd8748e
commit 1163c348f6
2 changed files with 77 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

82
feature_stream.go
View File

@ -80,6 +80,19 @@ func (b *Stream) writeByte(c byte) error {
return nil
}
func (b *Stream) writeTwoBytes(c1 byte, c2 byte) error {
if b.Error != nil {
return b.Error
}
if b.Available() <= 1 && b.Flush() != nil {
return b.Error
}
b.buf[b.n] = c1
b.buf[b.n + 1] = c2
b.n += 2
return nil
}
// Flush writes any buffered data to the underlying io.Writer.
func (b *Stream) Flush() error {
if b.Error != nil {
@ -118,20 +131,67 @@ func (b *Stream) WriteRaw(s string) {
b.n += n
}
func (b *Stream) WriteString(s string) {
b.writeByte('"')
for len(s) > b.Available() && b.Error == nil {
n := copy(b.buf[b.n:], s)
b.n += n
s = s[n:]
b.Flush()
func (stream *Stream) WriteString(s string) {
valLen := len(s)
toWriteLen := valLen
bufLengthMinusTwo := len(stream.buf) - 2 // make room for the quotes
if stream.n + toWriteLen > bufLengthMinusTwo {
toWriteLen = bufLengthMinusTwo - stream.n
}
if b.Error != nil {
if toWriteLen < 0 {
stream.Flush()
if stream.n + toWriteLen > bufLengthMinusTwo {
toWriteLen = bufLengthMinusTwo - stream.n
}
}
n := stream.n
stream.buf[n] = '"'
n++
// write string, the fast path, without utf8 and escape support
i := 0
for ; i < toWriteLen; i++ {
c := s[i]
if c > 31 && c != '"' && c != '\\' {
stream.buf[n] = c
n++
} else {
break;
}
}
if i == valLen {
stream.buf[n] = '"'
n++
stream.n = n
return
}
n := copy(b.buf[b.n:], s)
b.n += n
b.writeByte('"')
stream.n = n
// for the remaining parts, we process them char by char
stream.writeStringSlowPath(s, i, valLen);
stream.writeByte('"')
}
func (stream *Stream) writeStringSlowPath(s string, i int, valLen int) {
for ; i < valLen; i++ {
c := s[i]
switch (c) {
case '"':
stream.writeTwoBytes('\\', '"')
case '\\':
stream.writeTwoBytes('\\', '\\')
case '\b':
stream.writeTwoBytes('\\', 'b')
case '\f':
stream.writeTwoBytes('\\', 'f')
case '\n':
stream.writeTwoBytes('\\', 'n')
case '\r':
stream.writeTwoBytes('\\', 'r')
case '\t':
stream.writeTwoBytes('\\', 't')
default:
stream.writeByte(c);
}
}
}
func (stream *Stream) WriteNil() {