celogeek/photos-api
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add db parameter

Browse Source
This commit is contained in:
celogeek 2021-12-06 13:31:22 +01:00
parent 4206dcf433
commit e82e51af2d
Signed by: celogeek
GPG Key ID: E6B7BDCFC446233A
2 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/controllers/models/setup.go
View File

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

7
src/main.go
View File

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