forked from MontFerret/ferret
-
Notifications
You must be signed in to change notification settings - Fork 0
/
func_call.go
67 lines (53 loc) · 1.29 KB
/
func_call.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
package expressions
import (
"context"
"github.com/MontFerret/ferret/pkg/runtime/core"
"github.com/MontFerret/ferret/pkg/runtime/values"
)
type FunctionCallExpression struct {
src core.SourceMap
fun core.Function
args []core.Expression
}
func NewFunctionCallExpression(
src core.SourceMap,
fun core.Function,
args ...core.Expression,
) (*FunctionCallExpression, error) {
if fun == nil {
return nil, core.Error(core.ErrMissedArgument, "function")
}
return &FunctionCallExpression{src, fun, args}, nil
}
func (e *FunctionCallExpression) Arguments() []core.Expression {
return e.args
}
func (e *FunctionCallExpression) Function() core.Function {
return e.fun
}
func (e *FunctionCallExpression) Exec(ctx context.Context, scope *core.Scope) (core.Value, error) {
select {
case <-ctx.Done():
return values.None, core.ErrTerminated
default:
var out core.Value
var err error
if len(e.args) == 0 {
out, err = e.fun(ctx)
} else {
args := make([]core.Value, len(e.args))
for idx, arg := range e.args {
out, err := arg.Exec(ctx, scope)
if err != nil {
return values.None, core.SourceError(e.src, err)
}
args[idx] = out
}
out, err = e.fun(ctx, args...)
}
if err != nil {
return values.None, core.SourceError(e.src, err)
}
return out, nil
}
}