mirror of
https://github.com/astaxie/beego.git
synced 2024-11-26 04:51:30 +00:00
add structmap function for controller
This commit is contained in:
parent
4fc49b5803
commit
2dee30183d
68
router.go
68
router.go
@ -1,12 +1,14 @@
|
|||||||
package beego
|
package beego
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"net/http"
|
"net/http"
|
||||||
"net/url"
|
"net/url"
|
||||||
"reflect"
|
"reflect"
|
||||||
"regexp"
|
"regexp"
|
||||||
"runtime"
|
"runtime"
|
||||||
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -183,6 +185,69 @@ func (p *ControllerRegistor) FilterPrefixPath(path string, filter http.HandlerFu
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func structMap(vc reflect.Value, params *map[string]string) error {
|
||||||
|
for k, v := range *params {
|
||||||
|
names := strings.Split(k, ".")
|
||||||
|
var value reflect.Value = vc
|
||||||
|
for i, name := range names {
|
||||||
|
if i != len(names)-1 {
|
||||||
|
if value.Kind() != reflect.Struct {
|
||||||
|
return errors.New("arg error")
|
||||||
|
}
|
||||||
|
value := value.FieldByName(name)
|
||||||
|
if !value.IsValid() {
|
||||||
|
return errors.New("arg error")
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
tv := value.FieldByName(name)
|
||||||
|
fmt.Println(name, tv, tv.Kind())
|
||||||
|
if !tv.IsValid() {
|
||||||
|
return errors.New("arg error")
|
||||||
|
}
|
||||||
|
if !tv.CanSet() {
|
||||||
|
return errors.New("can not set " + name)
|
||||||
|
}
|
||||||
|
var l interface{}
|
||||||
|
switch k := tv.Kind(); k {
|
||||||
|
case reflect.String:
|
||||||
|
l = v
|
||||||
|
case reflect.Bool:
|
||||||
|
l = (v == "true")
|
||||||
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32:
|
||||||
|
x, err := strconv.Atoi(v)
|
||||||
|
if err != nil {
|
||||||
|
return errors.New("arg " + v + " as int: " + err.Error())
|
||||||
|
}
|
||||||
|
l = x
|
||||||
|
case reflect.Int64:
|
||||||
|
x, err := strconv.ParseInt(v, 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
return errors.New("arg " + v + " as int: " + err.Error())
|
||||||
|
}
|
||||||
|
l = x
|
||||||
|
case reflect.Float32, reflect.Float64:
|
||||||
|
x, err := strconv.ParseFloat(v, 64)
|
||||||
|
if err != nil {
|
||||||
|
return errors.New("arg " + v + " as float64: " + err.Error())
|
||||||
|
}
|
||||||
|
l = x
|
||||||
|
case reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
|
||||||
|
x, err := strconv.ParseUint(v, 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
return errors.New("arg " + v + " as int: " + err.Error())
|
||||||
|
}
|
||||||
|
l = x
|
||||||
|
case reflect.Struct:
|
||||||
|
fmt.Println("can not set an struct")
|
||||||
|
}
|
||||||
|
|
||||||
|
tv.Set(reflect.ValueOf(l))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
// AutoRoute
|
// AutoRoute
|
||||||
func (p *ControllerRegistor) ServeHTTP(rw http.ResponseWriter, r *http.Request) {
|
func (p *ControllerRegistor) ServeHTTP(rw http.ResponseWriter, r *http.Request) {
|
||||||
defer func() {
|
defer func() {
|
||||||
@ -351,6 +416,9 @@ func (p *ControllerRegistor) ServeHTTP(rw http.ResponseWriter, r *http.Request)
|
|||||||
init := vc.MethodByName("Init")
|
init := vc.MethodByName("Init")
|
||||||
in := make([]reflect.Value, 2)
|
in := make([]reflect.Value, 2)
|
||||||
ct := &Context{ResponseWriter: w, Request: r, Params: params}
|
ct := &Context{ResponseWriter: w, Request: r, Params: params}
|
||||||
|
|
||||||
|
structMap(vc.Elem(), ¶ms)
|
||||||
|
|
||||||
in[0] = reflect.ValueOf(ct)
|
in[0] = reflect.ValueOf(ct)
|
||||||
in[1] = reflect.ValueOf(runrouter.controllerType.Name())
|
in[1] = reflect.ValueOf(runrouter.controllerType.Name())
|
||||||
init.Call(in)
|
init.Call(in)
|
||||||
|
Loading…
Reference in New Issue
Block a user