From 4f71228ff1f148143ecb2c25c44f76714c1b06a9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9D=8E=E5=AD=90=E9=93=AD?= Date: Fri, 21 Mar 2025 16:19:44 +0800 Subject: [PATCH] =?UTF-8?q?=E6=BB=A1=E8=B6=B3=E5=8E=8B=E6=B5=8B=E5=A4=84?= =?UTF-8?q?=E7=90=86?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- internal/data/repoimpl/order.go | 4 +++- internal/data/repoimpl/product.go | 2 +- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/internal/data/repoimpl/order.go b/internal/data/repoimpl/order.go index 8342f1d..14fea1a 100644 --- a/internal/data/repoimpl/order.go +++ b/internal/data/repoimpl/order.go @@ -92,7 +92,7 @@ func (p *OrderRepoImpl) GetByOutBizNo(ctx context.Context, t vo.OrderType, outBi info := &model.Order{} - db := p.DB(ctx) + db := p.db.DB(ctx).WithContext(ctx).Model(model.Order{}) tx := db.Where(model.Order{Type: t.GetValue(), OutBizNo: outBizNo}).First(&info) if tx.Error != nil { @@ -122,9 +122,11 @@ func (p *OrderRepoImpl) GetByID(ctx context.Context, id uint64) (*bo.OrderBo, er tx := p.DB(ctx).Where(model.Order{ID: id}).First(&info) if tx.Error != nil { + if errors.Is(tx.Error, gorm.ErrRecordNotFound) { return nil, err2.ErrorDbNotFound("订单数据不存在") } + return nil, fmt.Errorf("order db fail %w", tx.Error) } diff --git a/internal/data/repoimpl/product.go b/internal/data/repoimpl/product.go index b592885..62a5814 100644 --- a/internal/data/repoimpl/product.go +++ b/internal/data/repoimpl/product.go @@ -85,7 +85,7 @@ func (r *ProductRepoImpl) GetByProductNo(ctx context.Context, productNo string) func (r *ProductRepoImpl) getByProductNo(ctx context.Context, item *model.Product, productNo string) (*model.Product, error) { - db := r.DB(ctx) + db := r.db.DB(ctx).WithContext(ctx).Model(model.Product{}) tx := db.Where(model.Product{ProductNo: productNo}).First(&item) if tx.Error != nil {