1
0
mirror of https://github.com/astaxie/beego.git synced 2024-11-25 20:00:55 +00:00

Add ParseForm function

This commit is contained in:
miraclesu 2013-07-25 22:25:12 +08:00
parent 23deaedd39
commit f552338822
2 changed files with 119 additions and 0 deletions

View File

@ -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
View 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")
}
}