task.go 839 B

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. package do
  2. import (
  3. "analyze/internal/model"
  4. "github.com/gogf/gf/v2/encoding/gjson"
  5. "github.com/gogf/gf/v2/frame/g"
  6. )
  7. type taskInfo struct {
  8. FirstRun bool
  9. TaskRun bool
  10. Ticker int
  11. }
  12. type taskConfig struct {
  13. Name string
  14. MainSwitch bool
  15. MarketAnalysis taskInfo
  16. }
  17. var (
  18. TaskConfig taskConfig
  19. )
  20. func init() {
  21. g.Log().Info(model.Ctx, "--定时任务配置初始化--开始--")
  22. var (
  23. task map[string]interface{}
  24. err error
  25. b []byte
  26. msg = "--定时任务配置初始化--异常--"
  27. )
  28. task, err = g.Cfg("task.yaml").Data(model.Ctx)
  29. if err == nil {
  30. b, err = gjson.Marshal(task)
  31. if err == nil {
  32. if err = gjson.Unmarshal(b, &TaskConfig); err == nil {
  33. msg = ""
  34. }
  35. }
  36. }
  37. if msg != "" {
  38. g.Log().Error(model.Ctx, msg)
  39. }
  40. g.Log().Info(model.Ctx, "--定时任务配置初始化--结束--")
  41. }