Skip to content

Commit

Permalink
Don't set the package variable for the new Resolver Template (#2725)
Browse files Browse the repository at this point in the history
  • Loading branch information
penard-cb committed Jul 24, 2023
1 parent febf956 commit abe3ffd
Showing 1 changed file with 6 additions and 5 deletions.
11 changes: 6 additions & 5 deletions plugin/resolvergen/resolver.go
Expand Up @@ -81,8 +81,9 @@ func (m *Plugin) generateSingleFile(data *codegen.Data) error {
OmitTemplateComment: data.Config.Resolver.OmitTemplateComment,
}

newResolverTemplate := resolverTemplate
if data.Config.Resolver.ResolverTemplate != "" {
resolverTemplate = readResolverTemplate(data.Config.Resolver.ResolverTemplate)
newResolverTemplate = readResolverTemplate(data.Config.Resolver.ResolverTemplate)
}

return templates.Render(templates.Options{
Expand All @@ -91,7 +92,7 @@ func (m *Plugin) generateSingleFile(data *codegen.Data) error {
Filename: data.Config.Resolver.Filename,
Data: resolverBuild,
Packages: data.Config.Packages,
Template: resolverTemplate,
Template: newResolverTemplate,
})
}

Expand Down Expand Up @@ -162,9 +163,9 @@ func (m *Plugin) generatePerSchema(data *codegen.Data) error {
file.imports = rewriter.ExistingImports(filename)
file.RemainingSource = rewriter.RemainingSource(filename)
}

newResolverTemplate := resolverTemplate
if data.Config.Resolver.ResolverTemplate != "" {
resolverTemplate = readResolverTemplate(data.Config.Resolver.ResolverTemplate)
newResolverTemplate = readResolverTemplate(data.Config.Resolver.ResolverTemplate)
}

for filename, file := range files {
Expand Down Expand Up @@ -194,7 +195,7 @@ func (m *Plugin) generatePerSchema(data *codegen.Data) error {
Filename: filename,
Data: resolverBuild,
Packages: data.Config.Packages,
Template: resolverTemplate,
Template: newResolverTemplate,
})
if err != nil {
return err
Expand Down

0 comments on commit abe3ffd

Please sign in to comment.