init

backend/providers/postgres/config.go

@@ -0,0 +1,136 @@
+package postgres
+
+import (
+	"fmt"
+	"strconv"
+	"time"
+
+	"go.ipao.vip/atom/container"
+	"go.ipao.vip/atom/opt"
+	"gorm.io/gorm/logger"
+)
+
+const DefaultPrefix = "Database"
+
+func DefaultProvider() container.ProviderContainer {
+	return container.ProviderContainer{
+		Provider: Provide,
+		Options: []opt.Option{
+			opt.Prefix(DefaultPrefix),
+		},
+	}
+}
+
+type Config struct {
+	Username     string
+	Password     string
+	Database     string
+	Schema       string
+	Host         string
+	Port         uint
+	SslMode      string
+	TimeZone     string
+	Prefix       string // 表前缀
+	Singular     bool   // 是否开启全局禁用复数，true表示开启
+	MaxIdleConns int    // 空闲中的最大连接数
+	MaxOpenConns int    // 打开到数据库的最大连接数
+	// 可选：连接生命周期配置（0 表示不设置）
+	ConnMaxLifetimeSeconds uint
+	ConnMaxIdleTimeSeconds uint
+
+	// 可选：GORM 日志与行为配置
+	LogLevel               string // silent|error|warn|info（默认info）
+	SlowThresholdMs        uint   // 慢查询阈值（毫秒）默认200
+	ParameterizedQueries   bool   // 占位符输出，便于日志安全与查询归并
+	PrepareStmt            bool   // 预编译语句缓存
+	SkipDefaultTransaction bool   // 跳过默认事务
+
+	// 可选：DSN 增强
+	UseSearchPath   bool   // 在 DSN 中附带 search_path
+	ApplicationName string // application_name
+}
+
+func (m Config) GormSlowThreshold() time.Duration {
+	if m.SlowThresholdMs == 0 {
+		return 200 * time.Millisecond // 默认200ms
+	}
+	return time.Duration(m.SlowThresholdMs) * time.Millisecond
+}
+
+func (m Config) GormLogLevel() logger.LogLevel {
+	switch m.LogLevel {
+	case "silent":
+		return logger.Silent
+	case "error":
+		return logger.Error
+	case "warn":
+		return logger.Warn
+	case "info", "":
+		return logger.Info
+	default:
+		return logger.Info
+	}
+}
+
+func (m *Config) checkDefault() {
+	if m.MaxIdleConns == 0 {
+		m.MaxIdleConns = 10
+	}
+
+	if m.MaxOpenConns == 0 {
+		m.MaxOpenConns = 100
+	}
+
+	if m.Username == "" {
+		m.Username = "postgres"
+	}
+
+	if m.SslMode == "" {
+		m.SslMode = "disable"
+	}
+
+	if m.TimeZone == "" {
+		m.TimeZone = "Asia/Shanghai"
+	}
+
+	if m.Port == 0 {
+		m.Port = 5432
+	}
+
+	if m.Schema == "" {
+		m.Schema = "public"
+	}
+}
+
+func (m *Config) EmptyDsn() string {
+	// 基本 DSN
+	dsnTpl := "host=%s user=%s password=%s port=%d dbname=%s sslmode=%s TimeZone=%s"
+	m.checkDefault()
+	base := fmt.Sprintf(dsnTpl, m.Host, m.Username, m.Password, m.Port, m.Database, m.SslMode, m.TimeZone)
+	// 附加可选参数
+	extras := ""
+	if m.UseSearchPath && m.Schema != "" {
+		extras += " search_path=" + m.Schema
+	}
+	if m.ApplicationName != "" {
+		extras += " application_name=" + strconv.Quote(m.ApplicationName)
+	}
+	return base + extras
+}
+
+// DSN connection dsn
+func (m *Config) DSN() string {
+	// 基本 DSN
+	dsnTpl := "host=%s user=%s password=%s dbname=%s port=%d sslmode=%s TimeZone=%s"
+	m.checkDefault()
+	base := fmt.Sprintf(dsnTpl, m.Host, m.Username, m.Password, m.Database, m.Port, m.SslMode, m.TimeZone)
+	// 附加可选参数
+	extras := ""
+	if m.UseSearchPath && m.Schema != "" {
+		extras += " search_path=" + m.Schema
+	}
+	if m.ApplicationName != "" {
+		extras += " application_name=" + strconv.Quote(m.ApplicationName)
+	}
+	return base + extras
+}

backend/providers/postgres/postgres.go

@@ -0,0 +1,91 @@
+package postgres
+
+import (
+	"context"
+	"database/sql"
+	"time"
+
+	"github.com/sirupsen/logrus"
+	"go.ipao.vip/atom/container"
+	"go.ipao.vip/atom/opt"
+	"gorm.io/driver/postgres"
+	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
+	"gorm.io/gorm/schema"
+)
+
+func Provide(opts ...opt.Option) error {
+	o := opt.New(opts...)
+	var conf Config
+	if err := o.UnmarshalConfig(&conf); err != nil {
+		return err
+	}
+
+	return container.Container.Provide(func() (*gorm.DB, *sql.DB, *Config, error) {
+		dbConfig := postgres.Config{DSN: conf.DSN()}
+
+		// 安全日志：不打印密码，仅输出关键连接信息
+		logrus.
+			WithFields(
+				logrus.Fields{
+					"host":   conf.Host,
+					"port":   conf.Port,
+					"db":     conf.Database,
+					"schema": conf.Schema,
+					"ssl":    conf.SslMode,
+				},
+			).
+			Info("opening PostgreSQL connection")
+
+		// 映射日志等级
+		lvl := conf.GormLogLevel()
+		slow := conf.GormSlowThreshold()
+
+		gormConfig := gorm.Config{
+			NamingStrategy: schema.NamingStrategy{
+				TablePrefix:   conf.Prefix,
+				SingularTable: conf.Singular,
+			},
+			DisableForeignKeyConstraintWhenMigrating: true,
+			PrepareStmt:                              conf.PrepareStmt,
+			SkipDefaultTransaction:                   conf.SkipDefaultTransaction,
+			Logger: logger.New(logrus.StandardLogger(), logger.Config{
+				SlowThreshold:             slow,
+				LogLevel:                  lvl,
+				IgnoreRecordNotFoundError: true,
+				Colorful:                  false,
+				ParameterizedQueries:      conf.ParameterizedQueries,
+			}),
+		}
+
+		db, err := gorm.Open(postgres.New(dbConfig), &gormConfig)
+		if err != nil {
+			return nil, nil, nil, err
+		}
+
+		sqlDB, err := db.DB()
+		if err != nil {
+			return nil, nil, nil, err
+		}
+		sqlDB.SetMaxIdleConns(conf.MaxIdleConns)
+		sqlDB.SetMaxOpenConns(conf.MaxOpenConns)
+		if conf.ConnMaxLifetimeSeconds > 0 {
+			sqlDB.SetConnMaxLifetime(time.Duration(conf.ConnMaxLifetimeSeconds) * time.Second)
+		}
+		if conf.ConnMaxIdleTimeSeconds > 0 {
+			sqlDB.SetConnMaxIdleTime(time.Duration(conf.ConnMaxIdleTimeSeconds) * time.Second)
+		}
+
+		// Ping 校验
+		ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+		defer cancel()
+		if err := sqlDB.PingContext(ctx); err != nil {
+			return nil, nil, nil, err
+		}
+
+		// 关闭钩子
+		container.AddCloseAble(func() { _ = sqlDB.Close() })
+
+		return db, sqlDB, &conf, nil
+	}, o.DiOptions()...)
+}