fix(db.go): added error checking for migration

GenerateNU · Oct 7, 2023 · 1aff58f · 1aff58f
1 parent 6a3843f
commit 1aff58f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/server/src/database/db.go b/server/src/database/db.go
@@ -51,7 +51,7 @@ func InitDB() (*gorm.DB, error) {
 		}
 	}
 
-	db.AutoMigrate(
+	if err := db.AutoMigrate(
 		&model.File{},
 		&model.Persona{},
 		&model.PersonaTaskJunction{},
@@ -61,7 +61,9 @@ func InitDB() (*gorm.DB, error) {
 		&model.Task{},
 		&model.UserProfile{},
 		&model.User{},
-	)
+	); err != nil {
+		return nil, fmt.Errorf("failed to perform database auto migration: %v", err)
+	}
 
 	return db, nil
 }