From e82e51af2d4600a66e73b01183b6a765adf19dcf Mon Sep 17 00:00:00 2001
From: celogeek <me@celogeek.com>
Date: Mon, 6 Dec 2021 13:31:22 +0100
Subject: [PATCH] add db parameter

---
 src/controllers/models/setup.go | 4 ++--
 src/main.go                     | 7 ++++++-
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/controllers/models/setup.go b/src/controllers/models/setup.go
index f4ed467..d982fbd 100644
--- a/src/controllers/models/setup.go
+++ b/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!")
diff --git a/src/main.go b/src/main.go
index 10c5247..af2136a 100644
--- a/src/main.go
+++ b/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)