feat: use pgxpool

Author: clstb

go.mod

@@ -11,6 +11,7 @@ require (
 	github.com/gofrs/uuid v4.2.0+incompatible
 	github.com/golang-migrate/migrate/v4 v4.15.1
 	github.com/grpc-ecosystem/go-grpc-middleware v1.3.0
+	github.com/jackc/pgconn v1.11.0
 	github.com/jackc/pgx/v4 v4.15.0
 	github.com/lithammer/fuzzysearch v1.1.3
 	github.com/shopspring/decimal v1.3.1
@@ -31,12 +32,12 @@ require (
 	github.com/hashicorp/errwrap v1.0.0 // indirect
 	github.com/hashicorp/go-multierror v1.1.0 // indirect
 	github.com/jackc/chunkreader/v2 v2.0.1 // indirect
-	github.com/jackc/pgconn v1.11.0 // indirect
 	github.com/jackc/pgio v1.0.0 // indirect
 	github.com/jackc/pgpassfile v1.0.0 // indirect
 	github.com/jackc/pgproto3/v2 v2.2.0 // indirect
 	github.com/jackc/pgservicefile v0.0.0-20200714003250-2b9c44734f2b // indirect
 	github.com/jackc/pgtype v1.10.0 // indirect
+	github.com/jackc/puddle v1.2.1 // indirect
 	github.com/lib/pq v1.10.2 // indirect
 	github.com/lucasb-eyer/go-colorful v1.2.0 // indirect
 	github.com/mattn/go-isatty v0.0.14 // indirect

go.sum

@@ -629,6 +629,7 @@ github.com/jackc/puddle v0.0.0-20190608224051-11cab39313c9/go.mod h1:m4B5Dj62Y0f
 github.com/jackc/puddle v1.1.0/go.mod h1:m4B5Dj62Y0fbyuIc15OsIqK0+JU8nkqQjsgx7dvjSWk=
 github.com/jackc/puddle v1.1.1/go.mod h1:m4B5Dj62Y0fbyuIc15OsIqK0+JU8nkqQjsgx7dvjSWk=
 github.com/jackc/puddle v1.1.3/go.mod h1:m4B5Dj62Y0fbyuIc15OsIqK0+JU8nkqQjsgx7dvjSWk=
+github.com/jackc/puddle v1.2.1 h1:gI8os0wpRXFd4FiAY2dWiqRK037tjj3t7rKFeO4X5iw=
 github.com/jackc/puddle v1.2.1/go.mod h1:m4B5Dj62Y0fbyuIc15OsIqK0+JU8nkqQjsgx7dvjSWk=
 github.com/jinzhu/inflection v1.0.0/go.mod h1:h+uFLlag+Qp1Va5pdKtLDYj+kHp5pxUVkryuEj+Srlc=
 github.com/jinzhu/now v1.1.1/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8=

go/cmd/auth/BUILD.bazel

@@ -17,6 +17,7 @@ go_library(
         "//go/pkg/util",
         "@com_github_golang_migrate_migrate_v4//source/file",
         "@com_github_grpc_ecosystem_go_grpc_middleware//:go-grpc-middleware",
+        "@com_github_jackc_pgx_v4//pgxpool",
         "@com_github_jackc_pgx_v4//stdlib",
         "@com_github_urfave_cli_v2//:cli",
         "@org_golang_google_grpc//:go_default_library",

go/cmd/auth/migrate.go

@@ -1,18 +1,18 @@
 package main
 
 import (
-	"database/sql"
 	"fmt"
 
 	authdb "github.com/clstb/phi/go/pkg/auth/db"
 	"github.com/clstb/phi/go/pkg/util"
 	_ "github.com/golang-migrate/migrate/v4/source/file"
+	"github.com/jackc/pgx/v4/pgxpool"
 	_ "github.com/jackc/pgx/v4/stdlib"
 	"github.com/urfave/cli/v2"
 )
 
 func Migrate(ctx *cli.Context) error {
-	db, ok := ctx.Context.Value("db").(*sql.DB)
+	db, ok := ctx.Context.Value("db").(*pgxpool.Pool)
 	if !ok {
 		return fmt.Errorf("missing db")
 	}

go/cmd/auth/server.go

@@ -1,21 +1,21 @@
 package main
 
 import (
-	"database/sql"
 	"fmt"
 
 	pb "github.com/clstb/phi/go/pkg/auth/pb"
 	"github.com/clstb/phi/go/pkg/auth/server"
 	"github.com/clstb/phi/go/pkg/interceptor"
 	"github.com/clstb/phi/go/pkg/util"
 	grpc_middleware "github.com/grpc-ecosystem/go-grpc-middleware"
+	"github.com/jackc/pgx/v4/pgxpool"
 	"github.com/urfave/cli/v2"
 	"go.uber.org/zap"
 	"google.golang.org/grpc"
 )
 
 func Server(ctx *cli.Context) error {
-	db, ok := ctx.Context.Value("db").(*sql.DB)
+	db, ok := ctx.Context.Value("db").(*pgxpool.Pool)
 	if !ok {
 		return fmt.Errorf("missing db")
 	}

go/cmd/tinkgw/BUILD.bazel

@@ -17,6 +17,7 @@ go_library(
         "//go/pkg/util",
         "@com_github_golang_migrate_migrate_v4//source/file",
         "@com_github_grpc_ecosystem_go_grpc_middleware//:go-grpc-middleware",
+        "@com_github_jackc_pgx_v4//pgxpool",
         "@com_github_jackc_pgx_v4//stdlib",
         "@com_github_urfave_cli_v2//:cli",
         "@org_golang_google_grpc//:go_default_library",

go/cmd/tinkgw/migrate.go

@@ -1,18 +1,18 @@
 package main
 
 import (
-	"database/sql"
 	"fmt"
 
 	tinkgwdb "github.com/clstb/phi/go/pkg/tinkgw/db"
 	"github.com/clstb/phi/go/pkg/util"
 	_ "github.com/golang-migrate/migrate/v4/source/file"
+	"github.com/jackc/pgx/v4/pgxpool"
 	_ "github.com/jackc/pgx/v4/stdlib"
 	"github.com/urfave/cli/v2"
 )
 
 func Migrate(ctx *cli.Context) error {
-	db, ok := ctx.Context.Value("db").(*sql.DB)
+	db, ok := ctx.Context.Value("db").(*pgxpool.Pool)
 	if !ok {
 		return fmt.Errorf("missing db")
 	}

go/cmd/tinkgw/server.go

@@ -1,21 +1,21 @@
 package main
 
 import (
-	"database/sql"
 	"fmt"
 
 	"github.com/clstb/phi/go/pkg/interceptor"
 	pb "github.com/clstb/phi/go/pkg/tinkgw/pb"
 	"github.com/clstb/phi/go/pkg/tinkgw/server"
 	"github.com/clstb/phi/go/pkg/util"
 	grpc_middleware "github.com/grpc-ecosystem/go-grpc-middleware"
+	"github.com/jackc/pgx/v4/pgxpool"
 	"github.com/urfave/cli/v2"
 	"go.uber.org/zap"
 	"google.golang.org/grpc"
 )
 
 func Server(ctx *cli.Context) error {
-	db, ok := ctx.Context.Value("db").(*sql.DB)
+	db, ok := ctx.Context.Value("db").(*pgxpool.Pool)
 	if !ok {
 		return fmt.Errorf("missing db")
 	}

go/pkg/auth/db/BUILD.bazel

@@ -15,5 +15,9 @@ go_library(
     ],
     importpath = "github.com/clstb/phi/go/pkg/auth/db",
     visibility = ["//visibility:public"],
-    deps = ["@com_github_gofrs_uuid//:uuid"],
+    deps = [
+        "@com_github_gofrs_uuid//:uuid",
+        "@com_github_jackc_pgconn//:pgconn",
+        "@com_github_jackc_pgx_v4//:pgx",
+    ],
 )