数据源和 oss

internal/data/data.go

@@ -4,55 +4,47 @@ import (
 	"context"
 	"fmt"
 	"log"
-	"os"
+	"sync"
 	"time"
 
-	_ "github.com/go-sql-driver/mysql"
 	"github.com/google/wire"
 	"github.com/redis/go-redis/v9"
-	"gorm.io/driver/mysql"
 	"gorm.io/gorm"
 	"kra/internal/conf"
-	"kra/internal/data/ent"
-	"kra/internal/data/ent/migrate"
 )
 
-// ProviderSet is data providers.
 var ProviderSet = wire.NewSet(NewData, NewAdminRepo, NewSystemRepo, NewAccessRepo, NewSettingsRepo, NewVersionRepo, NewExportRepo, NewAuditRepo, NewTaskRepo, NewMediaRepo, NewAnnouncementRepo, NewEmailRepo, NewCache, NewFileStorage)
 
-// Data is a struct that contains the database client.
 type Data struct {
-	db     *ent.Client
-	gormDB *gorm.DB
-	redis  *redis.Client
+	mu       sync.RWMutex
+	gormDB   *gorm.DB
+	redis    *redis.Client
+	database *conf.Data_Database
+	config   *conf.Data
+	admin    *conf.AdminBackend
 }
 
-// NewData creates a new Data instance.
-func NewData(c *conf.Data) (*Data, func(), error) {
+func NewData(c *conf.Data, admin *conf.AdminBackend) (*Data, func(), error) {
 	if c == nil || c.Database == nil {
 		return nil, nil, fmt.Errorf("database configuration is required")
 	}
-	db, err := ent.Open(c.Database.Driver, c.Database.Source)
+	d := &Data{database: c.Database, config: c, admin: admin}
+	db, err := openDatabase(c.Database, false)
 	if err != nil {
-		return nil, nil, fmt.Errorf("open ent database: %w", err)
-	}
-	gormDB, err := gorm.Open(mysql.Open(c.Database.Source), &gorm.Config{})
-	if err != nil {
-		_ = db.Close()
-		return nil, nil, err
-	}
-	// The announcement module owns its table lifecycle independently of the
-	// first-run database initializer.
-	if err = gormDB.AutoMigrate(&announcementPO{}); err != nil {
-		_ = db.Close()
-		return nil, nil, fmt.Errorf("migrate announcement table: %w", err)
-	}
-	var redisClient *redis.Client
-	if c.Redis != nil && c.Redis.Addr != "" {
-		options := &redis.Options{Addr: c.Redis.Addr}
-		if c.Redis.Network != "" {
-			options.Network = c.Redis.Network
+		// The initialization endpoint must remain available when the configured
+		// target database has not been created yet.
+		log.Printf("configured database unavailable before initialization: %v", err)
+		db, err = openFallbackDatabase()
+		if err != nil {
+			return nil, nil, fmt.Errorf("open bootstrap database: %w", err)
 		}
+	}
+	d.gormDB = db
+	if err = db.AutoMigrate(&announcementPO{}, &adminPO{}); err != nil {
+		return nil, nil, fmt.Errorf("migrate bootstrap tables: %w", err)
+	}
+	if c.Redis != nil && c.Redis.Addr != "" {
+		options := &redis.Options{Addr: c.Redis.Addr, Network: c.Redis.Network}
 		if c.Redis.ReadTimeout != nil {
 			options.ReadTimeout = c.Redis.ReadTimeout.AsDuration()
 		}
@@ -65,27 +57,42 @@ func NewData(c *conf.Data) (*Data, func(), error) {
 			log.Printf("redis unavailable, using in-memory cache: %v", pingErr)
 			_ = candidate.Close()
 		} else {
-			redisClient = candidate
+			d.redis = candidate
 		}
 		cancel()
 	}
-	if os.Getenv("DEPLOY_ENV") == "dev" {
-		// Enable debug mode for detailed logging.
-		db = db.Debug()
-		// Run the auto migration tool.
-		if err = db.Schema.Create(context.Background(), migrate.WithDropIndex(true)); err != nil {
-			return nil, nil, err
-		}
-	}
 	cleanup := func() {
-		_ = db.Close()
-		if redisClient != nil {
-			_ = redisClient.Close()
+		d.mu.RLock()
+		db := d.gormDB
+		d.mu.RUnlock()
+		if sqlDB, closeErr := db.DB(); closeErr == nil {
+			_ = sqlDB.Close()
+		}
+		if d.redis != nil {
+			_ = d.redis.Close()
 		}
 	}
-	return &Data{
-		db:     db,
-		gormDB: gormDB,
-		redis:  redisClient,
-	}, cleanup, nil
+	return d, cleanup, nil
+}
+
+func (d *Data) switchDatabase(config *conf.Data_Database) error {
+	db, err := openDatabase(config, true)
+	if err != nil {
+		return err
+	}
+	d.mu.Lock()
+	old := d.gormDB
+	d.gormDB = db
+	d.database = config
+	d.config.Database = config
+	d.mu.Unlock()
+	if err := d.persistConfig(); err != nil {
+		return fmt.Errorf("persist database configuration: %w", err)
+	}
+	if old != nil {
+		if sqlDB, e := old.DB(); e == nil {
+			_ = sqlDB.Close()
+		}
+	}
+	return nil
 }