init.go 1.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
  1. package crontask
  2. import (
  3. "context"
  4. "wechat-api/internal/svc"
  5. "github.com/robfig/cron/v3"
  6. "github.com/zeromicro/go-zero/core/logx"
  7. )
  8. type CronTask struct {
  9. logx.Logger
  10. ctx context.Context
  11. svcCtx *svc.ServiceContext
  12. }
  13. func NewCronTask(ctx context.Context, svcCtx *svc.ServiceContext) *CronTask {
  14. return &CronTask{
  15. Logger: logx.WithContext(ctx),
  16. ctx: ctx,
  17. svcCtx: svcCtx,
  18. }
  19. }
  20. func ScheduleRun(c *cron.Cron, serverCtx *svc.ServiceContext) {
  21. l := NewCronTask(context.Background(), serverCtx)
  22. c.AddFunc("* * * * *", func() {
  23. l.sendMsg()
  24. })
  25. sendWx := NewCronTask(context.Background(), serverCtx)
  26. c.AddFunc("* * * * *", func() {
  27. sendWx.sendWx()
  28. })
  29. sendWxOnTimeout := NewCronTask(context.Background(), serverCtx)
  30. c.AddFunc("* * * * *", func() {
  31. sendWxOnTimeout.sendWxOnTimeout()
  32. })
  33. computeStatistic := NewCronTask(context.Background(), serverCtx)
  34. c.AddFunc("0 * * * *", func() {
  35. computeStatistic.computeStatistic()
  36. })
  37. //syncWx := NewCronTask(context.Background(), serverCtx)
  38. //c.AddFunc("*/30 * * * *", func() {
  39. // syncWx.syncWx()
  40. //})
  41. contactForm := NewCronTask(context.Background(), serverCtx)
  42. c.AddFunc("1 0 * * *", func() {
  43. contactForm.analyze()
  44. })
  45. //contactForm := NewCronTask(context.Background(), serverCtx)
  46. //c.AddFunc("@every 60m", func() {
  47. // contactForm.analyze()
  48. //})
  49. //l = NewCronTask(context.Background(), serverCtx)
  50. //c.AddFunc("* * * * *", func() {
  51. // MaxWorker := 10
  52. // MaxChannel := 3
  53. // l.compApiCallback(MaxWorker, MaxChannel)
  54. //})
  55. }