Initial commit: 商品售卖网站

2026-04-13 07:20:09 +08:00
commit c6154273f2
865 changed files with 26573 additions and 0 deletions
@@ -0,0 +1,79 @@
+package utils
+
+import (
+	"fmt"
+	"log"
+
+	"sale/internal/config"
+	"sale/internal/models"
+
+	sqlite "github.com/glebarez/sqlite"
+	"gorm.io/driver/mysql"
+	"gorm.io/driver/postgres"
+	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
+)
+
+var DB *gorm.DB
+
+func InitDB() {
+	var err error
+	cfg := config.AppConfig.Database
+
+	switch cfg.Driver {
+	case "mysql":
+		dsn := fmt.Sprintf(
+			"%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
+			cfg.User, cfg.Password, cfg.Host, cfg.Port, cfg.DBName,
+		)
+		DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{
+			Logger: logger.Default.LogMode(logger.Info),
+		})
+	case "postgres":
+		dsn := fmt.Sprintf(
+			"host=%s port=%s user=%s password=%s dbname=%s sslmode=%s",
+			cfg.Host, cfg.Port, cfg.User, cfg.Password, cfg.DBName, cfg.SSLMode,
+		)
+		DB, err = gorm.Open(postgres.Open(dsn), &gorm.Config{
+			Logger: logger.Default.LogMode(logger.Info),
+		})
+	default:
+		DB, err = gorm.Open(sqlite.Open(cfg.FilePath), &gorm.Config{
+			Logger: logger.Default.LogMode(logger.Info),
+		})
+	}
+
+	if err != nil {
+		log.Fatalf("Failed to connect to database: %v", err)
+	}
+
+	log.Printf("Database connected successfully (driver: %s)", cfg.Driver)
+}
+
+func AutoMigrate() {
+	err := DB.AutoMigrate(
+		&models.User{},
+		&models.Category{},
+		&models.Brand{},
+		&models.Product{},
+		&models.ProductCustomField{},
+		&models.Inventory{},
+		&models.Cart{},
+		&models.Order{},
+		&models.OrderItem{},
+		&models.Address{},
+		&models.Lottery{},
+		&models.LotteryPrize{},
+		&models.LotteryParticipant{},
+		&models.LotteryWinner{},
+		&models.Ticket{},
+		&models.SystemSetting{},
+		&models.SupplierAuthorization{},
+		&models.Article{},
+	)
+	if err != nil {
+		log.Fatalf("Failed to migrate database: %v", err)
+	}
+
+	log.Println("Database migrated successfully")
+}