Add PostgreSQL support (#23)
Signed-off-by: Julien Riou <julien@riou.xyz>
This commit is contained in:
parent
b6f23d4bd5
commit
47c8a8ace9
10 changed files with 216 additions and 9 deletions
10
main.go
10
main.go
|
@ -10,7 +10,6 @@ import (
|
|||
"os"
|
||||
|
||||
log "github.com/sirupsen/logrus"
|
||||
"gorm.io/driver/sqlite"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
|
@ -102,11 +101,16 @@ func main() {
|
|||
}
|
||||
|
||||
// connect to the database
|
||||
db, err := gorm.Open(sqlite.Open(*databaseFileName), &gorm.Config{})
|
||||
var db *gorm.DB
|
||||
if config.HasDatabase() {
|
||||
db, err = NewDatabaseFromConfig(config.DatabaseConfig)
|
||||
} else {
|
||||
db, err = NewDatabaseFromFile(*databaseFileName)
|
||||
}
|
||||
if err != nil {
|
||||
log.Fatalf("cannot connect to database: %s", err)
|
||||
}
|
||||
log.Debugf("connected to database %s", *databaseFileName)
|
||||
log.Debugf("connected to database")
|
||||
|
||||
// create tables
|
||||
if err := db.AutoMigrate(&Product{}); err != nil {
|
||||
|
|
Reference in a new issue