feat:(ast) support arm64

Makefile

@@ -23,9 +23,9 @@ CPU_avx			:= amd64
 CPU_avx2		:= amd64
 CPU_sse  		:= amd64
 
-TMPL_avx		:= fastint_amd64_test fastfloat_amd64_test native_amd64_test recover_amd64_test
-TMPL_avx2		:= fastint_amd64_test fastfloat_amd64_test native_amd64_test recover_amd64_test
-TMPL_sse 		:= fastint_amd64_test fastfloat_amd64_test native_amd64_test recover_amd64_test
+TMPL_avx		:= native_test fastint_test fastfloat_test native_amd64 recover_test
+TMPL_avx2		:= native_test fastint_test fastfloat_test native_amd64 recover_test
+TMPL_sse 		:= native_test fastint_test fastfloat_test native_amd64 recover_test
 
 CFLAGS_avx		:= -msse -mno-sse4 -mavx -mpclmul -mno-avx2 -mstack-alignment=0 -DUSE_AVX=1 -DUSE_AVX2=0
 CFLAGS_avx2		:= -msse -mno-sse4 -mavx -mpclmul -mavx2 -mstack-alignment=0 -DUSE_AVX=1 -DUSE_AVX2=1 
@@ -54,10 +54,14 @@ define build_tmpl
 	$(eval @arch := $(1))
 	$(eval @tmpl := $(2))
 	$(eval @dest := $(3))
+	$(eval @cpu  := $(4))
 
 ${@dest}: ${@tmpl}
 	mkdir -p $(dir ${@dest})
-	echo '// Code generated by Makefile, DO NOT EDIT.' > ${@dest}
+	echo '// +build ${@cpu}' > ${@dest}
+	echo >> ${@dest}
+	echo >> ${@dest}
+	echo '// Code generated by Makefile, DO NOT EDIT.' >> ${@dest}
 	echo >> ${@dest}
 	sed -e 's/{{PACKAGE}}/${@arch}/g' ${@tmpl} >> ${@dest}
 endef
@@ -91,7 +95,8 @@ $(foreach 							\
 		build_tmpl,					\
 		$(1),						\
 		${OUT_DIR}/${tmpl}.tmpl,	\
-		${OUT_DIR}/$(1)/${tmpl}.go	\
+		${OUT_DIR}/$(1)/${tmpl}.go,	\
+		$(value CPU_$(1))           \
 	))								\
 )
 endef

ast/api_amd64.go → ast/api.go

@@ -1,4 +1,4 @@
-// +build amd64,go1.16,!go1.23
+// +build amd64,go1.16,!go1.23 arm64,go1.20,!go1.23
 
 /*
  * Copyright 2022 ByteDance Inc.
@@ -27,7 +27,6 @@ import (
     `github.com/bytedance/sonic/internal/native/types`
     `github.com/bytedance/sonic/internal/rt`
     uq `github.com/bytedance/sonic/unquote`
-    `github.com/chenzhuoyu/base64x`
 )
 
 var typeByte = rt.UnpackEface(byte(0)).Type
@@ -77,14 +76,6 @@ func unquote(src string) (string, types.ParsingError) {
     return uq.String(src)
 }
 
-func decodeBase64(src string) ([]byte, error) {
-    return base64x.StdEncoding.DecodeString(src)
-}
-
-func encodeBase64(src []byte) string {
-    return base64x.StdEncoding.EncodeToString(src)
-}
-
 func (self *Parser) decodeValue() (val types.JsonState) {
     sv := (*rt.GoString)(unsafe.Pointer(&self.s))
     flag := types.F_USE_NUMBER

ast/api_compat.go

@@ -1,111 +1,105 @@
-// +build !amd64 !go1.16 go1.23
+// +build !amd64,!arm64 go1.23 !go1.16 arm64,!go1.20
 
 /*
- * Copyright 2022 ByteDance Inc.
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *     http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
+* Copyright 2022 ByteDance Inc.
+*
+* Licensed under the Apache License, Version 2.0 (the "License");
+* you may not use this file except in compliance with the License.
+* You may obtain a copy of the License at
+*
+*     http://www.apache.org/licenses/LICENSE-2.0
+*
+* Unless required by applicable law or agreed to in writing, software
+* distributed under the License is distributed on an "AS IS" BASIS,
+* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+* See the License for the specific language governing permissions and
+* limitations under the License.
+*/
 
 package ast
 
 import (
-    `encoding/base64`
-    `encoding/json`
+	`encoding/json`
 
-    `github.com/bytedance/sonic/internal/native/types`
-    `github.com/bytedance/sonic/internal/rt`
+	`github.com/bytedance/sonic/internal/native/types`
+	`github.com/bytedance/sonic/internal/rt`
 )
 
 func init() {
-    println("WARNING: sonic only supports Go1.16~1.22 && CPU amd64, but your environment is not suitable")
+	println("WARNING:(ast) sonic only supports Go1.16~1.22, but your environment is not suitable")
 }
 
 func quote(buf *[]byte, val string) {
-    quoteString(buf, val)
+	quoteString(buf, val)
 }
 
 func unquote(src string) (string, types.ParsingError) {
-    sp := rt.IndexChar(src, -1)
-    out, ok := unquoteBytes(rt.BytesFrom(sp, len(src)+2, len(src)+2))
-    if !ok {
-        return "", types.ERR_INVALID_ESCAPE
-    }
-    return rt.Mem2Str(out), 0
+	sp := rt.IndexChar(src, -1)
+	out, ok := unquoteBytes(rt.BytesFrom(sp, len(src)+2, len(src)+2))
+	if !ok {
+		return "", types.ERR_INVALID_ESCAPE
+	}
+	return rt.Mem2Str(out), 0
 }
 
-func decodeBase64(src string) ([]byte, error) {
-    return base64.StdEncoding.DecodeString(src)
-}
-
-func encodeBase64(src []byte) string {
-    return base64.StdEncoding.EncodeToString(src)
-}
 
 func (self *Parser) decodeValue() (val types.JsonState) {
-    e, v := decodeValue(self.s, self.p, self.dbuf == nil)
-    if e < 0 {
-        return v
-    }
-    self.p = e
-    return v
+	e, v := decodeValue(self.s, self.p, self.dbuf == nil)
+	if e < 0 {
+		return v
+	}
+	self.p = e
+	return v
 }
 
 func (self *Parser) skip() (int, types.ParsingError) {
-    e, s := skipValue(self.s, self.p)
-    if e < 0 {
-        return self.p, types.ParsingError(-e)
-    }
-    self.p = e
-    return s, 0
+	e, s := skipValue(self.s, self.p)
+	if e < 0 {
+		return self.p, types.ParsingError(-e)
+	}
+	self.p = e
+	return s, 0
 }
 
 func (self *Parser) skipFast() (int, types.ParsingError) {
-    e, s := skipValueFast(self.s, self.p)
-    if e < 0 {
-        return self.p, types.ParsingError(-e)
-    }
-    self.p = e
-    return s, 0
+	e, s := skipValueFast(self.s, self.p)
+	if e < 0 {
+		return self.p, types.ParsingError(-e)
+	}
+	self.p = e
+	return s, 0
 }
 
 func (self *Node) encodeInterface(buf *[]byte) error {
-    out, err := json.Marshal(self.packAny())
-    if err != nil {
-        return err
-    }
-    *buf = append(*buf, out...)
-    return nil
+	out, err := json.Marshal(self.packAny())
+	if err != nil {
+		return err
+	}
+	*buf = append(*buf, out...)
+	return nil
 }
 
 func (self *Parser) getByPath(path ...interface{}) (int, types.ParsingError) {
-    var err types.ParsingError
     for _, p := range path {
         if idx, ok := p.(int); ok && idx >= 0 {
-            if err = self.searchIndex(idx); err != 0 {
-                return -1, err
+            if err := self.searchIndex(idx); err != 0 {
+                return self.p, err
             }
         } else if key, ok := p.(string); ok {
-            if err = self.searchKey(key); err != 0 {
-                return -1, err
+            if err := self.searchKey(key); err != 0 {
+                return self.p, err
             }
         } else {
             panic("path must be either int(>=0) or string")
         }
     }
-
-    var start int
-    if start, err = self.skip(); err != 0 {
-        return -1, err
+    start, e := self.skip()
+    if e != 0 {
+        return self.p, e
     }
+    // t := switchRawType(self.s[start])
+    // if t == _V_NUMBER {
+    //     self.p = 1 + backward(self.s, self.p-1)
+    // }
     return start, 0
 }