初始化

models/db.go

@@ -0,0 +1,83 @@
+// Package models 提供数据库初始化和连接管理功能
+package models
+
+import (
+	"fmt"
+	"log"
+
+	"excel-api/config"
+
+	"gorm.io/driver/mysql"
+	"gorm.io/gorm"
+)
+
+// DB 全局数据库连接实例
+var DB *gorm.DB
+
+// InitDB 初始化数据库连接
+// 读取配置文件中的数据库连接信息，建立MySQL连接并自动迁移表结构
+func InitDB() error {
+	cfg := config.App.Database
+	// 构建MySQL DSN（Data Source Name）连接字符串
+	dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=%s&parseTime=True&loc=Local",
+		cfg.Username,
+		cfg.Password,
+		cfg.Host,
+		cfg.Port,
+		cfg.DBName,
+		cfg.Charset,
+	)
+	var err error
+	DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
+	if err != nil {
+		return fmt.Errorf("数据库连接失败: %w", err)
+	}
+
+	// 获取底层 *sql.DB 用于配置连接池
+	sqlDB, err := DB.DB()
+	if err != nil {
+		return fmt.Errorf("获取数据库连接池失败: %w", err)
+	}
+
+	// 设置连接池参数
+	sqlDB.SetMaxIdleConns(10)  // 最大空闲连接数
+	sqlDB.SetMaxOpenConns(100) // 最大打开连接数
+
+	// 自动迁移：根据模型结构体创建/更新数据库表
+	// 注意：AutoMigrate 仅创建表和添加缺失列，不会删除已有列
+	// 完整的表结构变更请参照根目录 init.sql 文件
+	err = DB.AutoMigrate(
+		&User{},
+		&Role{},
+		&Permission{},
+		&RolePermission{},
+		&UserRole{},
+		&Document{},
+		&Workbook{},
+		&Sheet{},
+		&Cell{},
+		&DocumentRole{},
+		&UserDocument{},
+		&WorkbookPermission{},
+		&OperationLog{},
+		&ShareLink{},
+		&Comment{},
+		&Notification{},
+		&Tag{},
+		&DocumentTag{},
+		&Folder{},
+		&FileUpload{},
+		&Version{},
+		&CollaborationSession{},
+		&ExportHistory{},
+		&Setting{},
+		&ApiKey{},
+		&AuditLog{},
+	)
+	if err != nil {
+		return fmt.Errorf("数据库迁移失败: %w", err)
+	}
+
+	log.Println("数据库连接成功，表结构迁移完成")
+	return nil
+}