55 lines
1.4 KiB
Go
55 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
|
|
"customer-support-system/internal/database"
|
|
"customer-support-system/pkg/config"
|
|
"customer-support-system/pkg/logger"
|
|
)
|
|
|
|
func main() {
|
|
// Load configuration
|
|
if err := config.LoadConfig(); err != nil {
|
|
log.Fatalf("Failed to load configuration: %v", err)
|
|
}
|
|
|
|
// Initialize logger
|
|
logger.InitLogger()
|
|
|
|
// Connect to database
|
|
if err := database.Connect(); err != nil {
|
|
logger.WithError(err).Fatal("Failed to connect to database")
|
|
}
|
|
|
|
// Parse command line arguments
|
|
action := flag.String("action", "up", "Migration action (up or down)")
|
|
flag.Parse()
|
|
|
|
// Run migrations based on action
|
|
switch *action {
|
|
case "up":
|
|
if err := database.AutoMigrate(); err != nil {
|
|
logger.WithError(err).Fatal("Failed to run database migrations")
|
|
}
|
|
fmt.Println("Database migrations completed successfully")
|
|
case "down":
|
|
// Note: GORM doesn't support automatic rollback of migrations
|
|
// In a real application, you would need to implement a more sophisticated migration system
|
|
fmt.Println("Automatic rollback is not supported with GORM")
|
|
fmt.Println("Please manually revert the database schema changes")
|
|
default:
|
|
fmt.Printf("Unknown action: %s\n", *action)
|
|
fmt.Println("Usage: go run cmd/migrate.go -action=[up|down]")
|
|
os.Exit(1)
|
|
}
|
|
|
|
// Close database connection
|
|
if err := database.Close(); err != nil {
|
|
logger.WithError(err).Error("Failed to close database connection")
|
|
}
|
|
}
|