Przeglądaj źródła

:add 增加id号可配置 功能

fly 1 miesiąc temu
rodzic
commit
8d53d0a814
2 zmienionych plików z 20 dodań i 2 usunięć
  1. 3 0
      config.txt
  2. 17 2
      main.go

+ 3 - 0
config.txt

@@ -0,0 +1,3 @@
+20251204
+20251205
+20251208

+ 17 - 2
main.go

@@ -4,13 +4,16 @@ import (
 	"errors"
 	"github.com/sirupsen/logrus"
 	"github.com/xuri/excelize/v2"
+	"os"
 	"strconv"
+	"strings"
 	"time"
 )
 
 var (
 	log         = logrus.WithField("module", "main")
 	excelHandle *excelize.File
+	configFile  = "config.txt"
 	ids         = []string{"20251204", "20251205", "20251208"}
 	startId     = 16127287
 	endId       = 16127485
@@ -34,8 +37,20 @@ func init() {
 	//}
 	//logrus.SetOutput(logger) // logrus 设置日志的输出方式
 
-	for i := startId; i <= endId; i++ {
-		ids = append(ids, strconv.Itoa(i))
+	file, err := os.ReadFile(configFile)
+	if err != nil {
+		log.Fatalf("Read config file error: %s", err.Error())
+	}
+
+	lines := strings.Split(string(file), "\r\n")
+	//for i := startId; i <= endId; i++ {
+	//	ids = append(ids, strconv.Itoa(i))
+	//}
+	for _, line := range lines {
+		if len(line) < 8 {
+			continue
+		}
+		ids = append(ids, strings.TrimSpace(line))
 	}
 }