midsuit
/
wms-be
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add support for fresh migrations in command script

This commit is contained in:
Habib Fatkhul Rohman 2025-09-16 11:01:13 +07:00
parent a2790436d2
commit cc66c9c251
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
script/command.go
View File

@ -20,6 +20,7 @@ func Commands(injector *do.Injector) bool {
seed := false seed := false
run := false run := false
scriptFlag := false scriptFlag := false
fresh := false
for _, arg := range os.Args[1:] { for _, arg := range os.Args[1:] {
if arg == "--migrate" { if arg == "--migrate" {
@ -31,13 +32,21 @@ func Commands(injector *do.Injector) bool {
if arg == "--run" { if arg == "--run" {
run = true run = true
} }
if arg == "--fresh" {
fresh = true
}
if strings.HasPrefix(arg, "--script:") { if strings.HasPrefix(arg, "--script:") {
scriptFlag = true scriptFlag = true
scriptName = strings.TrimPrefix(arg, "--script:") scriptName = strings.TrimPrefix(arg, "--script:")
} }
} }
if migrate { if fresh {
if err := database.MigrateFresh(db); err != nil {
log.Fatalf("error migrate fresh: %v", err)
}
log.Println("fresh migration completed successfully")
} else if migrate {
if err := database.Migrate(db); err != nil { if err := database.Migrate(db); err != nil {
log.Fatalf("error migration: %v", err) log.Fatalf("error migration: %v", err)
} }