entgql/template: improve goimports time by adding missing imports (en…

…t#584)
prgres · Jun 17, 2024 · 2d81d47 · 2d81d47
1 parent d262910
commit 2d81d47
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 0 deletions.
diff --git a/entgql/template/collection.tmpl b/entgql/template/collection.tmpl
@@ -14,6 +14,7 @@ in the LICENSE file in the root directory of this source tree.
 {{ $gqlNodes := filterNodes $.Nodes (skipMode "type") }}
 
 import (
+	"entgo.io/contrib/entgql"
 	"github.com/99designs/gqlgen/graphql"
 	{{- range $n := $gqlNodes }}
 		"{{ $.Config.Package }}/{{ $n.Package }}"

diff --git a/entgql/template/edge.tmpl b/entgql/template/edge.tmpl
@@ -9,9 +9,12 @@ in the LICENSE file in the root directory of this source tree.
 {{ define "gql_edge" }}
 {{ template "header" $ }}
 
+{{ template "import" $ }}
+
 import (
 	"context"
 
+	"entgo.io/contrib/entgql"
 	"github.com/99designs/gqlgen/graphql"
 )
 

diff --git a/entgql/template/mutation_input.tmpl b/entgql/template/mutation_input.tmpl
@@ -11,6 +11,7 @@
 
 {{ $gqlNodes := filterNodes $.Nodes (skipMode "mutation_create_input" "mutation_update_input") }}
 import (
+    "entgo.io/contrib/entgql"
     {{- range $n := $gqlNodes }}
         {{- template "import/types" $n }}
         "{{ $.Config.Package }}/{{ $n.Package }}"

diff --git a/entgql/template/pagination.tmpl b/entgql/template/pagination.tmpl
@@ -30,6 +30,7 @@ import (
 	{{- range $n := $gqlNodes }}
 		"{{ $.Config.Package }}/{{ $n.Package }}"
 	{{- end }}
+	"{{ $.Config.Package }}/predicate"
 
 	"entgo.io/ent/dialect/sql"
 	"entgo.io/contrib/entgql"