ai-week/cmd/main.go

package main

import (
	"flag"
	"fmt"
	"github.com/jasonlvhit/gocron"
	"log"
	"time"
	"week/moonshot"
	"week/qywechat"
	"week/tools"
)

var maxRetries = 0
var now = time.Now()
var liveMode bool
var operateTime = 1

func main() {
	log.Println(` 
  _____                                                                        
_/ ____\______  ____   _____   _____    ____   __  _  _______    ____    ____  
\   __\\_  __ \/  _ \ /     \  \__  \  /  _ \  \ \/ \/ /\__  \  /    \  / ___\ 
 |  |   |  | \(  <_> )  Y Y  \  / __ \(  <_> )  \     /  / __ \|   |  \/ /_/  >
 |__|   |__|   \____/|__|_|  / (____  /\____/    \/\_/  (____  /___|  /\___  / 
                           \/       \/                       \/     \//_____/ `)
	fmt.Println("-----------------------------------------------------------------------------------------")
	fmt.Println("-----------------------------------------------------------------------------------------")
	flag.BoolVar(&liveMode, "livemode", false, "即时模式默认关闭")
	flag.Parse()
	if !liveMode {
		log.Println("目前是定时任务模式")
		log.Printf("等待任务的第%v次执行...", operateTime)
		// 定义任务，每周五的五点执行
		gocron.Every(1).Friday().At("17:00").Do(job)
		// 开始定时任务
		<-gocron.Start()
	} else {
		log.Println("目前是即时任务模式")
		job()
	}

}

func job() {
	//最多重试3次
	if maxRetries < 3 {
		nowDate := now.Format("2006-01-02 15:04:05") + " " + now.Weekday().String()
		log.Printf("开始执行任务，当前时间是：%v", nowDate)

		err := moonshot.CreateExcel()
		if err != nil {
			maxRetries++
			log.Printf("执行失败，开始第%v次重试...", maxRetries)
			log.Println("----------------------------------------------------------------------")
			job()

		}
		repContent, err := moonshot.AiChat()
		if err != nil {
			maxRetries++
			log.Printf("执行失败，开始第%v次重试...", maxRetries)
			log.Println("----------------------------------------------------------------------")
			job()
		}
		qywechat.Send(repContent)
		operateTime++
		log.Printf("任务执行成功,当前时间是:%v", nowDate)
		log.Printf("等待%v次任务执行还剩:%v", operateTime, tools.TimeUntilFriday())
	} else {
		log.Fatalln("最大重试次数已到达3，退出")
	}
}