Add API Framework Revel Source Files

[iec.git] / src / foundation / api / revel / invoker.go

diff --git a/src/foundation/api/revel/invoker.go b/src/foundation/api/revel/invoker.go
new file mode 100644 (file)
index 0000000..24d40fa
--- /dev/null
+++ b/src/foundation/api/revel/invoker.go
@@ -0,0 +1,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)
+       }
+}