feat: add NoJIT option (#67)

frugal.go

@@ -17,13 +17,14 @@
 package frugal
 
 import (
+	"github.com/cloudwego/frugal/internal/opts"
 	"github.com/cloudwego/frugal/internal/reflect"
 	"github.com/cloudwego/gopkg/protocol/thrift"
 )
 
 // EncodedSize measures the encoded size of val.
 func EncodedSize(val interface{}) int {
-	if nojit {
+	if !jit || opts.NoJIT {
 		return reflect.EncodedSize(val)
 	}
 	return jitEncodedSize(val)
@@ -32,15 +33,15 @@ func EncodedSize(val interface{}) int {
 // EncodeObject serializes val into buf with Thrift Binary Protocol, with optional Zero-Copy thrift.NocopyWriter.
 // buf must be large enough to contain the entire serialization result.
 func EncodeObject(buf []byte, w thrift.NocopyWriter, val interface{}) (int, error) {
-	if nojit {
+	if !jit || opts.NoJIT {
 		return reflect.Encode(buf, val) // TODO: impl thrift.NocopyWriter
 	}
 	return jitEncodeObject(buf, w, val)
 }
 
 // DecodeObject deserializes buf into val with Thrift Binary Protocol.
 func DecodeObject(buf []byte, val interface{}) (int, error) {
-	if nojit {
+	if !jit || opts.NoJIT {
 		return reflect.Decode(buf, val)
 	}
 	return jitDecodeObject(buf, val)

frugal_jit.go

@@ -19,9 +19,7 @@
 package frugal
 
 import (
-	"os"
 	"reflect"
-	"strconv"
 	"sync"
 
 	"github.com/cloudwego/frugal/internal/jit/decoder"
@@ -32,13 +30,7 @@ import (
 	"github.com/cloudwego/gopkg/protocol/thrift"
 )
 
-var nojit bool
-
-func init() {
-	if v, err := strconv.ParseBool(os.Getenv("FRUGAL_NO_JIT")); err == nil {
-		nojit = v
-	}
-}
+const jit = true
 
 func jitEncodedSize(val interface{}) int {
 	return encoder.EncodedSize(val)
@@ -74,7 +66,7 @@ func newty(ty *rt.GoType, d int) *_Ty {
 // Pretouch compiles vt ahead-of-time to avoid JIT compilation on-the-fly, in
 // order to reduce the first-hit latency.
 func Pretouch(vt reflect.Type, options ...Option) error {
-	if nojit {
+	if opts.NoJIT {
 		return nil
 	}
 	d := 0

frugal_jit_go124.go

@@ -21,10 +21,16 @@ package frugal
 import (
 	"reflect"
 
+	"github.com/cloudwego/frugal/internal/opts"
 	"github.com/cloudwego/gopkg/protocol/thrift"
 )
 
-const nojit = true
+const jit = false // not implemented for go1.24 || !amd64 || windows
+
+func init() {
+	// force set opts.NoJIT true coz jit not available
+	opts.NoJIT = true
+}
 
 func jitEncodedSize(val interface{}) int {
 	panic("not support JIT for Go version >= go1.24")

internal/opts/limits.go → internal/opts/defaults.go

@@ -29,6 +29,8 @@ const (
 var (
 	MaxInlineDepth  = parseOrDefault("FRUGAL_MAX_INLINE_DEPTH", _DefaultMaxInlineDepth, 1)
 	MaxInlineILSize = parseOrDefault("FRUGAL_MAX_INLINE_IL_SIZE", _DefaultMaxInlineILSize, 256)
+
+	NoJIT, _ = strconv.ParseBool(os.Getenv("FRUGAL_NO_JIT"))
 )
 
 func parseOrDefault(key string, def int, min int) int {

internal/opts/options.go

@@ -20,6 +20,8 @@ type Options struct {
 	MaxInlineDepth   int
 	MaxInlineILSize  int
 	MaxPretouchDepth int
+
+	NoJIT bool
 }
 
 func (self *Options) CanInline(sp int, pc int) bool {
@@ -35,5 +37,7 @@ func GetDefaultOptions() Options {
 		MaxInlineDepth:   MaxInlineDepth,
 		MaxInlineILSize:  MaxInlineILSize,
 		MaxPretouchDepth: 0,
+
+		NoJIT: NoJIT,
 	}
 }

options.go

@@ -22,13 +22,20 @@ import (
 	"github.com/cloudwego/frugal/internal/opts"
 )
 
+// Option is the property setter function for opts.Options.
+type Option func(*opts.Options)
+
+// NoJIT disables JIT encoder and decoder explicitly.
+//
+// This function will be deprecated along with the JIT implementation in the future.
+func NoJIT(v bool) {
+	opts.NoJIT = v
+}
+
 const (
 	_MinILSize = 1024
 )
 
-// Option is the property setter function for opts.Options.
-type Option func(*opts.Options)
-
 // WithMaxInlineDepth sets the maximum inlining depth for the JIT compiler.
 //
 // Increasing of this option makes the compiler inline more aggressively, which