Skip to content

Commit 535fc28

Browse files
committed
add github.com/jackc/pgtype import if there are enums using pgx/v4
1 parent 82b50aa commit 535fc28

File tree

14 files changed

+31
-0
lines changed

14 files changed

+31
-0
lines changed

internal/codegen/golang/imports.go

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -262,6 +262,11 @@ func (i *importer) modelImports() fileImports {
262262
if len(i.Enums) > 0 {
263263
std["fmt"] = struct{}{}
264264
std["database/sql/driver"] = struct{}{}
265+
266+
sqlpkg := parseDriver(i.Settings.Go.SqlPackage)
267+
if sqlpkg.Package() == "pgx/v4" {
268+
pkg[ImportSpec{Path: "github.com/jackc/pgtype"}] = struct{}{}
269+
}
265270
}
266271

267272
return sortedImports(std, pkg)

internal/endtoend/testdata/comment_on/postgresql/pgx/v4/go/models.go

Lines changed: 2 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

internal/endtoend/testdata/ddl_alter_type_add_value/postgresql/pgx/v4/go/models.go

Lines changed: 2 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

internal/endtoend/testdata/ddl_alter_type_rename/postgresql/pgx/v4/go/models.go

Lines changed: 2 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

internal/endtoend/testdata/ddl_alter_type_rename_and_update_columns/postgresql/pgx/v4/go/models.go

Lines changed: 2 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

internal/endtoend/testdata/ddl_alter_type_rename_value/postgresql/pgx/v4/go/models.go

Lines changed: 2 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

internal/endtoend/testdata/ddl_alter_type_set_schema/postgresql/pgx/v4/go/models.go

Lines changed: 2 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

internal/endtoend/testdata/ddl_comment/postgresql/pgx/v4/go/models.go

Lines changed: 2 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

internal/endtoend/testdata/ddl_create_enum/postgresql/pgx/v4/go/models.go

Lines changed: 2 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

internal/endtoend/testdata/emit_enum_valid_and_values/go/models.go

Lines changed: 2 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)