56 lines
1 KiB
Go
56 lines
1 KiB
Go
|
package cronservice
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"log"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"github.com/go-co-op/gocron/v2"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type RepoService interface {
|
|||
|
|
ForceUpdate()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type Service struct {
|
|||
|
|
repoService RepoService
|
|||
|
|
|
|||
|
|
scheduler gocron.Scheduler
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func New(repo RepoService) *Service {
|
|||
|
|
scheduler, err := gocron.NewScheduler(gocron.WithLocation(time.UTC))
|
|||
|
|
if err != nil {
|
|||
|
|
log.Fatalf("Не удалось создать планировщик: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return &Service{
|
|||
|
|
scheduler: scheduler,
|
|||
|
|
repoService: repo,
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (s *Service) SetupCronJobs() {
|
|||
|
|
_, err := s.scheduler.NewJob(
|
|||
|
|
gocron.CronJob("* * * * *", false),
|
|||
|
|
gocron.NewTask(func() {
|
|||
|
|
fmt.Println("Cron run!")
|
|||
|
|
s.repoService.ForceUpdate()
|
|||
|
|
}),
|
|||
|
|
)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Printf("Не удалось создать задание cron: %v", err)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (s *Service) Start() {
|
|||
|
|
s.scheduler.Start()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (s *Service) Shutdown() {
|
|||
|
|
err := s.scheduler.Shutdown()
|
|||
|
|
if err != nil {
|
|||
|
|
log.Printf("Не удалось корректно завершить работу планировщика: %v", err)
|
|||
|
|
}
|
|||
|
|
}
|