fix (#12)

Author: guonaihong

deepcopy.go

@@ -87,6 +87,7 @@ func isArraySlice(v reflect.Value) bool {
 // 拷贝slice array
 func (d *deepCopy) cpySliceArray(dst, src reflect.Value, depth int) error {
 
+	// dst只能是slice和array类型
 	if !isArraySlice(dst) {
 		return nil
 	}
@@ -96,17 +97,11 @@ func (d *deepCopy) cpySliceArray(dst, src reflect.Value, depth int) error {
 		l = dst.Len()
 	}
 
+	// 被拷贝dst类型是slice, 长度是0, src的长度有值
 	if dst.Kind() == reflect.Slice && dst.Len() == 0 && src.Len() > 0 {
-		elemType := src.Type().Elem()
-		if dst.Type().Elem().Kind() != elemType.Kind() {
-			return nil
-		}
 
-		// MakeSlice的类型用reflect.SliceOf(src.Index(0).Type())
-		// 而不用src.Type()的原因如下
-		// src.Type()拿到的是类型可能是array和slice。
-		// 实际需要的是元素T的slice类型, 使用src.Type(), 这是错误的。
-		newDst := reflect.MakeSlice(reflect.SliceOf(elemType), l, l)
+		dstElemType := dst.Type().Elem()
+		newDst := reflect.MakeSlice(reflect.SliceOf(dstElemType), l, l)
 		dst.Set(newDst)
 	}
 
@@ -293,7 +288,7 @@ func (d *deepCopy) cpyPtr(dst, src reflect.Value, depth int) error {
 		if !dst.CanSet() {
 			return nil
 		}
-		p := reflect.New(src.Type().Elem())
+		p := reflect.New(dst.Type().Elem())
 		dst.Set(p)
 	}
 
@@ -354,5 +349,4 @@ func (d *deepCopy) deepCopy(dst, src reflect.Value, depth int) error {
 		return d.cpyDefault(dst, src, depth)
 	}
 
-	return nil
 }

deepcopy_fix_test.go

@@ -0,0 +1,61 @@
+package deepcopy
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+// src
+type TopicItemSrc struct {
+	ID         string `protobuf:"bytes,1,opt,name=ID,proto3" json:"ID,omitempty"`
+	Name       string `protobuf:"bytes,2,opt,name=Name,proto3" json:"Name,omitempty"`
+	CreateTime string `protobuf:"bytes,3,opt,name=CreateTime,proto3" json:"CreateTime,omitempty"`
+}
+
+// src
+type AllTopicResp struct {
+	TopicItem []*TopicItemSrc `protobuf:"bytes,1,rep,name=TopicItem,proto3" json:"TopicItem,omitempty"`
+	Total     int32           `protobuf:"varint,2,opt,name=Total,proto3" json:"Total,omitempty"` //总条数
+}
+
+// desc
+type AllTopicResp_AllTopicRespData struct {
+	TopicItem []*TopicItemDst `protobuf:"bytes,1,rep,name=TopicItem,proto3" json:"TopicItem,omitempty"`
+	Total     int32           `protobuf:"varint,2,opt,name=Total,proto3" json:"Total,omitempty"` //总条数
+}
+
+type TopicItemDst struct {
+	ID         string `protobuf:"bytes,1,opt,name=ID,proto3" json:"ID,omitempty"`
+	Name       string `protobuf:"bytes,2,opt,name=Name,proto3" json:"Name,omitempty"`
+	CreateTime string `protobuf:"bytes,3,opt,name=CreateTime,proto3" json:"CreateTime,omitempty"`
+}
+
+type AllTopicResp2 struct {
+	Code    int32                          `protobuf:"varint,1,opt,name=Code,proto3" json:"Code,omitempty"`
+	Message string                         `protobuf:"bytes,2,opt,name=Message,proto3" json:"Message,omitempty"`
+	Data    *AllTopicResp_AllTopicRespData `protobuf:"bytes,3,opt,name=Data,proto3" json:"Data,omitempty"`
+}
+
+func TestCopy(t *testing.T) {
+	resp := AllTopicResp{Total: 100, TopicItem: []*TopicItemSrc{
+		{ID: "111", Name: "111"},
+		{ID: "111", Name: "222"},
+	}}
+
+	rsp := AllTopicResp2{Data: &AllTopicResp_AllTopicRespData{}}
+	Copy(rsp.Data, resp).Do()
+
+	fmt.Printf("%#v\n", resp)
+	fmt.Printf("%#v\n", rsp.Data)
+	assert.NotNil(t, rsp.Data)
+	assert.Equal(t, rsp.Data.Total, resp.Total)
+	assert.Equal(t, len(rsp.Data.TopicItem), len(resp.TopicItem))
+	assert.Greater(t, len(rsp.Data.TopicItem), 0)
+	assert.Greater(t, len(resp.TopicItem), 0)
+	for i, v := range resp.TopicItem {
+		assert.Equal(t, v.ID, rsp.Data.TopicItem[i].ID)
+		assert.Equal(t, v.Name, rsp.Data.TopicItem[i].Name)
+	}
+}