feat: add model auto migrate

peterxcli · Jan 13, 2024 · b0ee30d · b0ee30d
1 parent 98a32d4
commit b0ee30d
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/cmd/main.go b/cmd/main.go
@@ -3,6 +3,7 @@ package main
 import (
 	"bikefest/docs"
 	"bikefest/pkg/bootstrap"
+	"bikefest/pkg/model"
 	"bikefest/pkg/router"
 	"bikefest/pkg/service"
 	"fmt"
@@ -50,6 +51,16 @@ func main() {
 	// init config
 	app := bootstrap.App()
 
+	// TODO: needs to consider whether migrate the db schema everytime the service boot
+	err := app.Conn.AutoMigrate(
+		&model.User{},
+		&model.Event{},
+		&model.PsychoTest{},
+	)
+	if err != nil {
+		panic(err)
+	}
+
 	// init services
 	userService := service.NewUserService(app.Conn, app.Cache)
 	eventService := service.NewEventService(app.Conn, app.Cache)