20260507213409

code/app/app-go/internal/bootstrap/boots/pgsql.go

@@ -0,0 +1,100 @@
+package boots
+
+import (
+	"allapp-go/internal/config"
+	"context"
+	"fmt"
+	"net/url"
+	"time"
+
+	"github.com/jackc/pgx/v5/pgxpool"
+)
+
+func InitPgsql(cfg *config.Config) (*pgxpool.Pool, error) {
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+
+	pg := cfg.Postgres
+
+	// ssl
+	sslMode := "disable"
+	if pg.SslMode {
+		sslMode = "require"
+	}
+
+	// 密码转义
+	password := url.QueryEscape(pg.Password)
+
+	// DSN
+	dsn := fmt.Sprintf(
+		"postgres://%s:%s@%s:%d/%s?sslmode=%s",
+		pg.User,
+		password,
+		pg.Host,
+		pg.Port,
+		pg.Dbname,
+		sslMode,
+	)
+
+	// 解析配置
+	conf, err := pgxpool.ParseConfig(dsn)
+	if err != nil {
+		return nil, err
+	}
+
+	// ======================
+	// 连接池配置（带默认值）
+	// ======================
+
+	// 默认最大连接数
+	maxConns := int32(20)
+	if pg.MaxOpenConns > 0 {
+		maxConns = int32(pg.MaxOpenConns)
+	}
+
+	// 默认最小连接数
+	minConns := int32(2)
+	if pg.MaxIdleConns > 0 {
+		minConns = int32(pg.MaxIdleConns)
+	}
+
+	// 防止配置错误：min > max
+	if minConns > maxConns {
+		minConns = maxConns
+	}
+
+	// 应用配置
+	conf.MaxConns = maxConns
+	conf.MinConns = minConns
+
+	// 默认连接最大生命周期
+	if pg.ConnMaxLifetime > 0 {
+		conf.MaxConnLifetime = pg.ConnMaxLifetime
+	} else {
+		conf.MaxConnLifetime = time.Hour
+	}
+
+	// （推荐补充）空闲连接最大存活时间
+	conf.MaxConnIdleTime = 30 * time.Minute
+
+	// ======================
+	// 时区设置（正确方式 ⭐）
+	// ======================
+	if pg.TimeZone != "" {
+		conf.ConnConfig.RuntimeParams["TimeZone"] = pg.TimeZone
+	}
+
+	// 创建连接池
+	db, err := pgxpool.NewWithConfig(ctx, conf)
+	if err != nil {
+		return nil, err
+	}
+
+	// 测试连接
+	if err := db.Ping(ctx); err != nil {
+		db.Close()
+		return nil, err
+	}
+
+	return db, nil
+}