FileShare/server/internal/database/database.go

// Package database 负责数据库连接与迁移（需求 ⑧：双方言）：
//   - sqlite（默认）：modernc.org/sqlite 纯 Go 驱动（GORM 封装 glebarez/sqlite），零 CGO、零外部依赖；
//   - postgres：可选，配置 FCB_DB_DRIVER=postgres + FCB_DB_DSN 后启用。
//
// 两方言共用 GORM 抽象层，AutoMigrate 与全部业务查询保持方言无关；
// 唯一的原生 SQL（migrates 建表）已改为双方言分支。
package database

import (
	"context"
	"fmt"
	"os"
	"path/filepath"
	"strings"
	"time"

	"github.com/glebarez/sqlite"
	"gorm.io/driver/postgres"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"

	"fileshare/internal/config"
	"fileshare/internal/model"
)

// Options 连接选项（main.go 从 config.Env 装配）。
type Options struct {
	Driver string // sqlite | postgres（空按 sqlite 处理）
	DSN    string // postgres 连接串；sqlite 为文件路径（空回退 config.DefaultSQLitePath）
}

// Open 按驱动连接数据库并执行连接池设置与探活。
func Open(ctx context.Context, opts Options) (*gorm.DB, error) {
	driver := strings.ToLower(strings.TrimSpace(opts.Driver))
	if driver == "" {
		driver = config.DBDriverSQLite
	}
	var dialector gorm.Dialector
	switch driver {
	case config.DBDriverSQLite:
		path := strings.TrimSpace(opts.DSN)
		if path == "" {
			path = config.DefaultSQLitePath
		}
		// 自动创建父目录（如 ./data），对齐参考实现 data_root 语义
		if dir := filepath.Dir(path); dir != "" && dir != "." {
			if err := os.MkdirAll(dir, 0o755); err != nil {
				return nil, fmt.Errorf("database: 创建 SQLite 目录 %s 失败: %w", dir, err)
			}
		}
		// DSN 参数：busy_timeout 防写锁竞态；WAL 提升并发读写（Query 参数形式，驱动原生支持）
		dsn := path + "?_pragma=busy_timeout(10000)&_pragma=journal_mode(WAL)&_pragma=foreign_keys(1)"
		dialector = sqlite.Open(dsn)
	case config.DBDriverPostgres:
		if strings.TrimSpace(opts.DSN) == "" {
			return nil, fmt.Errorf("database: FCB_DB_DRIVER=postgres 需要提供 FCB_DB_DSN")
		}
		dialector = postgres.Open(opts.DSN)
	default:
		return nil, fmt.Errorf("database: 不支持的数据库驱动 %q（仅支持 sqlite|postgres）", driver)
	}

	db, err := gorm.Open(dialector, &gorm.Config{
		Logger: logger.Default.LogMode(logger.Warn),
		// 避免 GORM 生成方言特有子句；时间语义由应用层统一（容器本地时区）
		NowFunc: time.Now,
	})
	if err != nil {
		return nil, fmt.Errorf("database: 连接 %s 失败: %w", driver, err)
	}

	sqlDB, err := db.DB()
	if err != nil {
		return nil, err
	}
	// 连接池：SQLite 单文件场景保守设置；Postgres 沿用 v1 参数
	switch driver {
	case config.DBDriverSQLite:
		sqlDB.SetMaxOpenConns(8)
		sqlDB.SetMaxIdleConns(4)
		sqlDB.SetConnMaxLifetime(0) // 长连接文件句柄，无需轮换
	case config.DBDriverPostgres:
		sqlDB.SetMaxOpenConns(32)
		sqlDB.SetMaxIdleConns(8)
		sqlDB.SetConnMaxLifetime(time.Hour)
	}

	// 连接探活（带超时）
	pingCtx, cancel := context.WithTimeout(ctx, 10*time.Second)
	defer cancel()
	if err := sqlDB.PingContext(pingCtx); err != nil {
		return nil, fmt.Errorf("database: %s 探活失败: %w", driver, err)
	}
	return db, nil
}

// Migrate 执行迁移：先建迁移台账表（双方言分支），再 AutoMigrate 全部模型。
func Migrate(ctx context.Context, db *gorm.DB) error {
	if err := createMigratesTable(ctx, db); err != nil {
		return err
	}
	if err := model.AutoMigrate(db); err != nil {
		return fmt.Errorf("database: AutoMigrate 失败: %w", err)
	}
	return nil
}

// createMigratesTable 创建迁移台账表。
// 双方言差异：自增主键 postgres 用 BIGSERIAL、sqlite 用 INTEGER PRIMARY KEY AUTOINCREMENT；
// 时间戳默认值 postgres 用 CURRENT_TIMESTAMP、sqlite 用 CURRENT_TIMESTAMP（等价）。
func createMigratesTable(ctx context.Context, db *gorm.DB) error {
	ddl := `
		CREATE TABLE IF NOT EXISTS migrates (
			id INTEGER PRIMARY KEY AUTOINCREMENT,
			migration_file VARCHAR(255) NOT NULL UNIQUE,
			executed_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
		)`
	if db.Dialector.Name() == config.DBDriverPostgres {
		ddl = `
		CREATE TABLE IF NOT EXISTS migrates (
			id BIGSERIAL PRIMARY KEY,
			migration_file VARCHAR(255) NOT NULL UNIQUE,
			executed_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
		)`
	}
	if err := db.WithContext(ctx).Exec(ddl).Error; err != nil {
		return fmt.Errorf("database: 创建 migrates 表失败: %w", err)
	}
	return nil
}

// Close 关闭底层连接。
func Close(db *gorm.DB) error {
	sqlDB, err := db.DB()
	if err != nil {
		return err
	}
	return sqlDB.Close()
}