Batman

1 files changed, 18 insertions, 0 deletions

diff --git a/internal/storage/db.go b/internal/storage/db.go
new file mode 100644
index 0000000..06fe31c
--- /dev/null
+++ b/internal/storage/db.go
@@ -0,0 +1,18 @@
+package storage
+
+import (
+	"context"
+	"database/sql"
+	"log/slog"
+
+	"github.com/ChausseBenjamin/rafta/internal/util"
+)
+
+func GetDB(ctx context.Context) *sql.DB {
+	db, ok := ctx.Value(util.DBKey).(*sql.DB)
+	if !ok {
+		slog.Error("Unable to retrieve database from context")
+		return nil
+	}
+	return db
+}