From 9ff1a88233d051d177aed3d9862de6b07808a34f Mon Sep 17 00:00:00 2001
From: ziming <zi__mm@163.com>
Date: Fri, 4 Jul 2025 11:13:44 +0800
Subject: [PATCH] =?UTF-8?q?=E8=84=9A=E6=9C=AC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 internal/biz/timeslicequery/execute.go | 1 +
 internal/biz/timeslicequery/mq.go      | 1 +
 internal/pkg/timeslice/model.go        | 1 +
 3 files changed, 3 insertions(+)

diff --git a/internal/biz/timeslicequery/execute.go b/internal/biz/timeslicequery/execute.go
index c0f0059..5ac2ae5 100644
--- a/internal/biz/timeslicequery/execute.go
+++ b/internal/biz/timeslicequery/execute.go
@@ -47,6 +47,7 @@ func (v *Query) callbackFunc(ctx context.Context, req *timeslice.Task) error {
 		StartTime: currentStartTimeStr,
 		EndTime:   currentEndTimeStr,
 		ProductNo: req.Process.Manager.ProductNo,
+		BatchNo:   req.Process.Manager.BatchNo,
 	}
 
 	num := 0
diff --git a/internal/biz/timeslicequery/mq.go b/internal/biz/timeslicequery/mq.go
index 65d8462..5490938 100644
--- a/internal/biz/timeslicequery/mq.go
+++ b/internal/biz/timeslicequery/mq.go
@@ -85,6 +85,7 @@ func (v *Query) getManager(msg string) (*timeslice.Manager, error) {
 		StartTime:      start,
 		EndTime:        end,
 		ProductNo:      req.ProductNo,
+		BatchNo:        req.BatchNo,
 		GoNum:          timeslice.DefaultGoNum,          // 协程数量
 		TimeSliceHours: timeslice.DefaultTimeSliceHours, // 时间间隔
 	}
diff --git a/internal/pkg/timeslice/model.go b/internal/pkg/timeslice/model.go
index 30c8173..10f579d 100644
--- a/internal/pkg/timeslice/model.go
+++ b/internal/pkg/timeslice/model.go
@@ -9,6 +9,7 @@ type Manager struct {
 	StartTime      time.Time // 开始时间
 	EndTime        time.Time // 结束时间
 	ProductNo      string    // 产品编号
+	BatchNo        string    // 微信批次号
 	GoNum          int       // 并发数
 	TimeSliceHours int64     // 时间片"小时"
 }