98 lines
2.7 KiB
Go
98 lines
2.7 KiB
Go
package timeslicequery
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/go-kratos/kratos/v2/log"
|
|
"time"
|
|
"voucher/internal/biz/bo"
|
|
"voucher/internal/biz/do"
|
|
"voucher/internal/pkg/timeslice"
|
|
)
|
|
|
|
func (v *Query) execute(ctx context.Context, req *timeslice.Manager) error {
|
|
|
|
start := time.Now()
|
|
|
|
managerStartStr := req.StartTime.Format(time.DateTime)
|
|
managerEndStr := req.EndTime.Format(time.DateTime)
|
|
|
|
log.Warnf("微信券查询,%s到%s,开始", managerStartStr, managerEndStr)
|
|
fmt.Printf("微信券查询,%s到%s,开始\n", managerStartStr, managerEndStr)
|
|
|
|
taskCount, err := timeslice.NewManager(v.callbackFunc).Run(ctx, req)
|
|
if err != nil {
|
|
log.Errorf("微信券查询,%s到%s,失败:%v", managerStartStr, managerEndStr, err)
|
|
}
|
|
|
|
elapsed := time.Now().Sub(start).String()
|
|
|
|
log.Warnf("微信券查询,%s到%s,总任务数:%d,总耗时:%s", managerStartStr, managerEndStr, taskCount, elapsed)
|
|
fmt.Printf("微信券查询,%s到%s,总任务数:%d,总耗时:%s\n", managerStartStr, managerEndStr, taskCount, elapsed)
|
|
|
|
return nil
|
|
}
|
|
|
|
func (v *Query) callbackFunc(ctx context.Context, req *timeslice.Task) error {
|
|
|
|
startTimeStr := req.Process.Manager.StartTime.Format(time.DateTime)
|
|
endTimeStr := req.Process.Manager.EndTime.Format(time.DateTime)
|
|
|
|
currentStartTimeStr := req.CurrentStartTime.Format(time.DateTime)
|
|
currentEndTimeStr := req.CurrentEndTime.Format(time.DateTime)
|
|
|
|
start := time.Now()
|
|
|
|
bReq := &do.WechatQuery{
|
|
StartTime: currentStartTimeStr,
|
|
EndTime: currentEndTimeStr,
|
|
ProductNo: req.Process.Manager.ProductNo,
|
|
BatchNo: req.Process.Manager.BatchNo,
|
|
}
|
|
|
|
num := 0
|
|
errNum := 0
|
|
useNum := 0
|
|
|
|
err := v.orderRepo.FinSucByStockIdInBatches(ctx, bReq, func(ctx context.Context, rows []*bo.OrderBo) error {
|
|
|
|
for _, order := range rows {
|
|
|
|
num += 1
|
|
if err := v.wechatQuery(ctx, order, &useNum); err != nil {
|
|
|
|
errNum += 1
|
|
|
|
logFields := map[string]string{
|
|
"order_no": order.OrderNo,
|
|
"coupon_id": order.VoucherNo,
|
|
"open_id": order.Account,
|
|
"stock_id": order.BatchNo,
|
|
"err": err.Error(),
|
|
}
|
|
log.Errorf("微信券查询,%s到%s,taskId:%d,错误:%+v", startTimeStr, endTimeStr, req.TaskID, logFields)
|
|
|
|
if errNum > 20 {
|
|
return fmt.Errorf("微信券查询,%s到%s,第%d个任务,已经连续发生20次错误%+v", startTimeStr, endTimeStr, req.TaskID, logFields)
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
return nil
|
|
})
|
|
|
|
end := time.Now()
|
|
|
|
logFields := map[string]any{
|
|
"sTime": currentStartTimeStr + "到" + currentEndTimeStr,
|
|
"num": num,
|
|
"useNum": useNum,
|
|
"errNum": errNum,
|
|
"elapsed": end.Sub(start).String(),
|
|
}
|
|
log.Warnf("微信券查询,%s到%s,taskId:%d,处理完毕:%+v", startTimeStr, endTimeStr, req.TaskID, logFields)
|
|
|
|
return err
|
|
}
|