From edee30f156cfda5a8db2e0a970b90404fcdee257 Mon Sep 17 00:00:00 2001
From: ziming <zi__mm@163.com>
Date: Wed, 26 Mar 2025 15:06:19 +0800
Subject: [PATCH] ctx

---
 internal/data/db.go           | 4 ++--
 internal/service/cmb_order.go | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/internal/data/db.go b/internal/data/db.go
index 1e5c45d..a18626a 100644
--- a/internal/data/db.go
+++ b/internal/data/db.go
@@ -15,6 +15,6 @@ func NewDb(db *GormDb) *Db {
 	}
 }
 
-func (d *Db) DB(ctx context.Context) *gorm.DB {
-	return d.db.Client.WithContext(ctx)
+func (d *Db) DB(_ context.Context) *gorm.DB {
+	return d.db.Client
 }
diff --git a/internal/service/cmb_order.go b/internal/service/cmb_order.go
index f641bf9..a3867c8 100644
--- a/internal/service/cmb_order.go
+++ b/internal/service/cmb_order.go
@@ -49,7 +49,7 @@ func (c *CmbService) OrderFail(ctx context.Context, err error) (*v1.CmbReply, er
 
 func (c *CmbService) Order(ctx context.Context, request *v1.CmbRequest) (*v1.CmbReply, error) {
 
-	ctx2, cancel := context.WithCancel(ctx)
+	ctx2, cancel := context.WithTimeout(ctx, 10*time.Second)
 	defer cancel()
 
 	orderNo, err := c.order(ctx2, request)