This commit is contained in:
李子铭 2025-03-20 19:06:22 +08:00
parent 345f7c6da4
commit 31c2da9027
5 changed files with 14 additions and 14 deletions

View File

@ -36,7 +36,7 @@ func Test_db(t *testing.T) {
defer wg.Done() defer wg.Done()
var m model.Product var m model.Product
tx := gormDb.Model(model.Product{}).Where(model.Product{ProductNo: "001"}).Find(&m) tx := gormDb.Model(model.Product{}).Where(model.Product{ProductNo: "000"}).First(&m)
if tx.Error != nil { if tx.Error != nil {
if errors.Is(tx.Error, gorm.ErrRecordNotFound) { if errors.Is(tx.Error, gorm.ErrRecordNotFound) {
t.Errorf("未找到记录") t.Errorf("未找到记录")

View File

@ -2,6 +2,7 @@ package repoimpl
import ( import (
"context" "context"
"errors"
"fmt" "fmt"
"github.com/go-kratos/kratos/v2/log" "github.com/go-kratos/kratos/v2/log"
"gorm.io/gorm" "gorm.io/gorm"
@ -83,18 +84,17 @@ func (p *OrderRepoImpl) Create(ctx context.Context, req *bo.OrderBo) (*bo.OrderB
} }
func (p *OrderRepoImpl) GetByOutBizNo(ctx context.Context, t vo.OrderType, outBizNo string) (*bo.OrderBo, error) { func (p *OrderRepoImpl) GetByOutBizNo(ctx context.Context, t vo.OrderType, outBizNo string) (*bo.OrderBo, error) {
ctx, cancel := context.WithTimeout(ctx, 20*time.Second) // 设置查询超时
defer cancel()
info := &model.Order{} info := &model.Order{}
db := p.DB(ctx) db := p.DB(ctx)
tx := db.Where(model.Order{Type: t.GetValue(), OutBizNo: outBizNo}).Find(&info) tx := db.Where(model.Order{Type: t.GetValue(), OutBizNo: outBizNo}).First(&info)
if tx.Error != nil { if tx.Error != nil {
if errors.Is(tx.Error, context.DeadlineExceeded) {
sqlDB, _ := db.DB() sqlDB, _ := db.DB()
log.Warnf("order当前打开连接数:%d,空闲连接数:%d", sqlDB.Stats().OpenConnections, sqlDB.Stats().Idle) log.Warnf("order当前打开连接数:%d,空闲连接数:%d", sqlDB.Stats().OpenConnections, sqlDB.Stats().Idle)
}
return nil, fmt.Errorf("order db fail %w", tx.Error) return nil, fmt.Errorf("order db fail %w", tx.Error)
} }
@ -109,7 +109,7 @@ func (p *OrderRepoImpl) GetByOutBizNo(ctx context.Context, t vo.OrderType, outBi
func (p *OrderRepoImpl) GetByID(ctx context.Context, id uint64) (*bo.OrderBo, error) { func (p *OrderRepoImpl) GetByID(ctx context.Context, id uint64) (*bo.OrderBo, error) {
info := &model.Order{} info := &model.Order{}
tx := p.DB(ctx).Where(model.Order{ID: id}).Find(&info) tx := p.DB(ctx).Where(model.Order{ID: id}).First(&info)
if tx.Error != nil { if tx.Error != nil {
return nil, fmt.Errorf("order db fail %w", tx.Error) return nil, fmt.Errorf("order db fail %w", tx.Error)
@ -125,7 +125,7 @@ func (p *OrderRepoImpl) GetByID(ctx context.Context, id uint64) (*bo.OrderBo, er
func (p *OrderRepoImpl) GetByOrderNo(ctx context.Context, orderNo string) (*bo.OrderBo, error) { func (p *OrderRepoImpl) GetByOrderNo(ctx context.Context, orderNo string) (*bo.OrderBo, error) {
info := &model.Order{} info := &model.Order{}
tx := p.DB(ctx).Where(model.Order{OrderNo: orderNo}).Find(&info) tx := p.DB(ctx).Where(model.Order{OrderNo: orderNo}).First(&info)
if tx.Error != nil { if tx.Error != nil {
return nil, fmt.Errorf("order db fail %w", tx.Error) return nil, fmt.Errorf("order db fail %w", tx.Error)
@ -141,7 +141,7 @@ func (p *OrderRepoImpl) GetByOrderNo(ctx context.Context, orderNo string) (*bo.O
func (p *OrderRepoImpl) GetByMBV(ctx context.Context, merchantNo, batchNo, voucherNo string) (*bo.OrderBo, error) { func (p *OrderRepoImpl) GetByMBV(ctx context.Context, merchantNo, batchNo, voucherNo string) (*bo.OrderBo, error) {
info := &model.Order{} info := &model.Order{}
tx := p.DB(ctx).Where(model.Order{MerchantNo: merchantNo, BatchNo: batchNo, VoucherNo: voucherNo}).Find(&info) tx := p.DB(ctx).Where(model.Order{MerchantNo: merchantNo, BatchNo: batchNo, VoucherNo: voucherNo}).First(&info)
if tx.Error != nil { if tx.Error != nil {
return nil, fmt.Errorf("db fail %w", tx.Error) return nil, fmt.Errorf("db fail %w", tx.Error)

View File

@ -31,7 +31,7 @@ func (p *OrderNotifyRepoImpl) DB(ctx context.Context) *gorm.DB {
func (p *OrderNotifyRepoImpl) GetByID(ctx context.Context, id uint64) (*bo.OrderNotifyBo, error) { func (p *OrderNotifyRepoImpl) GetByID(ctx context.Context, id uint64) (*bo.OrderNotifyBo, error) {
info := &model.OrderNotify{} info := &model.OrderNotify{}
tx := p.DB(ctx).Where(model.OrderNotify{ID: id}).Find(&info) tx := p.DB(ctx).Where(model.OrderNotify{ID: id}).First(&info)
if tx.Error != nil { if tx.Error != nil {
return nil, tx.Error return nil, tx.Error

View File

@ -85,7 +85,7 @@ func (r *ProductRepoImpl) GetByPNO(ctx context.Context, PNO string) (*bo.Product
func (r *ProductRepoImpl) getByPNO(ctx context.Context, item *model.Product, PNO string) (*model.Product, error) { func (r *ProductRepoImpl) getByPNO(ctx context.Context, item *model.Product, PNO string) (*model.Product, error) {
db := r.DB(ctx) db := r.DB(ctx)
tx := db.Where(model.Product{ProductNo: PNO}).Find(&item) tx := db.Where(model.Product{ProductNo: PNO}).First(&item)
if tx.Error != nil { if tx.Error != nil {
sqlDB, _ := db.DB() sqlDB, _ := db.DB()

View File

@ -31,7 +31,7 @@ func (p *WechatNotifyRegisterTagRepoImpl) DB(ctx context.Context) *gorm.DB {
func (p *WechatNotifyRegisterTagRepoImpl) GetByStockIdAndMchId(ctx context.Context, stockCreatorMchID, stockId string) (*bo.WechatNotifyRegisterTagBo, error) { func (p *WechatNotifyRegisterTagRepoImpl) GetByStockIdAndMchId(ctx context.Context, stockCreatorMchID, stockId string) (*bo.WechatNotifyRegisterTagBo, error) {
info := &model.WechatNotifyRegisterTag{} info := &model.WechatNotifyRegisterTag{}
tx := p.DB(ctx).Where(model.WechatNotifyRegisterTag{StockCreatorMchID: stockCreatorMchID, StockID: stockId}).Find(&info) tx := p.DB(ctx).Where(model.WechatNotifyRegisterTag{StockCreatorMchID: stockCreatorMchID, StockID: stockId}).First(&info)
if tx.Error != nil { if tx.Error != nil {
return nil, tx.Error return nil, tx.Error