微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

golang 反射 reflect包 struct相互填充

最近在用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)
}

 

 

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。

相关推荐