向繁琐的赋值代码说不。deepcopy.Copy 深度拷贝来了

出发点

一次帮同事 review 代码，想在 go 里面找一个支持深度 Copy 的库，github 上少得可怜。最后找到 json marshal 加 unmarshal 的方式，但是这种方式有两个缺点，第 1 marshal 一次 reflect，unmarshal 一次 reflect，有两次 reflect 的过程，效率会垫底。第 2，不支持过滤条件，这点硬伤，改不了。特对这两点问题，所以想撸个改进版本(更快，更可控)。

项目地址

https://github.com/antlabs/deepcopy

作用

deepcopy.Copy 主要用于两个类型间的深度拷贝[从零实现]

feature

• 支持异构结构体拷贝, dst 和 src 可以是不同的类型，会拷贝 dst 和 src 交集的部分
• 多类型支持 struct/map/slice/array/int...int64/uint...uint64/ 等等
• 性能相比 json 序列化和反序列化的做法，拥有更快的执行速度
• 可以控制拷贝结构体层次
• 可以通过 tag 控制感兴趣的字段

内容

• Installation
• Quick start
• example
  • 1. 控制拷贝结构体最多深度
  • 2. 只拷贝设置 tag 的结构体成员
  • 3.拷贝 slice
  • 4.拷贝 map

Installation

go get github.com/antlabs/deepcopy 

Quick start

package main import ( "fmt" "github.com/antlabs/deepcopy" ) type dst struct { ID int Result string } type src struct{ ID int Text string } func main() { d, s := dst{}, src{ID:3} deepcopy.Copy(&d, &s).Do() fmt.Printf("%#v\n", d) } 

max copy depth

如果 src 的结构体嵌套了两套，MaxDepth 可以控制只拷贝一层

deepcopy.Copy(&dst{}, &src{}).MaxDepth(1).Do() 

copy only the specified tag

只拷贝结构体里面有 copy tag 的字段，比如下面只会拷贝 ID 成员

package main import ( "fmt" "github.com/antlabs/deepcopy" ) type dst struct { ID int `copy:"ID"` Result string } type src struct { ID int `copy:"ID"` Result string } func main() { d := dst{} s := src{ID: 3, Result: "use tag"} deepcopy.Copy(&d, &s).RegisterTagName("copy").Do() fmt.Printf("%#v\n", d) } 

copy slice

package main import ( "fmt" "github.com/antlabs/deepcopy" ) func main() { i := []int{1, 2, 3, 4, 5, 6} var o []int deepcopy.Copy(&o, &i).Do() fmt.Printf("%#v\n", o) } 

copy map

package main import ( "fmt" "github.com/antlabs/deepcopy" ) func main() { i := map[string]int{ "cat": 100, "head": 10, "tr": 3, "tail": 44, } var o map[string]int deepcopy.Copy(&o, &i).Do() fmt.Printf("%#v\n", o) } 

性能

从零实现的 deepcopy 相比 json 序列化与反序列化方式拥有更好的性能

goos: linux goarch: amd64 pkg: github.com/antlabs/deepcopy Benchmark_MiniCopy-12 243212 4987 ns/op Benchmark_DeepCopy-12 273775 4781 ns/op PASS ok github.com/antlabs/deepcopy 4.496s