refactor router

context/input.go

@@ -33,6 +33,7 @@ var (
 	acceptsHTMLRegex = regexp.MustCompile(`(text/html|application/xhtml\+xml)(?:,|$)`)
 	acceptsXMLRegex  = regexp.MustCompile(`(application/xml|text/xml)(?:,|$)`)
 	acceptsJSONRegex = regexp.MustCompile(`(application/json)(?:,|$)`)
+	maxParam         = 50
 )
 
 // BeegoInput operates the http request header, data, cookie and body.
@@ -40,16 +41,17 @@ var (
 type BeegoInput struct {
 	Context     *Context
 	CruSession  session.Store
-	Params      map[string]string
-	Data        map[interface{}]interface{} // store some values in this context when calling context in filter or controller.
+	pnames      []string
+	pvalues     []string
+	data        map[interface{}]interface{} // store some values in this context when calling context in filter or controller.
 	RequestBody []byte
 }
 
 // NewInput return BeegoInput generated by Context.
 func NewInput() *BeegoInput {
 	return &BeegoInput{
-		Params: make(map[string]string),
-		Data:   make(map[interface{}]interface{}),
+		pvalues: make([]string, maxParam),
+		data:    make(map[interface{}]interface{}),
 	}
 }
 
@@ -57,8 +59,8 @@ func NewInput() *BeegoInput {
 func (input *BeegoInput) Reset(ctx *Context) {
 	input.Context = ctx
 	input.CruSession = nil
-	input.Params = make(map[string]string)
-	input.Data = make(map[interface{}]interface{})
+	input.pnames = input.pnames[:0]
+	input.data = nil
 	input.RequestBody = []byte{}
 }
 
@@ -254,14 +256,27 @@ func (input *BeegoInput) UserAgent() string {
 	return input.Header("User-Agent")
 }
 
+// ParamsLen return the length of the params
+func (input *BeegoInput) ParamsLen() int {
+	return len(input.pnames)
+}
+
 // Param returns router param by a given key.
 func (input *BeegoInput) Param(key string) string {
-	if v, ok := input.Params[key]; ok {
-		return v
+	for i, v := range input.pnames {
+		if v == key {
+			return input.pvalues[i]
+		}
 	}
 	return ""
 }
 
+// SetParam will set the param with key and value
+func (input *BeegoInput) SetParam(key, val string) {
+	input.pvalues[len(input.pnames)] = val
+	input.pnames = append(input.pnames, key)
+}
+
 // Query returns input data item string by a given string.
 func (input *BeegoInput) Query(key string) string {
 	if val := input.Param(key); val != "" {
@@ -305,9 +320,17 @@ func (input *BeegoInput) CopyBody() []byte {
 	return requestbody
 }
 
+// Data return the implicit data in the input
+func (input *BeegoInput) Data() map[interface{}]interface{} {
+	if input.data == nil {
+		input.data = make(map[interface{}]interface{})
+	}
+	return input.data
+}
+
 // GetData returns the stored data in this context.
 func (input *BeegoInput) GetData(key interface{}) interface{} {
-	if v, ok := input.Data[key]; ok {
+	if v, ok := input.data[key]; ok {
 		return v
 	}
 	return nil
@@ -316,7 +339,10 @@ func (input *BeegoInput) GetData(key interface{}) interface{} {
 // SetData stores data with given key in this context.
 // This data are only available in this context.
 func (input *BeegoInput) SetData(key, val interface{}) {
-	input.Data[key] = val
+	if input.data == nil {
+		input.data = make(map[interface{}]interface{})
+	}
+	input.data[key] = val
 }
 
 // ParseFormOrMulitForm parseForm or parseMultiForm based on Content-type