最近在用Golang写一些简单的业务,经常需要在两类相似的结构体之间相互填充数据,如果struct只有少数几个field,相互赋值还好;struct有很多field的,实例代码如下:
type Request struct {
Field1 string
Field2 string
Field3 string
}
type ORM struct {
OtherField string
Field1 string
Field2 string
Field3 string
}
// src和dst必须是struct,且dst必须是point
func Fill(src interface{}, dst interface{}) error {
srcType := reflect.TypeOf(src)
srcValue := reflect.ValueOf(src)
dstValue := reflect.ValueOf(dst)
if srcType.Kind() != reflect.Struct {
return errors.New("src必须是struct")
}
if dstValue.Kind() != reflect.Ptr {
return errors.New("dst必须是point")
}
for i := 0; i < srcType.NumField(); i++ {
dstField := dstValue.Elem().FieldByName(srcType.Field(i).Name)
if dstField.CanSet() {
dstField.Set(srcValue.Field(i))
}
}
return nil
}
func main() {
request := Request{
Field1: "1",
Field2: "2",
Field3: "3",
}
orm := ORM{}
req := Request{}
if err := Fill(request, &orm); err != nil {
log.Fatal(err)
}
log.Printf("%#v", orm)
if err := Fill(orm, &req); err != nil {
log.Fatal(err)
}
log.Printf("%#v", req)
}