fix: db

do4-2022 · Mar 20, 2024 · 90b0aef · 90b0aef
1 parent f4347b0
commit 90b0aef
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/routes/controller.go b/routes/controller.go
@@ -1,4 +1,7 @@
 package routes
 
+import "gorm.io/gorm"
+
 type Controller struct {
+	DB *gorm.DB
 }
diff --git a/routes/router.go b/routes/router.go
@@ -9,7 +9,7 @@ func GetRoutes(db *gorm.DB) *gin.Engine {
 
 	router := gin.Default()
 
-	controller := Controller{}
+	controller := Controller{DB: db}
 
 	router.POST("/user", controller.createUser)