mirror of
https://github.com/astaxie/beego.git
synced 2024-11-22 13:20:55 +00:00
Add ParseForm function
This commit is contained in:
parent
23deaedd39
commit
f552338822
80
utils.go
80
utils.go
@ -4,7 +4,10 @@ import (
|
|||||||
"fmt"
|
"fmt"
|
||||||
"github.com/russross/blackfriday"
|
"github.com/russross/blackfriday"
|
||||||
"html/template"
|
"html/template"
|
||||||
|
"net/url"
|
||||||
|
"reflect"
|
||||||
"regexp"
|
"regexp"
|
||||||
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
@ -179,3 +182,80 @@ func inSlice(v string, sl []string) bool {
|
|||||||
}
|
}
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// parse form values to struct via tag
|
||||||
|
func ParseForm(form url.Values, obj interface{}) error {
|
||||||
|
objT := reflect.TypeOf(obj)
|
||||||
|
objV := reflect.ValueOf(obj)
|
||||||
|
if !(objT.Kind() == reflect.Ptr && objT.Elem().Kind() == reflect.Struct) {
|
||||||
|
return fmt.Errorf("%v must be a struct pointer", obj)
|
||||||
|
}
|
||||||
|
objT = objT.Elem()
|
||||||
|
objV = objV.Elem()
|
||||||
|
|
||||||
|
for i := 0; i < objT.NumField(); i++ {
|
||||||
|
fieldV := objV.Field(i)
|
||||||
|
if !fieldV.CanSet() {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
fieldT := objT.Field(i)
|
||||||
|
tag := fieldT.Tag.Get("form")
|
||||||
|
if len(tag) == 0 {
|
||||||
|
tag = fieldT.Name
|
||||||
|
}
|
||||||
|
value := form.Get(tag)
|
||||||
|
if len(value) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
switch fieldT.Type.Kind() {
|
||||||
|
case reflect.Bool:
|
||||||
|
b, err := strconv.ParseBool(value)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
fieldV.SetBool(b)
|
||||||
|
case reflect.Int:
|
||||||
|
fallthrough
|
||||||
|
case reflect.Int8:
|
||||||
|
fallthrough
|
||||||
|
case reflect.Int16:
|
||||||
|
fallthrough
|
||||||
|
case reflect.Int32:
|
||||||
|
fallthrough
|
||||||
|
case reflect.Int64:
|
||||||
|
x, err := strconv.ParseInt(value, 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
fieldV.SetInt(x)
|
||||||
|
case reflect.Uint:
|
||||||
|
fallthrough
|
||||||
|
case reflect.Uint8:
|
||||||
|
fallthrough
|
||||||
|
case reflect.Uint16:
|
||||||
|
fallthrough
|
||||||
|
case reflect.Uint32:
|
||||||
|
fallthrough
|
||||||
|
case reflect.Uint64:
|
||||||
|
x, err := strconv.ParseUint(value, 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
fieldV.SetUint(x)
|
||||||
|
case reflect.Float32:
|
||||||
|
fallthrough
|
||||||
|
case reflect.Float64:
|
||||||
|
x, err := strconv.ParseFloat(value, 64)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
fieldV.SetFloat(x)
|
||||||
|
case reflect.Interface:
|
||||||
|
fieldV.Set(reflect.ValueOf(value))
|
||||||
|
case reflect.String:
|
||||||
|
fieldV.SetString(value)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
39
utils_test.go
Normal file
39
utils_test.go
Normal file
@ -0,0 +1,39 @@
|
|||||||
|
package beego
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/url"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestParseForm(t *testing.T) {
|
||||||
|
type user struct {
|
||||||
|
Id int
|
||||||
|
tag string `form:tag`
|
||||||
|
Name interface{} `form:"username"`
|
||||||
|
Age int `form:"age"`
|
||||||
|
Email string
|
||||||
|
}
|
||||||
|
|
||||||
|
u := user{}
|
||||||
|
form := url.Values{
|
||||||
|
"tag": []string{"no"},
|
||||||
|
"username": []string{"test"},
|
||||||
|
"age": []string{"40"},
|
||||||
|
"Email": []string{"test@gmail.com"},
|
||||||
|
}
|
||||||
|
if err := ParseForm(form, u); err == nil {
|
||||||
|
t.Fatal("nothing will be changed")
|
||||||
|
}
|
||||||
|
if err := ParseForm(form, &u); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if u.Name.(string) != "test" {
|
||||||
|
t.Error("should be equal")
|
||||||
|
}
|
||||||
|
if u.Age != 40 {
|
||||||
|
t.Error("should be equal")
|
||||||
|
}
|
||||||
|
if u.Email != "test@gmail.com" {
|
||||||
|
t.Error("should be equal")
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user