mirror of
https://github.com/muety/wakapi.git
synced 2023-08-10 21:12:56 +03:00
273 lines
8.5 KiB
Go
273 lines
8.5 KiB
Go
package services
|
|
|
|
import (
|
|
"github.com/emvi/logbuch"
|
|
"github.com/leandro-lugaresi/hub"
|
|
"github.com/muety/artifex/v2"
|
|
"github.com/muety/wakapi/config"
|
|
"github.com/muety/wakapi/helpers"
|
|
"github.com/muety/wakapi/models"
|
|
"github.com/muety/wakapi/repositories"
|
|
"github.com/muety/wakapi/utils"
|
|
"github.com/patrickmn/go-cache"
|
|
"reflect"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
type LeaderboardService struct {
|
|
config *config.Config
|
|
cache *cache.Cache
|
|
eventBus *hub.Hub
|
|
repository repositories.ILeaderboardRepository
|
|
summaryService ISummaryService
|
|
userService IUserService
|
|
queueDefault *artifex.Dispatcher
|
|
queueWorkers *artifex.Dispatcher
|
|
}
|
|
|
|
func NewLeaderboardService(leaderboardRepo repositories.ILeaderboardRepository, summaryService ISummaryService, userService IUserService) *LeaderboardService {
|
|
srv := &LeaderboardService{
|
|
config: config.Get(),
|
|
cache: cache.New(6*time.Hour, 6*time.Hour),
|
|
eventBus: config.EventBus(),
|
|
repository: leaderboardRepo,
|
|
summaryService: summaryService,
|
|
userService: userService,
|
|
queueDefault: config.GetDefaultQueue(),
|
|
queueWorkers: config.GetQueue(config.QueueProcessing),
|
|
}
|
|
|
|
onUserUpdate := srv.eventBus.Subscribe(0, config.EventUserUpdate)
|
|
go func(sub *hub.Subscription) {
|
|
for m := range sub.Receiver {
|
|
|
|
// generate leaderboard for updated user, if leaderboard enabled and none present, yet
|
|
user := m.Fields[config.FieldPayload].(*models.User)
|
|
|
|
exists, err := srv.ExistsAnyByUser(user.ID)
|
|
if err != nil {
|
|
config.Log().Error("failed to check existing leaderboards upon user update - %v", err)
|
|
}
|
|
|
|
if user.PublicLeaderboard && !exists {
|
|
logbuch.Info("generating leaderboard for '%s' after settings update", user.ID)
|
|
srv.ComputeLeaderboard([]*models.User{user}, models.IntervalPast7Days, []uint8{models.SummaryLanguage})
|
|
} else if !user.PublicLeaderboard && exists {
|
|
logbuch.Info("clearing leaderboard for '%s' after settings update", user.ID)
|
|
if err := srv.repository.DeleteByUser(user.ID); err != nil {
|
|
config.Log().Error("failed to clear leaderboard for user '%s' - %v", user.ID, err)
|
|
}
|
|
srv.cache.Flush()
|
|
}
|
|
}
|
|
}(&onUserUpdate)
|
|
|
|
return srv
|
|
}
|
|
|
|
func (srv *LeaderboardService) Schedule() {
|
|
logbuch.Info("scheduling leaderboard generation")
|
|
|
|
generate := func() {
|
|
users, err := srv.userService.GetAllByLeaderboard(true)
|
|
if err != nil {
|
|
config.Log().Error("failed to get users for leaderboard generation - %v", err)
|
|
return
|
|
}
|
|
srv.ComputeLeaderboard(users, models.IntervalPast7Days, []uint8{models.SummaryLanguage})
|
|
}
|
|
|
|
for _, cronExp := range srv.config.App.GetLeaderboardGenerationTimeCron() {
|
|
if _, err := srv.queueDefault.DispatchCron(generate, cronExp); err != nil {
|
|
config.Log().Error("failed to schedule leaderboard generation (%s), %v", cronExp, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (srv *LeaderboardService) ComputeLeaderboard(users []*models.User, interval *models.IntervalKey, by []uint8) error {
|
|
logbuch.Info("generating leaderboard (%s) for %d users (%d aggregations)", (*interval)[0], len(users), len(by))
|
|
|
|
for _, user := range users {
|
|
if err := srv.repository.DeleteByUserAndInterval(user.ID, interval); err != nil {
|
|
config.Log().Error("failed to delete leaderboard items for user %s (interval %s) - %v", user.ID, (*interval)[0], err)
|
|
continue
|
|
}
|
|
|
|
item, err := srv.GenerateByUser(user, interval)
|
|
if err != nil {
|
|
config.Log().Error("failed to generate general leaderboard for user %s - %v", user.ID, err)
|
|
continue
|
|
}
|
|
|
|
if err := srv.repository.InsertBatch([]*models.LeaderboardItem{item}); err != nil {
|
|
config.Log().Error("failed to persist general leaderboard for user %s - %v", user.ID, err)
|
|
continue
|
|
}
|
|
|
|
for _, by := range by {
|
|
items, err := srv.GenerateAggregatedByUser(user, interval, by)
|
|
if err != nil {
|
|
config.Log().Error("failed to generate aggregated (by %s) leaderboard for user %s - %v", models.GetEntityColumn(by), user.ID, err)
|
|
continue
|
|
}
|
|
|
|
if len(items) == 0 {
|
|
continue
|
|
}
|
|
|
|
if err := srv.repository.InsertBatch(items); err != nil {
|
|
config.Log().Error("failed to persist aggregated (by %s) leaderboard for user %s - %v", models.GetEntityColumn(by), user.ID, err)
|
|
continue
|
|
}
|
|
}
|
|
}
|
|
|
|
srv.cache.Flush()
|
|
logbuch.Info("finished leaderboard generation")
|
|
return nil
|
|
}
|
|
|
|
func (srv *LeaderboardService) ExistsAnyByUser(userId string) (bool, error) {
|
|
count, err := srv.repository.CountAllByUser(userId)
|
|
return count > 0, err
|
|
}
|
|
|
|
func (srv *LeaderboardService) CountUsers() (int64, error) {
|
|
// check cache
|
|
cacheKey := "count_total"
|
|
if cacheResult, ok := srv.cache.Get(cacheKey); ok {
|
|
return cacheResult.(int64), nil
|
|
}
|
|
|
|
count, err := srv.repository.CountUsers()
|
|
if err != nil {
|
|
srv.cache.SetDefault(cacheKey, count)
|
|
}
|
|
return count, err
|
|
}
|
|
|
|
func (srv *LeaderboardService) GetByInterval(interval *models.IntervalKey, pageParams *utils.PageParams, resolveUsers bool) (models.Leaderboard, error) {
|
|
return srv.GetAggregatedByInterval(interval, nil, pageParams, resolveUsers)
|
|
}
|
|
|
|
func (srv *LeaderboardService) GetByIntervalAndUser(interval *models.IntervalKey, userId string, resolveUser bool) (models.Leaderboard, error) {
|
|
return srv.GetAggregatedByIntervalAndUser(interval, userId, nil, resolveUser)
|
|
}
|
|
|
|
func (srv *LeaderboardService) GetAggregatedByInterval(interval *models.IntervalKey, by *uint8, pageParams *utils.PageParams, resolveUsers bool) (models.Leaderboard, error) {
|
|
// check cache
|
|
cacheKey := srv.getHash(interval, by, "", pageParams)
|
|
if cacheResult, ok := srv.cache.Get(cacheKey); ok {
|
|
return cacheResult.([]*models.LeaderboardItemRanked), nil
|
|
}
|
|
|
|
items, err := srv.repository.GetAllAggregatedByInterval(interval, by, pageParams.Limit(), pageParams.Offset())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if resolveUsers {
|
|
users, err := srv.userService.GetManyMapped(models.Leaderboard(items).UserIDs())
|
|
if err != nil {
|
|
config.Log().Error("failed to resolve users for leaderboard item - %v", err)
|
|
} else {
|
|
for _, item := range items {
|
|
if u, ok := users[item.UserID]; ok {
|
|
item.User = u
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
srv.cache.SetDefault(cacheKey, items)
|
|
return items, nil
|
|
}
|
|
|
|
func (srv *LeaderboardService) GetAggregatedByIntervalAndUser(interval *models.IntervalKey, userId string, by *uint8, resolveUser bool) (models.Leaderboard, error) {
|
|
// check cache
|
|
cacheKey := srv.getHash(interval, by, userId, nil)
|
|
if cacheResult, ok := srv.cache.Get(cacheKey); ok {
|
|
return cacheResult.([]*models.LeaderboardItemRanked), nil
|
|
}
|
|
|
|
items, err := srv.repository.GetAggregatedByUserAndInterval(userId, interval, by, 0, 0)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if resolveUser {
|
|
u, err := srv.userService.GetUserById(userId)
|
|
if err != nil {
|
|
config.Log().Error("failed to resolve user for leaderboard item - %v", err)
|
|
} else {
|
|
for _, item := range items {
|
|
item.User = u
|
|
}
|
|
}
|
|
}
|
|
|
|
srv.cache.SetDefault(cacheKey, items)
|
|
return items, nil
|
|
}
|
|
|
|
func (srv *LeaderboardService) GenerateByUser(user *models.User, interval *models.IntervalKey) (*models.LeaderboardItem, error) {
|
|
err, from, to := helpers.ResolveIntervalTZ(interval, user.TZ())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
summary, err := srv.summaryService.Aliased(from, to, user, srv.summaryService.Retrieve, nil, false)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &models.LeaderboardItem{
|
|
User: user,
|
|
UserID: user.ID,
|
|
Interval: (*interval)[0],
|
|
Total: summary.TotalTime(),
|
|
}, nil
|
|
}
|
|
|
|
func (srv *LeaderboardService) GenerateAggregatedByUser(user *models.User, interval *models.IntervalKey, by uint8) ([]*models.LeaderboardItem, error) {
|
|
err, from, to := helpers.ResolveIntervalTZ(interval, user.TZ())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
summary, err := srv.summaryService.Aliased(from, to, user, srv.summaryService.Retrieve, nil, false)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
summaryItems := *summary.ItemsByType(by)
|
|
items := make([]*models.LeaderboardItem, summaryItems.Len())
|
|
|
|
for i := 0; i < summaryItems.Len(); i++ {
|
|
key := summaryItems[i].Key
|
|
items[i] = &models.LeaderboardItem{
|
|
User: user,
|
|
UserID: user.ID,
|
|
Interval: (*interval)[0],
|
|
By: &by,
|
|
Total: summary.TotalTimeByKey(by, key),
|
|
Key: &key,
|
|
}
|
|
}
|
|
|
|
return items, nil
|
|
}
|
|
|
|
func (srv *LeaderboardService) getHash(interval *models.IntervalKey, by *uint8, user string, pageParams *utils.PageParams) string {
|
|
k := strings.Join(*interval, "__") + "__" + user
|
|
if by != nil && !reflect.ValueOf(by).IsNil() {
|
|
k += "__" + models.GetEntityColumn(*by)
|
|
}
|
|
if pageParams != nil {
|
|
k += "__" + strconv.Itoa(pageParams.Page) + "__" + strconv.Itoa(pageParams.PageSize)
|
|
}
|
|
return k
|
|
}
|