forked from revel/revel
-
Notifications
You must be signed in to change notification settings - Fork 0
/
invoker.go
50 lines (44 loc) · 1.43 KB
/
invoker.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
// Copyright (c) 2012-2016 The Revel Framework Authors, All rights reserved.
// Revel Framework source code and usage is governed by a MIT style
// license that can be found in the LICENSE file.
package revel
import (
"io"
"reflect"
)
var (
controllerPtrType = reflect.TypeOf(&Controller{})
websocketType = reflect.TypeOf((*ServerWebSocket)(nil)).Elem()
)
func ActionInvoker(c *Controller, _ []Filter) {
// Instantiate the method.
methodValue := reflect.ValueOf(c.AppController).MethodByName(c.MethodType.Name)
// Collect the values for the method's arguments.
var methodArgs []reflect.Value
for _, arg := range c.MethodType.Args {
// If they accept a websocket connection, treat that arg specially.
var boundArg reflect.Value
if arg.Type.Implements(websocketType) {
boundArg = reflect.ValueOf(c.Request.WebSocket)
} else {
boundArg = Bind(c.Params, arg.Name, arg.Type)
// #756 - If the argument is a closer, defer a Close call,
// so we don't risk on leaks.
if closer, ok := boundArg.Interface().(io.Closer); ok {
defer func() {
_ = closer.Close()
}()
}
}
methodArgs = append(methodArgs, boundArg)
}
var resultValue reflect.Value
if methodValue.Type().IsVariadic() {
resultValue = methodValue.CallSlice(methodArgs)[0]
} else {
resultValue = methodValue.Call(methodArgs)[0]
}
if resultValue.Kind() == reflect.Interface && !resultValue.IsNil() {
c.Result = resultValue.Interface().(Result)
}
}