Skip to content
Snippets Groups Projects
Verified Commit ffb39b7a authored by Ernesto Rico Schmidt's avatar Ernesto Rico Schmidt
Browse files

feat: add support for SQLite database

parent 92ab6474
No related branches found
No related tags found
No related merge requests found
......@@ -22,11 +22,11 @@ import (
"context"
"errors"
"fmt"
"gorm.io/gorm/logger"
"os"
"gorm.io/driver/postgres"
"gorm.io/driver/sqlite"
"gorm.io/gorm"
"gorm.io/gorm/logger"
"os"
)
type Store struct {
......@@ -58,14 +58,16 @@ func postgresConnectionString(cfg DBConfig) (conn string) {
}
func openDB(cfg DBConfig) (db *gorm.DB, err error) {
var conn string
switch cfg.Type {
case "postgres":
conn = postgresConnectionString(cfg)
conn := postgresConnectionString(cfg)
if db, err = gorm.Open(postgres.Open(conn), &gorm.Config{Logger: logger.Default.LogMode(cfg.LogLevel)}); err != nil {
return nil, errors.New("failed to connect to database: " + err.Error())
}
case "sqlite3":
if db, err = gorm.Open(sqlite.Open(":memory:"), &gorm.Config{}); err != nil {
return nil, errors.New("failed to open database: " + err.Error())
}
default:
return nil, errors.New("unknown database engine: " + cfg.Type)
}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment