voucher/internal/data/gorm_test.go

package data

import (
	"context"
	"errors"
	"fmt"
	"google.golang.org/protobuf/types/known/durationpb"
	"gorm.io/gorm"
	"gorm.io/hints"
	"sync"
	"testing"
	"time"
	"voucher/internal/biz/vo"
	"voucher/internal/conf"
	"voucher/internal/data/model"
)

func Test_gorm_db(t *testing.T) {
	// 镜像mysql
	data := &conf.Data_Database{
		Driver:      "mysql",
		Source:      "root:lsxddb123.@tcp(47.108.53.72:3306)/voucher?parseTime=True&loc=Local",
		MaxIdle:     20,                 // 空闲连接池
		MaxOpen:     200,                // 最大连接数
		MaxLifetime: durationpb.New(60), // 5分钟
		IsDebug:     false,
	}
	//data := &conf.Data_Database{
	//	Driver:      "mysql",
	//	Source:      "root:lansexiongdi6,@tcp(47.97.27.195:3306)/voucher?parseTime=True&loc=Local",
	//	MaxIdle:     2,  // 空闲连接池
	//	MaxOpen:     10, // 最大连接数
	//	MaxLifetime: durationpb.New(60),
	//	IsDebug:     false,
	//}
	gormDb, cleanup := db(data)
	defer cleanup()

	start2 := time.Now()

	concurrency := 5000 // 调整并发数
	errCount := 0

	var wg sync.WaitGroup
	wg.Add(concurrency)

	for i := 0; i < concurrency; i++ {
		go func(i int) {
			defer wg.Done()

			ctx := context.Background()
			var info model.Order
			tx := gormDb.
				WithContext(ctx).
				Model(model.Order{}).
				//Clauses(hints.UseIndex("udx_out_biz_no_type")).
				Where(model.Order{Type: vo.OrderTypeCmb.GetValue(), OutBizNo: fmt.Sprintf("174252390990605ngywYrSAGE83e1%d", i)}).
				First(&info)
			if tx.Error != nil {
				if errors.Is(tx.Error, gorm.ErrRecordNotFound) {
					t.Errorf("未找到记录")
				} else {
					fmt.Printf("请求错误: %v\n", tx.Error)
					errCount += 1
				}
			}
		}(i)
	}

	wg.Wait()
	t.Logf("\n--------------连接请求,总请求耗时: %v,总数: %d, 失败次数: %d--------------\n", time.Since(start2), concurrency, errCount)
}

func Test_db(t *testing.T) {
	maxLifetime := durationpb.New(300) // 5分钟
	data := &conf.Data_Database{
		Driver:      "mysql",
		Source:      "root:lansexiongdi6,@tcp(47.97.27.195:3306)/voucher?parseTime=True&loc=Local",
		MaxIdle:     2,  // 空闲连接池
		MaxOpen:     10, // 最大连接数
		MaxLifetime: maxLifetime,
		IsDebug:     false,
	}
	gormDb, cleanup := db(data)
	defer cleanup()

	start2 := time.Now()

	errCount := 0
	const concurrency = 1

	for i := 0; i < concurrency; i++ {
		ctx := context.Background()

		var info model.Order
		tx := gormDb.
			WithContext(ctx).
			Model(model.Order{}).
			Clauses(hints.ForceIndex("udx_out_biz_no_type")).
			Where(model.Order{Type: vo.OrderTypeCmb.GetValue(), OutBizNo: fmt.Sprintf("174252390990605ngywYrSAGE83e1%d", i)}).
			First(&info)
		if tx.Error != nil {
			if errors.Is(tx.Error, gorm.ErrRecordNotFound) {
				t.Errorf("未找到记录")
			} else {
				fmt.Printf("请求错误: %v\n", tx.Error)
				errCount += 1
			}
		}
	}

	t.Logf("\n--------------连接请求,总请求耗时: %v,总数: %d, 失败次数: %d--------------\n", time.Since(start2), concurrency, errCount)
}

func Test_orderCreate(t *testing.T) {
	//maxLifetime := durationpb.New(300) // 5分钟
	//data := &conf.Data_Database{
	//	Driver:      "mysql",
	//	Source:      "root:lansexiongdi6,@tcp(47.97.27.195:3306)/voucher?parseTime=True&loc=Local",
	//	MaxIdle:     2,  // 空闲连接池
	//	MaxOpen:     10, // 最大连接数
	//	MaxLifetime: maxLifetime,
	//	IsDebug:     false,
	//}
	//gormDb, cleanup := db(data)
	//defer cleanup()
	//
	//now := time.Now()
	//
	//info := &model.Order{
	//	OrderNo:     req.OrderNo,
	//	OutBizNo:    req.OutBizNo,
	//	ProductNo:   req.ProductNo,
	//	BatchNo:     req.BatchNo,
	//	Account:     req.Account,
	//	AccountType: req.AccountType.GetValue(),
	//	Status:      req.Status.GetValue(),
	//	Type:        req.Type.GetValue(),
	//	AppID:       req.AppID,
	//	MerchantNo:  req.MerchantNo,
	//	Channel:     req.Channel.GetValue(),
	//	NotifyUrl:   req.NotifyUrl,
	//	Attach:      req.Attach,
	//	CreateTime:  &now,
	//	UpdateTime:  &now,
	//}
	//
	//if req.ProductNo == "001" {
	//	info.VoucherNo = req.OrderNo
	//	info.Status = vo.OrderStatusSuccess.GetValue()
	//}
	//
	//tx := p.DB(ctx).Create(info)
	//if tx.Error != nil {
	//	return nil, fmt.Errorf("create db fail %w", tx.Error)
	//}
}