mirror of
https://github.com/astaxie/beego.git
synced 2024-12-23 02:20:49 +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"
|
||||
"github.com/russross/blackfriday"
|
||||
"html/template"
|
||||
"net/url"
|
||||
"reflect"
|
||||
"regexp"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
@ -179,3 +182,80 @@ func inSlice(v string, sl []string) bool {
|
||||
}
|
||||
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