From 5389e1a5d26fdbf2441fa5a1e101999e8449b9d1 Mon Sep 17 00:00:00 2001
From: Benjamin Chausse <benjamin@chausse.xyz>
Date: Mon, 3 Feb 2025 01:12:45 -0500
Subject: Batman

---
 internal/storage/db.go | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)
 create mode 100644 internal/storage/db.go

(limited to 'internal/storage/db.go')

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
+}
-- 
cgit v1.2.3