cron.go 1002 B

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. package cron
  2. import (
  3. "github.com/0xJacky/Nginx-UI/internal/cert"
  4. "github.com/0xJacky/Nginx-UI/internal/logger"
  5. "github.com/0xJacky/Nginx-UI/internal/logrotate"
  6. "github.com/0xJacky/Nginx-UI/settings"
  7. "github.com/go-co-op/gocron"
  8. "time"
  9. )
  10. var s *gocron.Scheduler
  11. func init() {
  12. s = gocron.NewScheduler(time.UTC)
  13. }
  14. var logrotateJob *gocron.Job
  15. func InitCronJobs() {
  16. job, err := s.Every(30).Minute().SingletonMode().Do(cert.AutoCert)
  17. if err != nil {
  18. logger.Fatalf("AutoCert Job: %v, Err: %v\n", job, err)
  19. }
  20. startLogrotate()
  21. s.StartAsync()
  22. }
  23. func RestartLogrotate() {
  24. logger.Debug("Restart Logrotate")
  25. if logrotateJob != nil {
  26. s.RemoveByReference(logrotateJob)
  27. }
  28. startLogrotate()
  29. }
  30. func startLogrotate() {
  31. if !settings.LogrotateSettings.Enabled {
  32. return
  33. }
  34. var err error
  35. logrotateJob, err = s.Every(settings.LogrotateSettings.Interval).Minute().SingletonMode().Do(logrotate.Exec)
  36. if err != nil {
  37. logger.Fatalf("LogRotate Job: %v, Err: %v\n", logrotateJob, err)
  38. }
  39. }