1 files changed, 23 insertions, 0 deletions
diff --git a/internal/context/db.go b/internal/context/db.go
new file mode 100644
index 0000000..a9f53d3
--- /dev/null
+++ b/internal/context/db.go
@@ -0,0 +1,23 @@
+package context
+
+import (
+	"context"
+
+	"punchcard/internal/queries"
+)
+
+type dbContextKey struct{}
+
+// WithDB returns a new context with the database queries instance
+func WithDB(ctx context.Context, q *queries.Queries) context.Context {
+	return context.WithValue(ctx, dbContextKey{}, q)
+}
+
+// GetDB retrieves the database queries instance from context
+func GetDB(ctx context.Context) *queries.Queries {
+	if q, ok := ctx.Value(dbContextKey{}).(*queries.Queries); ok {
+		return q
+	}
+	return nil
+}
+