-
Notifications
You must be signed in to change notification settings - Fork 0
/
callable.go
73 lines (59 loc) · 1.35 KB
/
callable.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
68
69
70
71
72
73
package container
import (
"errors"
"github.com/goal-web/contracts"
"reflect"
)
var (
FuncTypeError = errors.New("参数必须是一个函数")
)
type magicalFunc struct {
in int
out int
value reflect.Value
arguments []reflect.Type
returns []reflect.Type
}
func NewMagicalFunc(fn interface{}) contracts.MagicalFunc {
var (
argValue = reflect.ValueOf(fn)
argType = reflect.TypeOf(fn)
)
if argValue.Kind() != reflect.Func {
panic(FuncTypeError)
}
var (
arguments = make([]reflect.Type, 0)
returns = make([]reflect.Type, 0)
argumentsLen = argType.NumIn()
returnsLen = argType.NumOut()
)
for argIndex := 0; argIndex < argumentsLen; argIndex++ {
arguments = append(arguments, argType.In(argIndex))
}
for outIndex := 0; outIndex < returnsLen; outIndex++ {
returns = append(returns, argType.Out(outIndex))
}
return &magicalFunc{
in: argumentsLen,
out: returnsLen,
value: argValue,
arguments: arguments,
returns: returns,
}
}
func (fn *magicalFunc) Call(in []reflect.Value) []reflect.Value {
return fn.value.Call(in)
}
func (fn *magicalFunc) Arguments() []reflect.Type {
return fn.arguments
}
func (fn *magicalFunc) Returns() []reflect.Type {
return fn.returns
}
func (fn *magicalFunc) NumOut() int {
return fn.out
}
func (fn *magicalFunc) NumIn() int {
return fn.in
}