fix(codegen): GoDefine 支持指针等类型

Author: caixw

codegen/object.go

@@ -28,17 +28,13 @@ func GoDefine(t reflect.Type, m map[reflect.Type]string, unexported bool) string
 	goDefine(buf, 0, t, m, unexported, false)
 	s := buf.String()
 
-	if len(s) > 0 && s[0] == '{' { // 结构可能由于 m 的关系返回一个非结构体的类型定义，所以只能由开头是否为 { 判断是否为结构体。
-		return "type " + t.Name() + " struct " + s
+	if strings.HasPrefix(s, "struct {") { // 结构可能由于 m 的关系返回一个非结构体的类型定义，所以只能由开头是否为 struct { 判断是否为结构体。
+		return "type " + t.Name() + " " + s
 	}
 	return buf.String()
 }
 
 func goDefine(buf *errwrap.Buffer, indent int, t reflect.Type, m map[reflect.Type]string, unexported, anonymous bool) {
-	for t.Kind() == reflect.Pointer {
-		t = t.Elem()
-	}
-
 	if len(m) > 0 {
 		if s, found := m[t]; found {
 			buf.WString(s)
@@ -48,21 +44,35 @@ func goDefine(buf *errwrap.Buffer, indent int, t reflect.Type, m map[reflect.Typ
 
 	switch t.Kind() {
 	case reflect.Func, reflect.Chan: // 忽略
+	case reflect.Pointer:
+		buf.WByte('*')
+		goDefine(buf, indent, t.Elem(), m, unexported, anonymous)
 	case reflect.Slice:
-		buf.WString("[]").WString(t.Elem().Name())
+		buf.WString("[]")
+		goDefine(buf, indent, t.Elem(), m, unexported, anonymous)
 	case reflect.Array:
-		buf.WByte('[').WString(strconv.Itoa(t.Len())).WByte(']').WString(t.Elem().Name())
+		buf.WByte('[').WString(strconv.Itoa(t.Len())).WByte(']')
+		goDefine(buf, indent, t.Elem(), m, unexported, anonymous)
 	case reflect.Struct:
 		if !anonymous {
-			buf.WString("{\n")
+			if t.NumField() == 0 {
+				buf.WString("struct {}")
+				return
+			}
+
+			buf.WString("struct {\n")
 			indent++
 		}
 
 		for i := 0; i < t.NumField(); i++ {
 			f := t.Field(i)
 
 			if f.Anonymous {
-				goDefine(buf, indent, f.Type, m, unexported, true)
+				tt := f.Type
+				for tt.Kind() == reflect.Pointer { // 匿名字段需要去掉指针类型
+					tt = tt.Elem()
+				}
+				goDefine(buf, indent, tt, m, unexported, true)
 				continue
 			}
 

codegen/object_test.go

@@ -40,13 +40,18 @@ type object5 struct {
 	Str string
 }
 
+type object6 struct {
+	XMLName struct{} `json:"root"`
+	Str     []*object3
+}
+
 func TestGoDefine(t *testing.T) {
 	a := assert.New(t, false)
 
 	wont := "type object1 struct {\n\tInt\tint\t`json:\"int\" yaml:\"int\"`\n\tArray\t[5]int\n\tSlice\t[]string\n\tByte\tuint8\n}"
 	a.Equal(GoDefine(reflect.TypeFor[object1](), nil, false), wont)
 
-	wont = "type object2 struct {\n\tInt\tint\t`json:\"int\" yaml:\"int\"`\n\tObject\t{\n\t\tInt\tint\t`json:\"int\" yaml:\"int\"`\n\t\tArray\t[5]int\n\t\tSlice\t[]string\n\t\tByte\tuint8\n\t}\t`json:\"object\"`\n}"
+	wont = "type object2 struct {\n\tInt\tint\t`json:\"int\" yaml:\"int\"`\n\tObject\t*struct {\n\t\tInt\tint\t`json:\"int\" yaml:\"int\"`\n\t\tArray\t[5]int\n\t\tSlice\t[]string\n\t\tByte\tuint8\n\t}\t`json:\"object\"`\n}"
 	a.Equal(GoDefine(reflect.TypeFor[*object2](), nil, false), wont)
 
 	a.Equal(GoDefine(reflect.TypeFor[int](), nil, false), "int")
@@ -71,4 +76,6 @@ func TestGoDefine(t *testing.T) {
 
 	a.Equal(GoDefine(reflect.TypeFor[time.Time](), m, false), "string")
 	a.Equal(GoDefine(reflect.TypeFor[time.Time](), nil, false), "type Time struct {\n}")
+
+	a.Equal(GoDefine(reflect.TypeFor[object6](), m, false), "type object6 struct {\n\tXMLName\tstruct {}\t`json:\"root\"`\n\tStr\t[]*struct {\n\t\tT\tstring\t`json:\"t\"`\n\t}\n}")
 }