feat: 字典类型增加 len() 方法

sealdice · Oct 21, 2023 · 5153493 · 5153493
1 parent d841825
commit 5153493
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/types_methods.go b/types_methods.go
@@ -137,6 +137,16 @@ func funcDictItems(ctx *Context, this *VMValue, params []*VMValue) *VMValue {
 	return VMValueNewArrayRaw(arr)
 }
 
+func funcDictLen(ctx *Context, this *VMValue, params []*VMValue) *VMValue {
+	d := this.MustReadDictData()
+	var size int64
+	d.Dict.Range(func(key string, value *VMValue) bool {
+		size++
+		return true
+	})
+	return VMValueNewInt(size)
+}
+
 var builtinProto = map[VMValueType]*VMDictValue{
 	VMTypeArray: VMValueMustNewDictWithArray(
 		VMValueNewStr("kh"), nnf(&ndf{"Array.kh", []string{"num"}, []*VMValue{VMValueNewInt(1)}, nil, funcArrayKeepHigh}),
@@ -154,6 +164,7 @@ var builtinProto = map[VMValueType]*VMDictValue{
 		VMValueNewStr("keys"), nnf(&ndf{"Dict.keys", []string{}, nil, nil, funcDictKeys}),
 		VMValueNewStr("values"), nnf(&ndf{"Dict.values", []string{}, nil, nil, funcDictValues}),
 		VMValueNewStr("items"), nnf(&ndf{"Dict.items", []string{}, nil, nil, funcDictItems}),
+		VMValueNewStr("len"), nnf(&ndf{"Dict.len", []string{}, nil, nil, funcDictLen}),
 	),
 }