fix go-gorm#1032: add import pkg paths to single query files

ch3nnn · Jan 16, 2024 · d8e7ed2 · d8e7ed2
1 parent 16a35f1
commit d8e7ed2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/generator.go b/generator.go
@@ -389,7 +389,7 @@ func (g *Generator) generateSingleQueryFile(data *genInfo) (err error) {
 	}
 	err = render(tmpl.Header, &buf, map[string]interface{}{
 		"Package":        g.queryPkgName,
-		"ImportPkgPaths": importList.Add(structPkgPath).Add(getImportPkgPaths(data)...).Paths(),
+		"ImportPkgPaths": importList.Add(g.importPkgPaths...).Add(structPkgPath).Add(getImportPkgPaths(data)...).Paths(),
 	})
 	if err != nil {
 		return err