golang之copier
今天我们要介绍的copier
库就能处理不同类型之间的赋值。除此之外,copier
还能:
- 调用同名方法为字段赋值;
- 以源对象字段为参数调用目标对象的方法,从而为目标对象赋值(当然也可以做其它的任何事情);
- 将切片赋值给切片(可以是不同类型哦);
- 将结构体追加到切片中。
快速使用
先安装:
$ go get github.com/jinzhu/copier
后使用:
package main import ( "fmt" "github.com/jinzhu/copier" ) type User struct { Name string Age int } type Employee struct { Name string Age int Role string } func main() { user := User{Name: "dj", Age: 18} employee := Employee{} copier.Copy(&employee, &user) fmt.Printf("%#v\n", employee) }
将user
对象中的字段赋值到employee
的同名字段中。如果目标对象中没有同名的字段,则该字段被忽略。
高级特性
方法赋值
目标对象中的一些字段,源对象中没有,但是源对象有同名的方法。这时Copy
会调用这个方法,将返回值赋值给目标对象中的字段:
type User struct { Name string Age int } func (u *User) DoubleAge() int { return 2 * u.Age } type Employee struct { Name string DoubleAge int Role string } func main() { user := User{Name: "dj", Age: 18} employee := Employee{} copier.Copy(&employee, &user) fmt.Printf("%#v\n", employee) }
我们给User
添加一个DoubleAge
方法。Employee
结构有字段DoubleAge
,User
中没有,但是User
有一个同名的方法,这时Copy
调用user
的DoubleAge
方法为employee
的DoubleAge
赋值,得到 36。
调用目标方法
有时候源对象中的某个字段没有出现在目标对象中,但是目标对象有一个同名的方法,方法接受一个同类型的参数,这时Copy
会以源对象的这个字段作为参数调用目标对象的该方法:
type User struct { Name string Age int Role string } type Employee struct { Name string Age int SuperRole string } func (e *Employee) Role(role string) { e.SuperRole = "Super" + role } func main() { user := User{Name: "dj", Age: 18, Role: "Admin"} employee := Employee{} copier.Copy(&employee, &user) fmt.Printf("%#v\n", employee) }
我们给Employee
添加了一个Role
方法,User
的字段Role
没有出现在Employee
中,但是Employee
有一个同名方法。Copy
函数内部会以user
对象的Role
字段为参数调用employee
的Role
方法。最终,我们的employee
对象的SuperRole
值变为SuperAdmin
。实际上,这个方法中可以执行任何操作,不一定是赋值。
切片赋值
使用一个切片来为另一个切片赋值。如果类型相同,那好办,直接append
就行。如果类型不同呢?copier
就派上大用场了:
type User struct { Name string Age int } type Employee struct { Name string Age int Role string } func main() { users := []User{ {Name: "dj", Age: 18}, {Name: "dj2", Age: 18}, } employees := []Employee{} copier.Copy(&employees, &users) fmt.Printf("%#v\n", employees) }
这个实际上就是将源切片中每个元素分别赋值到目标切片中。
将结构赋值到切片
这个不难,实际上就是根据源对象生成一个和目标切片类型相符合的对象,然后append
到目标切片中:
type User struct { Name string Age int } type Employee struct { Name string Age int Role string } func main() { user := User{Name: "dj", Age: 18} employees := []Employee{} copier.Copy(&employees, &user) fmt.Printf("%#v\n", employees) }
上面代码中,Copy
先通过user
生成一个Employee
对象,然后append
到切片employees
中。
汇总
最后将所有的特性汇总在一个例子中,其实就是Copier
的 GitHub 仓库首页的例子:
type User struct { Name string Age int Role string } func (u *User) DoubleAge() int { return u.Age * 2 } type Employee struct { Name string Age int SuperRole string } func (e *Employee) Role(role string) { e.SuperRole = "Super" + role } func main() { var ( user = User{Name: "dj", Age: 18} users = []User{ {Name: "dj", Age: 18, Role: "Admin"}, {Name: "dj2", Age: 18, Role: "Dev"}, } employee = Employee{} employees = []Employee{} ) copier.Copy(&employee, &user) fmt.Printf("%#v\n", employee) copier.Copy(&employees, &user) fmt.Printf("%#v\n", employees) // employees = []Employee{} copier.Copy(&employees, &users) fmt.Printf("%#v\n", employees) }
上面例子中,我故意把employees = []Employee{}
这一行注释掉,最后输出的employees
是 3 个元素,能更清楚的看出切片到切片是append
的,目标切片原来的元素还是保留的。