remove lib/pq imports, add shim to register jackc/pgx stdlib driver as 'postgres'

Author: maxtheaxe

go.mod

@@ -92,7 +92,6 @@ require (
 	github.com/klauspost/compress v1.18.1
 	github.com/klauspost/pgzip v1.2.6
 	github.com/knights-analytics/hugot v0.4.3
-	github.com/lib/pq v1.10.9
 	github.com/linkedin/goavro/v2 v2.12.0
 	github.com/matoous/go-nanoid/v2 v2.0.0
 	github.com/microcosm-cc/bluemonday v1.0.27

go.sum

@@ -1484,8 +1484,8 @@ github.com/lib/pq v1.1.0/go.mod h1:5WUZQaWbwv1U+lTReE5YruASi9Al49XbQIvNi/34Woo=
 github.com/lib/pq v1.2.0/go.mod h1:5WUZQaWbwv1U+lTReE5YruASi9Al49XbQIvNi/34Woo=
 github.com/lib/pq v1.10.2/go.mod h1:AlVN5x4E4T544tWzH6hKfbfQvm3HdbOxrmggDNAPY9o=
 github.com/lib/pq v1.10.4/go.mod h1:AlVN5x4E4T544tWzH6hKfbfQvm3HdbOxrmggDNAPY9o=
-github.com/lib/pq v1.10.9 h1:YXG7RB+JIjhP29X+OtkiDnYaXQwpS4JEWq7dtCCRUEw=
-github.com/lib/pq v1.10.9/go.mod h1:AlVN5x4E4T544tWzH6hKfbfQvm3HdbOxrmggDNAPY9o=
+github.com/lib/pq v1.10.7 h1:p7ZhMD+KsSRozJr34udlUrhboJwWAgCg34+/ZZNvZZw=
+github.com/lib/pq v1.10.7/go.mod h1:AlVN5x4E4T544tWzH6hKfbfQvm3HdbOxrmggDNAPY9o=
 github.com/linkedin/goavro/v2 v2.12.0 h1:rIQQSj8jdAUlKQh6DttK8wCRv4t4QO09g1C4aBWXslg=
 github.com/linkedin/goavro/v2 v2.12.0/go.mod h1:KXx+erlq+RPlGSPmLF7xGo6SAbh8sCQ53x064+ioxhk=
 github.com/lucasb-eyer/go-colorful v1.2.0 h1:1nnpGOrhyZZuNyfu1QjKiUICQ74+3FNCN69Aj6K7nkY=

internal/driver/postgrespgx/driver.go

@@ -0,0 +1,18 @@
+package postgrespgx
+
+import (
+	"database/sql"
+
+	"github.com/jackc/pgx/v4/stdlib"
+)
+
+func init() {
+	// Register pgx under the postgres driver name so existing configurations
+	// using `postgres` continue to work with the pgx stdlib driver.
+	for _, name := range sql.Drivers() {
+		if name == "postgres" {
+			return
+		}
+	}
+	sql.Register("postgres", stdlib.GetDefaultDriver())
+}

internal/impl/cockroachdb/exploration_test.go

@@ -13,7 +13,7 @@ import (
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 
-	_ "github.com/lib/pq"
+	_ "github.com/warpstreamlabs/bento/internal/driver/postgrespgx"
 
 	_ "github.com/warpstreamlabs/bento/public/components/io"
 	_ "github.com/warpstreamlabs/bento/public/components/pure"

internal/impl/cockroachdb/input_changefeed.go

@@ -17,8 +17,6 @@ import (
 	"github.com/Jeffail/shutdown"
 
 	"github.com/warpstreamlabs/bento/public/service"
-
-	_ "github.com/lib/pq"
 )
 
 var sampleString = `{

internal/impl/sql/driver_postgres.go

@@ -0,0 +1,5 @@
+package sql
+
+import (
+	_ "github.com/warpstreamlabs/bento/internal/driver/postgrespgx"
+)

public/components/sql/package.go

@@ -13,8 +13,8 @@ import (
 	_ "github.com/denisenkom/go-mssqldb"
 	_ "github.com/go-sql-driver/mysql"
 	_ "github.com/googleapis/go-sql-spanner"
-	_ "github.com/lib/pq"
 	_ "github.com/microsoft/gocosmos"
 	_ "github.com/sijms/go-ora/v2"
 	_ "github.com/trinodb/trino-go-client/trino"
+	_ "github.com/warpstreamlabs/bento/internal/driver/postgrespgx"
 )