feat: add custom date flag

tasks.go

@@ -10,14 +10,18 @@ import (
 	"github.com/ShugetsuSoft/pixivel-back/modules/responser/task"
 	"github.com/ShugetsuSoft/pixivel-back/tools/incrementor"
 	"log"
+	"time"
 )
 
 func main() {
 	ctx, cancel := context.WithCancel(context.Background())
 	defer cancel()
 
 	var configPath string
+	var date string
 	flag.StringVar(&configPath, "config", "config.yaml", "Config File Path")
+	flag.StringVar(&date, "date", "", "Specify rank date")
+
 	flag.Parse()
 
 	conf, err := modules.ReadConfig(configPath)
@@ -49,8 +53,16 @@ func main() {
 	ope := operations.NewDatabaseOperations(ctx, db, ft, es, ndb)
 
 	taskgen := task.NewTaskGenerator(mq, config.CrawlTaskQueue, conf.General.SpiderRetry, tracer)
-
-	err = incrementor.CrawlRank(taskgen, ope)
+	var dateobj time.Time
+	if date != "" {
+		dateobj = time.Now().AddDate(0, 0, -2)
+	} else {
+		dateobj, err = time.Parse("20060102", date)
+		if err != nil {
+			log.Fatal(err)
+		}
+	}
+	err = incrementor.CrawlRank(taskgen, ope, dateobj)
 	if err != nil {
 		log.Fatal(err)
 	}

tools/incrementor/incrementor.go

@@ -8,7 +8,7 @@ import (
 	"github.com/ShugetsuSoft/pixivel-back/modules/responser/task"
 )
 
-func CrawlRank(taskgen *task.TaskGenerator, ope *operations.DatabaseOperations) error {
+func CrawlRank(taskgen *task.TaskGenerator, ope *operations.DatabaseOperations, date time.Time) error {
 	ctx := context.Background()
 
 	contents := map[string][]string{
@@ -17,7 +17,7 @@ func CrawlRank(taskgen *task.TaskGenerator, ope *operations.DatabaseOperations)
 		"manga":  {"daily", "weekly", "monthly", "rookie"},
 		"ugoira": {"daily", "weekly"},
 	}
-	today := time.Now().AddDate(0, 0, -2).Format("20060102")
+	today := date.Format("20060102")
 	for content := range contents {
 		for index := range contents[content] {
 			exist, err := ope.IsRankExist(ctx, contents[content][index], today, content)