chore(tfstated): improve error messages and refactored PRAGMA code in the database module

3 files changed, 22 insertions, 20 deletions

diff --git a/.gitignore b/.gitignore
index 122fc54..d824783 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,4 +1,4 @@
-tfstated
+cover.out
 tfstate.db
 tfstate.db-shm
 tfstate.db-wal
diff --git a/go.mod b/go.mod
index 2733025..5cdb0f4 100644
--- a/go.mod
+++ b/go.mod
@@ -2,9 +2,8 @@ module git.adyxax.org/adyxax/tfstated
 
 go 1.23.3
 
-require github.com/mattn/go-sqlite3 v1.14.24
-
 require (
-	go.n16f.net/uuid v0.0.0-20240707135755-e4fd26b968ad // indirect
-	golang.org/x/crypto v0.29.0 // indirect
+	github.com/mattn/go-sqlite3 v1.14.24
+	go.n16f.net/uuid v0.0.0-20240707135755-e4fd26b968ad
+	golang.org/x/crypto v0.29.0
 )
diff --git a/pkg/database/db.go b/pkg/database/db.go
index 0777556..b7f0e89 100644
--- a/pkg/database/db.go
+++ b/pkg/database/db.go
@@ -12,7 +12,7 @@ import (
 func initDB(ctx context.Context, url string) (*sql.DB, error) {
 	db, err := sql.Open("sqlite3", url)
 	if err != nil {
-		return nil, err
+		return nil, fmt.Errorf("failed to open database: %w", err)
 	}
 	defer func() {
 		if err != nil {
@@ -20,7 +20,7 @@ func initDB(ctx context.Context, url string) (*sql.DB, error) {
 		}
 	}()
 	if _, err = db.ExecContext(ctx, "PRAGMA busy_timeout = 5000"); err != nil {
-		return nil, err
+		return nil, fmt.Errorf("failed to set pragma: %w", err)
 	}
 
 	return db, nil
@@ -37,7 +37,7 @@ type DB struct {
 func NewDB(ctx context.Context, url string) (*DB, error) {
 	readDB, err := initDB(ctx, url)
 	if err != nil {
-		return nil, err
+		return nil, fmt.Errorf("failed to init read database connection: %w", err)
 	}
 	defer func() {
 		if err != nil {
@@ -48,7 +48,7 @@ func NewDB(ctx context.Context, url string) (*DB, error) {
 
 	writeDB, err := initDB(ctx, url)
 	if err != nil {
-		return nil, err
+		return nil, fmt.Errorf("failed to init write database connection: %w", err)
 	}
 	defer func() {
 		if err != nil {
@@ -63,20 +63,22 @@ func NewDB(ctx context.Context, url string) (*DB, error) {
 		versionsHistoryLimit: 64,
 		writeDB:              writeDB,
 	}
-	if _, err = db.Exec("PRAGMA foreign_keys = ON"); err != nil {
-		return nil, err
+	pragmas := []struct {
+		key   string
+		value string
+	}{
+		{"foreign_keys", "ON"},
+		{"cache_size", "10000000"},
+		{"journal_mode", "WAL"},
+		{"synchronous", "NORMAL"},
 	}
-	if _, err = db.Exec("PRAGMA cache_size = 10000000"); err != nil {
-		return nil, err
-	}
-	if _, err = db.Exec("PRAGMA journal_mode = WAL"); err != nil {
-		return nil, err
-	}
-	if _, err = db.Exec("PRAGMA synchronous = NORMAL"); err != nil {
-		return nil, err
+	for _, pragma := range pragmas {
+		if _, err = db.Exec(fmt.Sprintf("PRAGMA %s = %s", pragma.key, pragma.value)); err != nil {
+			return nil, fmt.Errorf("failed to set pragma: %w", err)
+		}
 	}
 	if err = db.migrate(); err != nil {
-		return nil, err
+		return nil, fmt.Errorf("failed to migrate: %w", err)
 	}
 
 	return &db, nil
@@ -85,6 +87,7 @@ func NewDB(ctx context.Context, url string) (*DB, error) {
 func (db *DB) Close() error {
 	if err := db.readDB.Close(); err != nil {
 		_ = db.writeDB.Close()
+		return fmt.Errorf("failed to close read database connection: %w", err)
 	}
 	return db.writeDB.Close()
 }