1
0
mirror of https://github.com/muety/wakapi.git synced 2023-08-10 21:12:56 +03:00
wakapi/services/report.go

122 lines
3.1 KiB
Go
Raw Normal View History

2021-04-30 15:07:14 +03:00
package services
import (
"github.com/emvi/logbuch"
"github.com/go-co-op/gocron"
2021-04-30 17:20:08 +03:00
"github.com/leandro-lugaresi/hub"
2021-04-30 15:07:14 +03:00
"github.com/muety/wakapi/config"
"github.com/muety/wakapi/models"
"sync"
"time"
)
var reportLock = sync.Mutex{}
type ReportService struct {
config *config.Config
2021-04-30 17:20:08 +03:00
eventBus *hub.Hub
2021-04-30 15:07:14 +03:00
summaryService ISummaryService
userService IUserService
mailService IMailService
schedulersWeekly map[string]*gocron.Scheduler // user id -> scheduler
}
func NewReportService(summaryService ISummaryService, userService IUserService, mailService IMailService) *ReportService {
2021-04-30 17:20:08 +03:00
srv := &ReportService{
2021-04-30 15:07:14 +03:00
config: config.Get(),
2021-04-30 17:20:08 +03:00
eventBus: config.EventBus(),
2021-04-30 15:07:14 +03:00
summaryService: summaryService,
userService: userService,
mailService: mailService,
schedulersWeekly: map[string]*gocron.Scheduler{},
}
2021-04-30 17:20:08 +03:00
sub := srv.eventBus.Subscribe(0, config.EventUserUpdate)
go func(sub *hub.Subscription) {
for m := range sub.Receiver {
srv.SyncSchedule(m.Fields[config.FieldPayload].(*models.User))
}
}(&sub)
return srv
2021-04-30 15:07:14 +03:00
}
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 {
2021-04-30 17:20:08 +03:00
srv.SyncSchedule(u)
2021-04-30 15:07:14 +03:00
}
}
2021-04-30 17:20:08 +03:00
// SyncSchedule syncs the currently active schedulers with the user's wish about whether or not to receive reports.
// Returns whether a scheduler is active after this operation has run.
func (srv *ReportService) SyncSchedule(u *models.User) bool {
2021-04-30 15:07:14 +03:00
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)
2021-04-30 17:20:08 +03:00
return false
2021-04-30 15:07:14 +03:00
}
// 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
}
2021-04-30 17:20:08 +03:00
return u.ReportsWeekly
2021-04-30 15:07:14 +03:00
}
func (srv *ReportService) Run(user *models.User, duration time.Duration) error {
2021-04-30 17:20:08 +03:00
if user.Email == "" {
logbuch.Warn("not generating report for '%s' as no e-mail address is set")
}
if !srv.SyncSchedule(user) {
logbuch.Info("reports for user '%s' were turned off in the meanwhile since last report job ran")
return nil
}
2021-04-30 15:07:14 +03:00
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
}