test: 采集时间限制

tools/pji/pji_time/main/main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"fmt"
+	"os"
 	"time"
 )
 
@@ -14,13 +15,13 @@ type CollectWindowStruct struct {
 	EndTime   time.Time
 }
 
-func IsTimeAllowed(currentTime time.Time) bool {
+func IsTimeAllowed(currentTime time.Time, startStr, endStr string) bool {
 	// 模拟解析数据
 	cw := CollectWindowStruct{
 		Flag:  1,
 		Days:  []string{},
-		Start: "09:00",
-		End:   "17:00",
+		Start: startStr,
+		End:   endStr,
 	}
 	// 单独解析采集时间
 	startTime, err := time.Parse("15:04", cw.Start)
@@ -76,7 +77,16 @@ func IsTimeAllowed(currentTime time.Time) bool {
 }
 
 func main() {
-	flag := IsTimeAllowed(time.Now())
+	if len(os.Args) == 2 {
+		fmt.Println("接收到的参数为:", os.Args[1:])
+	} else {
+		fmt.Println("参数数量应该为【2】，请输入开始时间和结束时间")
+	}
+	startStr := os.Args[1]
+	endStr := os.Args[2]
+	fmt.Println("输入的开始时间：", startStr)
+	fmt.Println("输入的结束时间：", endStr)
+	flag := IsTimeAllowed(time.Now(), startStr, endStr)
 	if flag {
 		fmt.Println("当前时间段符合规定，允许采集")
 	} else {