2019-02-02 16:18:25 +08:00
|
|
|
// Copyright 2017 gf Author(https://github.com/gogf/gf). All Rights Reserved.
|
2018-11-19 21:11:43 +08:00
|
|
|
//
|
|
|
|
// This Source Code Form is subject to the terms of the MIT License.
|
|
|
|
// If a copy of the MIT was not distributed with this file,
|
2019-02-02 16:18:25 +08:00
|
|
|
// You can obtain one at https://github.com/gogf/gf.
|
2018-11-19 21:11:43 +08:00
|
|
|
|
|
|
|
package ghttp
|
|
|
|
|
2019-07-29 21:01:19 +08:00
|
|
|
import "github.com/gogf/gf/container/gvar"
|
2018-11-19 21:11:43 +08:00
|
|
|
|
2019-11-29 22:02:19 +08:00
|
|
|
// SetParam sets custom parameter with key-value pair.
|
2018-11-19 21:11:43 +08:00
|
|
|
func (r *Request) SetParam(key string, value interface{}) {
|
2019-12-01 14:07:36 +08:00
|
|
|
if r.paramsMap == nil {
|
|
|
|
r.paramsMap = make(map[string]interface{})
|
2019-06-19 09:06:52 +08:00
|
|
|
}
|
2019-12-01 14:07:36 +08:00
|
|
|
r.paramsMap[key] = value
|
2018-11-19 21:11:43 +08:00
|
|
|
}
|
|
|
|
|
2019-11-29 22:02:19 +08:00
|
|
|
// GetParam returns custom parameter with given name <key>.
|
|
|
|
// It returns <def> if <key> does not exist.
|
|
|
|
// It returns nil if <def> is not passed.
|
2019-09-18 23:20:45 +08:00
|
|
|
func (r *Request) GetParam(key string, def ...interface{}) interface{} {
|
2019-12-01 14:07:36 +08:00
|
|
|
if r.paramsMap != nil {
|
|
|
|
return r.paramsMap[key]
|
2019-06-19 09:06:52 +08:00
|
|
|
}
|
|
|
|
if len(def) > 0 {
|
2019-09-18 23:20:45 +08:00
|
|
|
return def[0]
|
2019-06-19 09:06:52 +08:00
|
|
|
}
|
2019-09-18 23:20:45 +08:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2019-11-29 22:02:19 +08:00
|
|
|
// GetParamVar returns custom parameter with given name <key> as *gvar.Var.
|
|
|
|
// It returns <def> if <key> does not exist.
|
|
|
|
// It returns nil if <def> is not passed.
|
2019-09-18 23:20:45 +08:00
|
|
|
func (r *Request) GetParamVar(key string, def ...interface{}) *gvar.Var {
|
|
|
|
return gvar.New(r.GetParam(key, def...))
|
2018-11-19 21:11:43 +08:00
|
|
|
}
|