add db parameter

src/controllers/models/setup.go

@@ -7,8 +7,8 @@ import (
 
 var DB *gorm.DB
 
-func ConnectDatabase() {
-	database, err := gorm.Open("sqlite3", "../test.db")
+func ConnectDatabase(filename *string) {
+	database, err := gorm.Open("sqlite3", *filename)
 
 	if err != nil {
 		panic("Failed to connect to database!")

src/main.go

@@ -1,15 +1,20 @@
 package main
 
 import (
+	"flag"
+
 	"github.com/gin-gonic/gin"
 	"gitlab.celogeek.com/photos/api/src/controllers"
 	"gitlab.celogeek.com/photos/api/src/controllers/models"
 )
 
 func main() {
+	dbFilename := flag.String("db", "test.db", "database filename")
+	flag.Parse()
+
 	r := gin.Default()
 
-	models.ConnectDatabase()
+	models.ConnectDatabase(dbFilename)
 
 	r.GET("/", controllers.DefaultRoot)