go-trustlog/api/persistence/config.go

package persistence

import (
	"database/sql"
	"fmt"
	"time"
)

// DBConfig 数据库配置
type DBConfig struct {
	// DriverName 数据库驱动名称，如 "postgres", "mysql", "sqlite3" 等
	DriverName string
	// DSN 数据源名称（连接字符串）
	DSN string
	// MaxOpenConns 最大打开连接数
	MaxOpenConns int
	// MaxIdleConns 最大空闲连接数
	MaxIdleConns int
	// ConnMaxLifetime 连接最大生命周期
	ConnMaxLifetime time.Duration
	// ConnMaxIdleTime 连接最大空闲时间
	ConnMaxIdleTime time.Duration
}

// DefaultDBConfig 返回默认数据库配置
func DefaultDBConfig(driverName, dsn string) DBConfig {
	return DBConfig{
		DriverName:      driverName,
		DSN:             dsn,
		MaxOpenConns:    25,
		MaxIdleConns:    5,
		ConnMaxLifetime: time.Hour,
		ConnMaxIdleTime: 10 * time.Minute,
	}
}

// NewDB 创建并配置数据库连接
func NewDB(config DBConfig) (*sql.DB, error) {
	db, err := sql.Open(config.DriverName, config.DSN)
	if err != nil {
		return nil, fmt.Errorf("failed to open database: %w", err)
	}

	// 配置连接池
	db.SetMaxOpenConns(config.MaxOpenConns)
	db.SetMaxIdleConns(config.MaxIdleConns)
	db.SetConnMaxLifetime(config.ConnMaxLifetime)
	db.SetConnMaxIdleTime(config.ConnMaxIdleTime)

	// 测试连接
	if err := db.Ping(); err != nil {
		db.Close()
		return nil, fmt.Errorf("failed to ping database: %w", err)
	}

	return db, nil
}