mirror of
https://github.com/muety/wakapi.git
synced 2023-08-10 21:12:56 +03:00
97 lines
2.3 KiB
Go
97 lines
2.3 KiB
Go
|
package services
|
|||
|
|
|||
|
import (
|
|||
|
"github.com/emvi/logbuch"
|
|||
|
"github.com/go-co-op/gocron"
|
|||
|
"github.com/muety/wakapi/config"
|
|||
|
"github.com/muety/wakapi/models"
|
|||
|
"sync"
|
|||
|
"time"
|
|||
|
)
|
|||
|
|
|||
|
var reportLock = sync.Mutex{}
|
|||
|
|
|||
|
type ReportService struct {
|
|||
|
config *config.Config
|
|||
|
summaryService ISummaryService
|
|||
|
userService IUserService
|
|||
|
mailService IMailService
|
|||
|
schedulersWeekly map[string]*gocron.Scheduler // user id -> scheduler
|
|||
|
}
|
|||
|
|
|||
|
func NewReportService(summaryService ISummaryService, userService IUserService, mailService IMailService) *ReportService {
|
|||
|
return &ReportService{
|
|||
|
config: config.Get(),
|
|||
|
summaryService: summaryService,
|
|||
|
userService: userService,
|
|||
|
mailService: mailService,
|
|||
|
schedulersWeekly: map[string]*gocron.Scheduler{},
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
func (srv *ReportService) Schedule() {
|
|||
|
logbuch.Info("initializing report service")
|
|||
|
|
|||
|
users, err := srv.userService.GetAllByReports(true)
|
|||
|
if err != nil {
|
|||
|
config.Log().Fatal("%v", err)
|
|||
|
}
|
|||
|
|
|||
|
logbuch.Info("scheduling reports for %d users", len(users))
|
|||
|
for _, u := range users {
|
|||
|
srv.UpdateUserSchedule(u)
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
func (srv *ReportService) UpdateUserSchedule(u *models.User) {
|
|||
|
reportLock.Lock()
|
|||
|
defer reportLock.Unlock()
|
|||
|
|
|||
|
// unschedule
|
|||
|
if s, ok := srv.schedulersWeekly[u.ID]; ok && !u.ReportsWeekly {
|
|||
|
s.Stop()
|
|||
|
s.Clear()
|
|||
|
delete(srv.schedulersWeekly, u.ID)
|
|||
|
return
|
|||
|
}
|
|||
|
|
|||
|
// schedule
|
|||
|
if _, ok := srv.schedulersWeekly[u.ID]; !ok && u.ReportsWeekly {
|
|||
|
s := gocron.NewScheduler(u.TZ())
|
|||
|
s.
|
|||
|
Every(1).
|
|||
|
Week().
|
|||
|
Weekday(srv.config.App.GetWeeklyReportDay()).
|
|||
|
At(srv.config.App.GetWeeklyReportTime()).
|
|||
|
Do(srv.Run, u, 7*24*time.Hour)
|
|||
|
s.StartAsync()
|
|||
|
srv.schedulersWeekly[u.ID] = s
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
func (srv *ReportService) Run(user *models.User, duration time.Duration) error {
|
|||
|
end := time.Now().In(user.TZ())
|
|||
|
start := time.Now().Add(-1 * duration)
|
|||
|
|
|||
|
summary, err := srv.summaryService.Aliased(start, end, user, srv.summaryService.Retrieve, false)
|
|||
|
if err != nil {
|
|||
|
config.Log().Error("failed to generate report for '%s' – %v", user.ID, err)
|
|||
|
return err
|
|||
|
}
|
|||
|
|
|||
|
report := &models.Report{
|
|||
|
From: start,
|
|||
|
To: end,
|
|||
|
User: user,
|
|||
|
Summary: summary,
|
|||
|
}
|
|||
|
|
|||
|
if err := srv.mailService.SendReport(user, report); err != nil {
|
|||
|
config.Log().Error("failed to send report for '%s' – %v", user.ID, err)
|
|||
|
return err
|
|||
|
}
|
|||
|
|
|||
|
logbuch.Info("sent report to user '%s'", user.ID)
|
|||
|
return nil
|
|||
|
}
|