38 lines
600 B
Go
38 lines
600 B
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"inou/lib"
|
|
)
|
|
|
|
func main() {
|
|
if len(os.Args) < 2 {
|
|
fmt.Println("Usage: migrate <dob|category>")
|
|
os.Exit(1)
|
|
}
|
|
|
|
if err := lib.Init(); err != nil {
|
|
fmt.Printf("Error initializing: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
defer lib.DBClose()
|
|
|
|
switch os.Args[1] {
|
|
case "dob":
|
|
if err := lib.MigrateDOB(); err != nil {
|
|
fmt.Printf("Error: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
case "category":
|
|
if err := lib.MigrateCategory(); err != nil {
|
|
fmt.Printf("Error: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
default:
|
|
fmt.Printf("Unknown migration: %s\n", os.Args[1])
|
|
os.Exit(1)
|
|
}
|
|
}
|