fixed generation for alias ptr

humans-group · Mar 23, 2021 · 697a73d · 697a73d
1 parent 0cfd086
commit 697a73d
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 1 deletion.
diff --git a/field_generator.go b/field_generator.go
@@ -132,6 +132,9 @@ func (g *fieldGenerator) writePointerDef(ctx fieldGenContext) {
 	case astparser.TypeSimple:
 		g.writeSimpleDef(ctx.withFieldName("*"+ctx.fieldName))
 	case astparser.TypeCustom:
+		if tt.AliasType != nil {
+			ctx = ctx.withFieldName("*"+ctx.fieldName)
+		}
 		g.writeCustomDef(ctx)
 	default:
 		panic(fmt.Sprintf("unsupported array pointer innter type %T field name %s", tt, ctx.fieldName))

diff --git a/fixtures_test/custom.go b/fixtures_test/custom.go
@@ -43,6 +43,7 @@ type Dep struct {
 	ConstInt      ConstInt
 	ConstStrArray []ConstStr
 	ConstStrMap   map[string]ConstStr
+	ConstStrPtr   *ConstStr
 }
 
 type Optional struct {

diff --git a/fixtures_test/custom.zap.go b/fixtures_test/custom.zap.go
@@ -152,6 +152,10 @@ func (m *Dep) MarshalLogObject(enc zapcore.ObjectEncoder) error {
 		return nil
 	}))
 
+	keyName = "ConstStrPtr"
+	if m.ConstStrPtr != nil {
+		enc.AddString(keyName, string(*m.ConstStrPtr))
+	}
 	return nil
 }
 

diff --git a/generator_test.go b/generator_test.go
@@ -22,7 +22,7 @@ var _ = Describe("backoff", func() {
 	It("should generate golden files", func() {
 		cfg := astparser.Config{
 			InputDir:      "fixtures_test",
-			IncludeRegexp: "primitives.go|custom.go",
+			IncludeRegexp: "custom.go",
 		}
 		sources, err := astparser.Load(cfg)
 		Ω(err).ShouldNot(HaveOccurred())