save for 调整 非全局方式使用

Author: glennliao

config/functions/functions.go

@@ -19,6 +19,15 @@ func Reg(name string, f Func) {
 	funcMap[name] = f
 }
 
+func RegHandler(name string, handler func(ctx context.Context, param g.Map) (res any, err error)) {
+	if _, exists := funcMap[name]; exists {
+		panic(fmt.Errorf(" function %s has exists", name))
+	}
+	funcMap[name] = struct {
+		Handler func(ctx context.Context, param g.Map) (res any, err error)
+	}{Handler: handler}
+}
+
 func Call(ctx context.Context, name string, param g.Map) (any, error) {
 	return funcMap[name].Handler(ctx, param)
 }

query/node.go

@@ -389,8 +389,9 @@ func (n *Node) parse() {
 		}
 
 	case NodeTypeFunc:
-		functionName, _ := util.ParseFunctionsStr(n.simpleReqVal)
-		n.simpleReqVal = functionName
+		// todo
+		//functionName, _ := util.ParseFunctionsStr(n.simpleReqVal)
+		//n.simpleReqVal = functionName
 
 	}
 	if n.queryContext.PrintProcessLog {
@@ -572,8 +573,15 @@ func (n *Node) fetch() {
 			n.total = n.children[n.primaryTableKey].total
 		}
 	case NodeTypeFunc:
+		functionName, paramKeys := util.ParseFunctionsStr(n.simpleReqVal)
+		//n.simpleReqVal = functionName
+		// todo 如何传递参数
 		param := model.Map{}
-		n.ret, n.err = functions.Call(n.ctx, n.simpleReqVal, param)
+		for _, key := range paramKeys {
+			param[key] = n.queryContext.pathNodes[key].simpleReqVal
+		}
+
+		n.ret, n.err = functions.Call(n.ctx, functionName, param)
 	}
 
 }