From 3ccef3d34500d2d7b5935197336ba8579ab75bc4 Mon Sep 17 00:00:00 2001
From: ziming <zi__mm@163.com>
Date: Wed, 26 Mar 2025 14:53:14 +0800
Subject: [PATCH] ctx

---
 internal/data/db.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/internal/data/db.go b/internal/data/db.go
index a18626a..1e5c45d 100644
--- a/internal/data/db.go
+++ b/internal/data/db.go
@@ -15,6 +15,6 @@ func NewDb(db *GormDb) *Db {
 	}
 }
 
-func (d *Db) DB(_ context.Context) *gorm.DB {
-	return d.db.Client
+func (d *Db) DB(ctx context.Context) *gorm.DB {
+	return d.db.Client.WithContext(ctx)
 }