keep imports when scattering resolvers between files

99designs · Feb 4, 2020 · cf4a3eb · cf4a3eb
1 parent da7c1e4
commit cf4a3eb
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/plugin/resolvergen/resolver.go b/plugin/resolvergen/resolver.go
@@ -89,9 +89,7 @@ func (m *Plugin) generatePerSchema(data *codegen.Data) error {
 		if o.HasResolvers() {
 			fn := gqlToResolverName(data.Config.Resolver.Dir(), o.Position.Src.Name)
 			if files[fn] == nil {
-				files[fn] = &File{
-					imports: rewriter.ExistingImports(fn),
-				}
+				files[fn] = &File{}
 			}
 
 			rewriter.MarkStructCopied(templates.LcFirst(o.Name) + templates.UcFirst(data.Config.Resolver.Type))
@@ -120,6 +118,7 @@ func (m *Plugin) generatePerSchema(data *codegen.Data) error {
 	}
 
 	for filename, file := range files {
+		file.imports = rewriter.ExistingImports(filename)
 		file.RemainingSource = rewriter.RemainingSource(filename)
 	}