优化流式传输&fix bugs

internal/db/db.go

@@ -2,8 +2,10 @@
 package db
 
 import (
+	"fmt"
 	"gemini-balancer/internal/config"
 	"gemini-balancer/internal/db/dialect"
+	"gemini-balancer/internal/db/migrations"
 	stdlog "log"
 	"os"
 	"path/filepath"
@@ -86,3 +88,16 @@ func NewDB(cfg *config.Config, appLogger *logrus.Logger) (*gorm.DB, dialect.Dial
 	Logger.Info("Database connection established successfully.")
 	return db, adapter, nil
 }
+
+func NewDBWithMigrations(cfg *config.Config, logger *logrus.Logger) (*gorm.DB, dialect.DialectAdapter, error) {
+	gormDB, adapter, err := NewDB(cfg, logger)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	if err := migrations.RunVersionedMigrations(gormDB, cfg, logger); err != nil {
+		return nil, nil, fmt.Errorf("failed to run versioned migrations: %w", err)
+	}
+
+	return gormDB, adapter, nil
+}