init.go 1.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364
  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("@every 5m", func() {
  43. contactForm.analyze()
  44. })
  45. //l = NewCronTask(context.Background(), serverCtx)
  46. //c.AddFunc("* * * * *", func() {
  47. // MaxWorker := 10
  48. // MaxChannel := 3
  49. // l.compApiCallback(MaxWorker, MaxChannel)
  50. //})
  51. }