mirror of https://github.com/astaxie/beego.git
68 lines
1.3 KiB
Go
68 lines
1.3 KiB
Go
package param
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
//Keeps param information to be auto passed to controller methods
|
|
type MethodParam struct {
|
|
name string
|
|
location paramLocation
|
|
required bool
|
|
defValue string
|
|
}
|
|
|
|
type paramLocation byte
|
|
|
|
const (
|
|
param paramLocation = iota
|
|
path
|
|
body
|
|
header
|
|
)
|
|
|
|
func New(name string, opts ...MethodParamOption) *MethodParam {
|
|
return newParam(name, nil, opts)
|
|
}
|
|
|
|
func newParam(name string, parser paramParser, opts []MethodParamOption) (param *MethodParam) {
|
|
param = &MethodParam{name: name}
|
|
for _, option := range opts {
|
|
option(param)
|
|
}
|
|
return
|
|
}
|
|
|
|
func Make(list ...*MethodParam) []*MethodParam {
|
|
if len(list) > 0 {
|
|
return list
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (mp *MethodParam) String() string {
|
|
options := []string{}
|
|
result := "param.New(\"" + mp.name + "\""
|
|
if mp.required {
|
|
options = append(options, "param.IsRequired")
|
|
}
|
|
switch mp.location {
|
|
case path:
|
|
options = append(options, "param.InPath")
|
|
case body:
|
|
options = append(options, "param.InBody")
|
|
case header:
|
|
options = append(options, "param.InHeader")
|
|
}
|
|
if mp.defValue != "" {
|
|
options = append(options, fmt.Sprintf(`param.Default("%s")`, mp.defValue))
|
|
}
|
|
if len(options) > 0 {
|
|
result += ", "
|
|
}
|
|
result += strings.Join(options, ", ")
|
|
result += ")"
|
|
return result
|
|
}
|