mirror of
https://github.com/hanzoai/expr.git
synced 2026-09-24 16:58:58 +00:00
a60d994e25
* export the checker and ensure no methods or fields are exported * make Checker reusable * add PatchAndCheck method to Checker, and reuse it in visitors * avoid unnecessary map and consolidate structs * make all Nature methods use pointer-receiver * added cache for type and nature information * optimize type and nature checks * more aggressively cache some partial results and add cheap checks * cache reused values * reduce copying in Nature by breaking down optional fields * enforce use of cache * improve ergonomics of FromType and NatureOf * improve struct method fetching * make cache private to Nature * first iteration improving struct reflect caching * cleaned up struct fields data * improve caching of functions data * simplify checker code * simplify and remove dead code * add checker benchmarks * address PR comments and minor improvements * rename from SetCache to Bind * make the benchmarks more fair * fix benchmark option not being applied * move Cache to be argument instead of field * merge FuncData into Optional * rename Optional to TypeData * simplify struct logic * cache builtin numeric types --------- Co-authored-by: Anton Medvedev <[email protected]>
101 lines
2.1 KiB
Go
101 lines
2.1 KiB
Go
package conf
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
|
|
"github.com/expr-lang/expr/ast"
|
|
"github.com/expr-lang/expr/builtin"
|
|
"github.com/expr-lang/expr/checker/nature"
|
|
"github.com/expr-lang/expr/vm/runtime"
|
|
)
|
|
|
|
var (
|
|
// DefaultMemoryBudget represents default maximum allowed memory usage by the vm.VM.
|
|
DefaultMemoryBudget uint = 1e6
|
|
|
|
// DefaultMaxNodes represents default maximum allowed AST nodes by the compiler.
|
|
DefaultMaxNodes uint = 1e4
|
|
)
|
|
|
|
type FunctionsTable map[string]*builtin.Function
|
|
|
|
type Config struct {
|
|
EnvObject any
|
|
Env nature.Nature
|
|
Expect reflect.Kind
|
|
ExpectAny bool
|
|
Optimize bool
|
|
Strict bool
|
|
Profile bool
|
|
MaxNodes uint
|
|
ConstFns map[string]reflect.Value
|
|
Visitors []ast.Visitor
|
|
Functions FunctionsTable
|
|
Builtins FunctionsTable
|
|
Disabled map[string]bool // disabled builtins
|
|
NtCache nature.Cache
|
|
}
|
|
|
|
// CreateNew creates new config with default values.
|
|
func CreateNew() *Config {
|
|
c := &Config{
|
|
Optimize: true,
|
|
MaxNodes: DefaultMaxNodes,
|
|
ConstFns: make(map[string]reflect.Value),
|
|
Functions: make(map[string]*builtin.Function),
|
|
Builtins: make(map[string]*builtin.Function),
|
|
Disabled: make(map[string]bool),
|
|
}
|
|
for _, f := range builtin.Builtins {
|
|
c.Builtins[f.Name] = f
|
|
}
|
|
return c
|
|
}
|
|
|
|
// New creates new config with environment.
|
|
func New(env any) *Config {
|
|
c := CreateNew()
|
|
c.WithEnv(env)
|
|
return c
|
|
}
|
|
|
|
func (c *Config) WithEnv(env any) {
|
|
c.EnvObject = env
|
|
c.Env = EnvWithCache(&c.NtCache, env)
|
|
c.Strict = c.Env.Strict
|
|
}
|
|
|
|
func (c *Config) ConstExpr(name string) {
|
|
if c.EnvObject == nil {
|
|
panic("no environment is specified for ConstExpr()")
|
|
}
|
|
fn := reflect.ValueOf(runtime.Fetch(c.EnvObject, name))
|
|
if fn.Kind() != reflect.Func {
|
|
panic(fmt.Errorf("const expression %q must be a function", name))
|
|
}
|
|
c.ConstFns[name] = fn
|
|
}
|
|
|
|
type Checker interface {
|
|
Check()
|
|
}
|
|
|
|
func (c *Config) Check() {
|
|
for _, v := range c.Visitors {
|
|
if c, ok := v.(Checker); ok {
|
|
c.Check()
|
|
}
|
|
}
|
|
}
|
|
|
|
func (c *Config) IsOverridden(name string) bool {
|
|
if _, ok := c.Functions[name]; ok {
|
|
return true
|
|
}
|
|
if _, ok := c.Env.Get(&c.NtCache, name); ok {
|
|
return true
|
|
}
|
|
return false
|
|
}
|