feat(platform): 开源发布 - 初始化 Open AI Canvas

backend/internal/database/database.go

@@ -0,0 +1,59 @@
+package database
+
+import (
+	"errors"
+	"fmt"
+	"os"
+	"strings"
+
+	"gorm.io/driver/postgres"
+	"gorm.io/driver/sqlite"
+	"gorm.io/gorm"
+)
+
+type Config struct {
+	Driver  string
+	DSN     string
+	DataDir string
+}
+
+func Open(config Config) (*gorm.DB, error) {
+	driver := strings.ToLower(strings.TrimSpace(config.Driver))
+	if driver == "" {
+		driver = "sqlite"
+	}
+	switch driver {
+	case "sqlite":
+		dsn := strings.TrimSpace(config.DSN)
+		if dsn == "" {
+			if err := os.MkdirAll(config.DataDir, 0o755); err != nil {
+				return nil, err
+			}
+			dsn = config.DataDir + "/open_ai_canvas.db?_busy_timeout=5000&_journal_mode=WAL&_foreign_keys=on&_synchronous=NORMAL"
+		}
+		return gorm.Open(sqlite.Open(dsn), &gorm.Config{})
+	case "postgres", "postgresql":
+		dsn := strings.TrimSpace(config.DSN)
+		if dsn == "" {
+			return nil, errors.New("PostgreSQL 模式必须配置 DATABASE_URL")
+		}
+		return gorm.Open(postgres.Open(dsn), &gorm.Config{})
+	default:
+		return nil, fmt.Errorf("不支持的数据库驱动：%s", driver)
+	}
+}
+
+func ConfigurePool(db *gorm.DB) error {
+	sqlDB, err := db.DB()
+	if err != nil {
+		return err
+	}
+	if db.Dialector.Name() == "postgres" {
+		sqlDB.SetMaxOpenConns(30)
+		sqlDB.SetMaxIdleConns(10)
+		return nil
+	}
+	sqlDB.SetMaxOpenConns(8)
+	sqlDB.SetMaxIdleConns(4)
+	return nil
+}