tg模式和定时模式不再冲突
This commit is contained in:
parent
62d0e1c1f1
commit
326585bcdb
37
main.go
37
main.go
|
@ -35,7 +35,9 @@ func init() {
|
|||
log.SetOutput(io.MultiWriter(w, os.Stdout))
|
||||
log.SetFormatter(logFormatter)
|
||||
level, err := log.ParseLevel(config.LogLevel)
|
||||
|
||||
if err != nil {
|
||||
log.SetLevel(log.DebugLevel)
|
||||
}
|
||||
log.SetLevel(level)
|
||||
}
|
||||
|
||||
|
@ -56,7 +58,7 @@ var (
|
|||
func init() {
|
||||
_, err := os.Stat(`./config/`)
|
||||
if err != nil {
|
||||
os.Mkdir("./config/", 0666)
|
||||
os.Mkdir("./config/", 0666) //nolint:errcheck
|
||||
return
|
||||
}
|
||||
}
|
||||
|
@ -76,8 +78,16 @@ func main() {
|
|||
log.Infoln(fmt.Sprintf("将等待%d秒后启动程序", config.StartWait))
|
||||
time.Sleep(time.Second * time.Duration(config.StartWait))
|
||||
}
|
||||
switch {
|
||||
case config.Cron != "":
|
||||
|
||||
if config.Cron != "" {
|
||||
go func() {
|
||||
defer func() {
|
||||
err := recover()
|
||||
if err != nil {
|
||||
log.Errorln("定时任务执行出现问题")
|
||||
log.Errorln(err)
|
||||
}
|
||||
}()
|
||||
log.Infoln("已采用定时执行模式")
|
||||
c := cron.New()
|
||||
_, err := c.AddFunc(config.Cron, func() {
|
||||
|
@ -96,7 +106,18 @@ func main() {
|
|||
}
|
||||
c.Start()
|
||||
select {}
|
||||
case config.TG.Enable:
|
||||
}()
|
||||
}
|
||||
|
||||
if config.TG.Enable {
|
||||
go func() {
|
||||
defer func() {
|
||||
err := recover()
|
||||
if err != nil {
|
||||
log.Errorln("TG模式执行出现问题")
|
||||
log.Errorln(err)
|
||||
}
|
||||
}()
|
||||
log.Infoln("已采用tg交互模式")
|
||||
telegram := lib.Telegram{
|
||||
Token: config.TG.Token,
|
||||
|
@ -105,14 +126,16 @@ func main() {
|
|||
}
|
||||
telegram.Init()
|
||||
select {}
|
||||
default:
|
||||
}()
|
||||
}
|
||||
|
||||
if !config.TG.Enable && config.Cron == "" {
|
||||
log.Infoln("已采用普通学习模式")
|
||||
do()
|
||||
}
|
||||
}
|
||||
|
||||
func do() {
|
||||
|
||||
defer func() {
|
||||
err := recover()
|
||||
if err != nil {
|
||||
|
|
Loading…
Reference in New Issue