refactor(pg):更换数据库驱动为pgx。

global/db.go

@@ -1,16 +1,16 @@
 package global
 
 import (
-	"database/sql"
 	"fmt"
 	"time"
 
 	"electricity_bill_calc/config"
 	"electricity_bill_calc/logger"
 
+	"github.com/jackc/pgx/v4"
+	"github.com/jackc/pgx/v4/log/zapadapter"
+	"github.com/jackc/pgx/v4/stdlib"
 	_ "github.com/lib/pq"
-	sqldblogger "github.com/simukti/sqldb-logger"
-	"github.com/simukti/sqldb-logger/logadapter/zapadapter"
 	"github.com/uptrace/bun"
 	"github.com/uptrace/bun/dialect/pgdialect"
 )
@@ -22,19 +22,24 @@ var (
 func SetupDatabaseConnection() error {
 	var err error
 	connStr := fmt.Sprintf(
-		"host=%s user=%s password=%s dbname=%s port=%d sslmode=disable TimeZone=Asia/Shanghai connect_timeout=0",
+		"host=%s user=%s password=%s dbname=%s port=%d sslmode=disable TimeZone=Asia/Shanghai connect_timeout=0 tcp_user_timeout=180000",
 		config.DatabaseSettings.Host,
 		config.DatabaseSettings.User,
 		config.DatabaseSettings.Pass,
 		config.DatabaseSettings.DB,
 		config.DatabaseSettings.Port,
 	)
-	db, err := sql.Open("postgres", connStr)
+	pgxConn, err := pgx.ParseConfig(connStr)
 	if err != nil {
-		return err
+		panic(err)
 	}
-	db = sqldblogger.OpenDriver(connStr, db.Driver(), zapadapter.New(logger.Named("PG")))
-	DB = bun.NewDB(db, pgdialect.New())
+	pgxConn.PreferSimpleProtocol = true
+	if config.ServerSettings.RunMode == "debug" {
+		pgxConn.Logger = zapadapter.NewLogger(logger.Named("PGX"))
+		pgxConn.LogLevel = pgx.LogLevelDebug
+	}
+	sqldb := stdlib.OpenDB(*pgxConn)
+	DB = bun.NewDB(sqldb, pgdialect.New())
 	DB.SetMaxIdleConns(config.DatabaseSettings.MaxIdleConns)
 	DB.SetMaxOpenConns(config.DatabaseSettings.MaxOpenConns)
 	DB.SetConnMaxIdleTime(10 * time.Minute)