fix #316

context/context.go

@@ -1,9 +1,8 @@
 package context
 
 import (
-	"net/http"
-
 	"github.com/astaxie/beego/middleware"
+	"net/http"
 )
 
 type Context struct {

context/input.go

@@ -11,14 +11,16 @@ import (
 
 type BeegoInput struct {
 	CruSession  session.SessionStore
-	Param       map[string]string
+	Params      map[string]string
+	Data        map[interface{}]interface{}
 	req         *http.Request
 	RequestBody []byte
 }
 
 func NewInput(req *http.Request) *BeegoInput {
 	return &BeegoInput{
-		Param: make(map[string]string),
+		Params: make(map[string]string),
+		Data:   make(map[interface{}]interface{}),
 		req:    req,
 	}
 }
@@ -129,8 +131,8 @@ func (input *BeegoInput) UserAgent() string {
 	return input.Header("User-Agent")
 }
 
-func (input *BeegoInput) Params(key string) string {
+func (input *BeegoInput) Param(key string) string {
-	if v, ok := input.Param[key]; ok {
+	if v, ok := input.Params[key]; ok {
 		return v
 	}
 	return ""
@@ -164,3 +166,14 @@ func (input *BeegoInput) Body() []byte {
 	input.RequestBody = requestbody
 	return requestbody
 }
+
+func (input *BeegoInput) GetData(key interface{}) interface{} {
+	if v, ok := input.Data[key]; ok {
+		return v
+	}
+	return nil
+}
+
+func (input *BeegoInput) SetData(key, val interface{}) {
+	input.Data[key] = val
+}