From 540d873eacb470bbbf88111f6d7c9350259add9c Mon Sep 17 00:00:00 2001
From: ziming <zi__mm@163.com>
Date: Thu, 12 Jun 2025 10:31:57 +0800
Subject: [PATCH] timeSliceQueryPush

---
 internal/pkg/script/script.go      | 8 ++++++++
 internal/pkg/script/script_test.go | 2 +-
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/internal/pkg/script/script.go b/internal/pkg/script/script.go
index 9774a02..6bc68ce 100644
--- a/internal/pkg/script/script.go
+++ b/internal/pkg/script/script.go
@@ -25,7 +25,11 @@ func timeSliceQueryPush(startTime, endTime time.Time, duration time.Duration, re
 
 	// 每指定间隔时间发送一次请求
 	for t := startTime; t.Before(endTime); t = t.Add(duration) {
+
 		end := t.Add(duration) // 计算每次请求的结束时间
+		if end.After(endTime) {
+			end = endTime
+		}
 
 		// 创建请求体
 		requestBody := map[string]any{
@@ -60,7 +64,11 @@ func pushWechatQuery(startTime, endTime time.Time, duration time.Duration, reque
 
 	// 每指定间隔时间发送一次请求
 	for t := startTime; t.Before(endTime); t = t.Add(duration) {
+
 		end := t.Add(duration) // 计算每次请求的结束时间
+		if end.After(endTime) {
+			end = endTime
+		}
 
 		// 创建请求体
 		requestBody := map[string]any{
diff --git a/internal/pkg/script/script_test.go b/internal/pkg/script/script_test.go
index c5f7bee..8fa5d22 100644
--- a/internal/pkg/script/script_test.go
+++ b/internal/pkg/script/script_test.go
@@ -13,7 +13,7 @@ func Test_script(t *testing.T) {
 		return
 	}
 
-	endTime, err := time.Parse(time.DateTime, "2025-05-01 2:00:03")
+	endTime, err := time.Parse(time.DateTime, "2025-05-01 02:00:03")
 	if err != nil {
 		t.Error(err)
 		return