From cc2c6004babae10366908e5ab5e75f3c9db2920d Mon Sep 17 00:00:00 2001 From: Pea Nut <peanut2@systemli.org> Date: Fri, 15 Nov 2024 18:50:31 +0100 Subject: [PATCH] Check for empty dburi in OpenDatabase --- storage/db.go | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/storage/db.go b/storage/db.go index 76985e8..c2f9fb0 100644 --- a/storage/db.go +++ b/storage/db.go @@ -1,6 +1,7 @@ package storage import ( + "errors" "fmt" "strings" @@ -10,6 +11,11 @@ import ( ) func OpenDatabase(dburi string) (*sqlx.DB, error) { + + if len(dburi) == 0 { + return nil, errors.New("Could not open database (dburi is empty)") + } + // https://www.sqlite.org/wal.html // https://www.sqlite.org/foreignkeys.html if !strings.Contains(dburi, "?") { @@ -25,9 +31,8 @@ func OpenDatabase(dburi string) (*sqlx.DB, error) { db.SetMaxOpenConns(1) _, err = db.Exec(`CREATE TABLE IF NOT EXISTS tokens ( - key TEXT PRIMARY KEY NOT NULL, - buckets TEXT NOT NULL - )`) + key TEXT PRIMARY KEY NOT NULL, + buckets TEXT NOT NULL)`) if err != nil { return nil, fmt.Errorf("Error creating tokens table: %w", err) -- GitLab