go-waku/waku/persistence/utils/db.go

66 lines
1.5 KiB
Go
Raw Normal View History

2023-07-06 21:40:57 +00:00
package utils
2023-01-04 17:58:14 +00:00
import (
"database/sql"
"errors"
"regexp"
"strings"
"github.com/waku-org/go-waku/waku/persistence/postgres"
"github.com/waku-org/go-waku/waku/persistence/sqlite"
"go.uber.org/zap"
2023-01-04 17:58:14 +00:00
)
func validateDBUrl(val string) error {
matched, err := regexp.Match(`^[\w\+]+:\/\/[\w\/\\\.\:\@]+\?{0,1}.*$`, []byte(val))
if !matched || err != nil {
return errors.New("invalid db url option format")
}
return nil
}
// DBSettings hold db specific configuration settings required during the db initialization
type DBSettings struct {
2023-08-09 17:14:02 +00:00
Vacuum bool
}
2023-07-06 21:40:57 +00:00
// ExtractDBAndMigration will return a database connection, and migration function that should be used depending on a database connection string
func ExtractDBAndMigration(databaseURL string, dbSettings DBSettings, logger *zap.Logger) (*sql.DB, func(*sql.DB) error, error) {
2023-01-04 17:58:14 +00:00
var db *sql.DB
var migrationFn func(*sql.DB) error
var err error
logger = logger.Named("db-setup")
2023-01-04 17:58:14 +00:00
dbURL := ""
if databaseURL != "" {
err := validateDBUrl(databaseURL)
if err != nil {
return nil, nil, err
}
dbURL = databaseURL
} else {
// In memoryDB
2023-02-17 03:03:22 +00:00
dbURL = "sqlite3://:memory:"
2023-01-04 17:58:14 +00:00
}
dbURLParts := strings.Split(dbURL, "://")
dbEngine := dbURLParts[0]
dbParams := dbURLParts[1]
switch dbEngine {
case "sqlite3":
2023-08-09 17:14:02 +00:00
db, migrationFn, err = sqlite.NewDB(dbParams, dbSettings.Vacuum, logger)
2023-01-04 17:58:14 +00:00
case "postgresql":
2023-08-09 17:14:02 +00:00
db, migrationFn, err = postgres.NewDB(dbURL, dbSettings.Vacuum, logger)
2023-01-04 17:58:14 +00:00
default:
err = errors.New("unsupported database engine")
}
if err != nil {
return nil, nil, err
}
return db, migrationFn, nil
}