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.SetOutput(io.MultiWriter(w, os.Stdout))
|
||||||
log.SetFormatter(logFormatter)
|
log.SetFormatter(logFormatter)
|
||||||
level, err := log.ParseLevel(config.LogLevel)
|
level, err := log.ParseLevel(config.LogLevel)
|
||||||
|
if err != nil {
|
||||||
|
log.SetLevel(log.DebugLevel)
|
||||||
|
}
|
||||||
log.SetLevel(level)
|
log.SetLevel(level)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -56,7 +58,7 @@ var (
|
||||||
func init() {
|
func init() {
|
||||||
_, err := os.Stat(`./config/`)
|
_, err := os.Stat(`./config/`)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
os.Mkdir("./config/", 0666)
|
os.Mkdir("./config/", 0666) //nolint:errcheck
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -76,8 +78,16 @@ func main() {
|
||||||
log.Infoln(fmt.Sprintf("将等待%d秒后启动程序", config.StartWait))
|
log.Infoln(fmt.Sprintf("将等待%d秒后启动程序", config.StartWait))
|
||||||
time.Sleep(time.Second * time.Duration(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("已采用定时执行模式")
|
log.Infoln("已采用定时执行模式")
|
||||||
c := cron.New()
|
c := cron.New()
|
||||||
_, err := c.AddFunc(config.Cron, func() {
|
_, err := c.AddFunc(config.Cron, func() {
|
||||||
|
@ -96,7 +106,18 @@ func main() {
|
||||||
}
|
}
|
||||||
c.Start()
|
c.Start()
|
||||||
select {}
|
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交互模式")
|
log.Infoln("已采用tg交互模式")
|
||||||
telegram := lib.Telegram{
|
telegram := lib.Telegram{
|
||||||
Token: config.TG.Token,
|
Token: config.TG.Token,
|
||||||
|
@ -105,14 +126,16 @@ func main() {
|
||||||
}
|
}
|
||||||
telegram.Init()
|
telegram.Init()
|
||||||
select {}
|
select {}
|
||||||
default:
|
}()
|
||||||
|
}
|
||||||
|
|
||||||
|
if !config.TG.Enable && config.Cron == "" {
|
||||||
log.Infoln("已采用普通学习模式")
|
log.Infoln("已采用普通学习模式")
|
||||||
do()
|
do()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func do() {
|
func do() {
|
||||||
|
|
||||||
defer func() {
|
defer func() {
|
||||||
err := recover()
|
err := recover()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|
Loading…
Reference in New Issue