Parameterize dbFile location for InitSqlDatabase

jcardarelli · Oct 12, 2023 · 01a64b3 · 01a64b3
1 parent 83f5a6c
commit 01a64b3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/cmd/init.go b/cmd/init.go
@@ -18,7 +18,7 @@ Cobra is a CLI library for Go that empowers applications.
 This application is a tool to generate the needed files
 to quickly create a Cobra application.`,
 	Run: func(cmd *cobra.Command, args []string) {
-		err := db.InitSqlDatabase()
+		err := db.InitSqlDatabase("fancy-cli.db")
 		if err != nil {
 			log.Fatalln(err)
 		}

diff --git a/database/sqlite.go b/database/sqlite.go
@@ -26,9 +26,9 @@ func OpenConnectionPool(sqliteDbFile string) error {
 }
 
 // Setup database connection
-func InitSqlDatabase() error {
+func InitSqlDatabase(dbFile string) error {
 	// Attempt to open a connection to the sqlite file
-	connErr := OpenConnectionPool("fancy-cli.db")
+	connErr := OpenConnectionPool(dbFile)
 	if connErr != nil {
 		log.Fatalln("error connecting to sqlite", connErr)
 	}