用户登录和接口鉴权

vendor/github.com/bytedance/sonic/ast/encode.go

@@ -17,12 +17,12 @@
 package ast
 
 import (
-	"sync"
-	"unicode/utf8"
+    `sync`
+    `unicode/utf8`
+)
 
-	"github.com/bytedance/gopkg/lang/dirtmake"
-	"github.com/bytedance/sonic/internal/rt"
-	"github.com/bytedance/sonic/option"
+const (
+    _MaxBuffer = 1024    // 1KB buffer size
 )
 
 func quoteString(e *[]byte, s string) {
@@ -30,7 +30,7 @@ func quoteString(e *[]byte, s string) {
     start := 0
     for i := 0; i < len(s); {
         if b := s[i]; b < utf8.RuneSelf {
-            if rt.SafeSet[b] {
+            if safeSet[b] {
                 i++
                 continue
             }
@@ -54,8 +54,8 @@ func quoteString(e *[]byte, s string) {
                 // user-controlled strings are rendered into JSON
                 // and served to some browsers.
                 *e = append(*e, `u00`...)
-                *e = append(*e, rt.Hex[b>>4])
-                *e = append(*e, rt.Hex[b&0xF])
+                *e = append(*e, hex[b>>4])
+                *e = append(*e, hex[b&0xF])
             }
             i++
             start = i
@@ -76,7 +76,7 @@ func quoteString(e *[]byte, s string) {
                 *e = append(*e, s[start:i]...)
             }
             *e = append(*e, `\u202`...)
-            *e = append(*e, rt.Hex[c&0xF])
+            *e = append(*e, hex[c&0xF])
             i += size
             start = i
             continue
@@ -92,29 +92,16 @@ func quoteString(e *[]byte, s string) {
 var bytesPool   = sync.Pool{}
 
 func (self *Node) MarshalJSON() ([]byte, error) {
-	if self == nil {
-		return bytesNull, nil
-	}
-
-    // fast path for raw node
-    if self.isRaw() {
-        return rt.Str2Mem(self.toString()), nil
-    }
-
     buf := newBuffer()
     err := self.encode(buf)
     if err != nil {
         freeBuffer(buf)
         return nil, err
     }
-    var ret []byte
-    if !rt.CanSizeResue(cap(*buf)) {
-        ret = *buf
-    } else {
-        ret = dirtmake.Bytes(len(*buf), len(*buf))
-        copy(ret, *buf)
-        freeBuffer(buf)
-    }
+
+    ret := make([]byte, len(*buf))
+    copy(ret, *buf)
+    freeBuffer(buf)
     return ret, err
 }
 
@@ -122,24 +109,21 @@ func newBuffer() *[]byte {
     if ret := bytesPool.Get(); ret != nil {
         return ret.(*[]byte)
     } else {
-        buf := make([]byte, 0, option.DefaultAstBufferSize)
+        buf := make([]byte, 0, _MaxBuffer)
         return &buf
     }
 }
 
 func freeBuffer(buf *[]byte) {
-    if !rt.CanSizeResue(cap(*buf)) {
-        return
-    }
     *buf = (*buf)[:0]
     bytesPool.Put(buf)
 }
 
 func (self *Node) encode(buf *[]byte) error {
-    if self.isRaw() {
+    if self.IsRaw() {
         return self.encodeRaw(buf)
     }
-    switch int(self.itype()) {
+    switch self.Type() {
         case V_NONE  : return ErrNotExist
         case V_ERROR : return self.Check()
         case V_NULL  : return self.encodeNull(buf)
@@ -155,21 +139,16 @@ func (self *Node) encode(buf *[]byte) error {
 }
 
 func (self *Node) encodeRaw(buf *[]byte) error {
-    lock := self.rlock()
-    if !self.isRaw() {
-        self.runlock()
-        return self.encode(buf)
-    }
-    raw := self.toString()
-    if lock {
-        self.runlock()
+    raw, err := self.Raw()
+    if err != nil {
+        return err
     }
     *buf = append(*buf, raw...)
     return nil
 }
 
 func (self *Node) encodeNull(buf *[]byte) error {
-    *buf = append(*buf, strNull...)
+    *buf = append(*buf, bytesNull...)
     return nil
 }