writer: handle imports for slice params

enosi-rl · Apr 17, 2022 · da24f99 · da24f99
1 parent ec925f3
commit da24f99
Show file tree

Hide file tree

Showing 5 changed files with 165 additions and 0 deletions.
diff --git a/cmd/gtrace/writer.go b/cmd/gtrace/writer.go
@@ -173,6 +173,9 @@ func (w *Writer) typeImports(dst []dep, t types.Type) []dep {
 	if p, ok := t.(*types.Pointer); ok {
 		return w.typeImports(dst, p.Elem())
 	}
+	if p, ok := t.(*types.Slice); ok {
+		return w.typeImports(dst, p.Elem())
+	}
 	n, ok := t.(*types.Named)
 	if !ok {
 		return dst

diff --git a/test/test_imports_pointer.go b/test/test_imports_pointer.go
@@ -0,0 +1,13 @@
+package test
+
+import "github.com/gobwas/gtrace/test/internal"
+
+//go:generate gtrace -v
+
+//gtrace:gen
+//gtrace:set shortcut
+// NOTE: must compile without unused imports error.
+type TracePointer struct {
+	OnSomethingA func(Type)
+	OnSomethingB func(*internal.Type)
+}
diff --git a/test/test_imports_pointer_gtrace.go b/test/test_imports_pointer_gtrace.go
diff --git a/test/test_imports_slice.go b/test/test_imports_slice.go
@@ -0,0 +1,13 @@
+package test
+
+import "github.com/gobwas/gtrace/test/internal"
+
+//go:generate gtrace -v
+
+//gtrace:gen
+//gtrace:set shortcut
+// NOTE: must compile without unused imports error.
+type TraceArray struct {
+	OnSomethingA func(Type)
+	OnSomethingB func([]internal.Type)
+}
diff --git a/test/test_imports_slice_gtrace.go b/test/test_imports_slice_gtrace.go