timeSliceQueryPush

internal/pkg/script/script.go

@@ -2,6 +2,7 @@ package script
 
 import (
 	"bytes"
+	"encoding/json"
 	"fmt"
 	"io"
 	"net/http"
@@ -10,25 +11,42 @@ import (
 
 const (
 	URL     = "http://127.0.0.1:15000/voucher/timeSliceQueryPush"
-	DEV_URL = "http://open.cszfan.com/voucher/cmb/timeSliceQueryPush"
+	DEV_URL = "http://open.cszfan.com/voucher/timeSliceQueryPush"
-	PRO_URL = "https://voucher.86698.cn/voucher/cmb/timeSliceQueryPush"
+	PRO_URL = "https://voucher.86698.cn/voucher/timeSliceQueryPush"
 )
 
 const (
-	SINGLE_URL     = "http://127.0.0.1:15000/voucher/pushWechatQuery"
+	SINGLE_URL     = "http://127.0.0.1:15000//voucher/pushWechatQuery"
-	DEV_SINGLE_URL = "http://open.cszfan.com/voucher/cmb/pushWechatQuery"
+	DEV_SINGLE_URL = "http://open.cszfan.com//voucher/pushWechatQuery"
-	PRO_SINGLE_URL = "https://voucher.86698.cn/voucher/cmb/pushWechatQuery"
+	PRO_SINGLE_URL = "https://voucher.86698.cn//voucher/pushWechatQuery"
 )
 
-func script(startTime, endTime time.Time, duration time.Duration, body []byte, URL string) error {
+func timeSliceQueryPush(startTime, endTime time.Time, duration time.Duration, requestURL string) error {
 
 	// 每指定间隔时间发送一次请求
 	for t := startTime; t.Before(endTime); t = t.Add(duration) {
 		end := t.Add(duration) // 计算每次请求的结束时间
 
+		// 创建请求体
+		requestBody := map[string]any{
+			"go_num":           2, // 并发数量
+			"time_slice_hours": 1, // 时间间隔
+			"product_no":       "",
+			"start_time":       t.Format(time.DateTime),
+			"end_time":         end.Format(time.DateTime),
+		}
+
+		// 将请求体转换为 JSON 格式
+		bodyBytes, err := json.Marshal(requestBody)
+		if err != nil {
+			return err
+		}
+
+		fmt.Printf("body:%s\n", string(bodyBytes))
+
 		// 发送请求
-		if err := sendRequest(t, end, body, URL); err != nil {
+		if err2 := sendRequest(bodyBytes, requestURL); err2 != nil {
-			fmt.Printf("Error sending request: %v\n", err)
+			fmt.Printf("Error sending request: %v\n", err2)
 		}
 
 		// 等待一段时间后再发送下一个请求
@@ -38,9 +56,43 @@ func script(startTime, endTime time.Time, duration time.Duration, body []byte, U
 	return nil
 }
 
-func sendRequest(startTime, endTime time.Time, body []byte, URL string) error {
+func pushWechatQuery(startTime, endTime time.Time, duration time.Duration, requestURL string) error {
 
-	resp, err := http.Post(URL, "application/json", bytes.NewBuffer(body))
+	// 每指定间隔时间发送一次请求
+	for t := startTime; t.Before(endTime); t = t.Add(duration) {
+		end := t.Add(duration) // 计算每次请求的结束时间
+
+		// 创建请求体
+		requestBody := map[string]any{
+			"order_no":   "",
+			"product_no": "",
+			"start_time": t.Format(time.DateTime),
+			"end_time":   end.Format(time.DateTime),
+		}
+
+		// 将请求体转换为 JSON 格式
+		bodyBytes, err := json.Marshal(requestBody)
+		if err != nil {
+			return err
+		}
+
+		fmt.Printf("body:%s\n", string(bodyBytes))
+
+		// 发送请求
+		if err2 := sendRequest(bodyBytes, requestURL); err2 != nil {
+			fmt.Printf("Error sending request: %v\n", err2)
+		}
+
+		// 等待一段时间后再发送下一个请求
+		time.Sleep(1 * time.Second) // 可以根据需要调整间隔时间
+	}
+
+	return nil
+}
+
+func sendRequest(body []byte, requestURL string) error {
+
+	resp, err := http.Post(requestURL, "application/json", bytes.NewBuffer(body))
 	if err != nil {
 		return fmt.Errorf("failed to send POST request: %v", err)
 	}

internal/pkg/script/script_test.go

@@ -1,7 +1,6 @@
 package script
 
 import (
-	"encoding/json"
 	"testing"
 	"time"
 )
@@ -14,7 +13,7 @@ func Test_script(t *testing.T) {
 		return
 	}
 
-	endTime, err := time.Parse(time.DateTime, "2025-05-01 5:00:00")
+	endTime, err := time.Parse(time.DateTime, "2025-05-01 2:00:03")
 	if err != nil {
 		t.Error(err)
 		return
@@ -22,23 +21,10 @@ func Test_script(t *testing.T) {
 
 	duration := 5 * time.Hour
 
-	// 创建请求体
+	//requestUrl := URL
-	requestBody := map[string]any{
+	requestUrl := DEV_URL
-		"go_num":           2, // 并发数量
-		"time_slice_hours": 2, // 时间间隔
-		"product_no":       "",
-		"start_time":       startTime.Format(time.DateTime),
-		"end_time":         endTime.Format(time.DateTime),
-	}
 
-	// 将请求体转换为 JSON 格式
+	if err = timeSliceQueryPush(startTime, endTime, duration, requestUrl); err != nil {
-	bodyBytes, err := json.Marshal(requestBody)
-	if err != nil {
-		t.Error(err)
-		return
-	}
-
-	if err = script(startTime, endTime, duration, bodyBytes, URL); err != nil {
 		t.Error(err)
 	}
 }
@@ -59,22 +45,9 @@ func Test_script2(t *testing.T) {
 
 	duration := 50 * time.Hour
 
-	// 创建请求体
+	requestUrl := SINGLE_URL
-	requestBody := map[string]any{
-		"order_no":   "",
-		"product_no": "",
-		"start_time": startTime.Format(time.DateTime),
-		"end_time":   endTime.Format(time.DateTime),
-	}
 
-	// 将请求体转换为 JSON 格式
+	if err = pushWechatQuery(startTime, endTime, duration, requestUrl); err != nil {
-	bodyBytes, err := json.Marshal(requestBody)
-	if err != nil {
-		t.Error(err)
-		return
-	}
-
-	if err = script(startTime, endTime, duration, bodyBytes, SINGLE_URL); err != nil {
 		t.Error(err)
 	}
 }