mirror of https://github.com/muety/wakapi.git
109 lines
2.8 KiB
Go
109 lines
2.8 KiB
Go
package repositories
|
|
|
|
import (
|
|
"github.com/muety/wakapi/models"
|
|
"github.com/muety/wakapi/utils"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/clause"
|
|
)
|
|
|
|
type LeaderboardRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewLeaderboardRepository(db *gorm.DB) *LeaderboardRepository {
|
|
return &LeaderboardRepository{db: db}
|
|
}
|
|
|
|
func (r *LeaderboardRepository) InsertBatch(items []*models.LeaderboardItem) error {
|
|
if err := r.db.
|
|
Clauses(clause.OnConflict{DoNothing: true}).
|
|
Create(&items).Error; err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (r *LeaderboardRepository) CountAllByUser(userId string) (int64, error) {
|
|
var count int64
|
|
err := r.db.
|
|
Table("leaderboard_items").
|
|
Where("user_id = ?", userId).
|
|
Count(&count).Error
|
|
return count, err
|
|
}
|
|
|
|
func (r *LeaderboardRepository) CountUsers() (int64, error) {
|
|
var count int64
|
|
err := r.db.
|
|
Table("leaderboard_items").
|
|
Distinct("user_id").
|
|
Count(&count).Error
|
|
return count, err
|
|
}
|
|
|
|
func (r *LeaderboardRepository) GetAllAggregatedByInterval(key *models.IntervalKey, by *uint8, limit, skip int) ([]*models.LeaderboardItemRanked, error) {
|
|
// TODO: distinct by (user, key) to filter out potential duplicates ?
|
|
|
|
var items []*models.LeaderboardItemRanked
|
|
subq := r.db.
|
|
Table("leaderboard_items").
|
|
Select("*, rank() over (partition by \"key\" order by total desc) as \"rank\"").
|
|
Where("\"interval\" in ?", *key)
|
|
subq = utils.WhereNullable(subq, "\"by\"", by)
|
|
|
|
q := r.db.Table("(?) as ranked", subq)
|
|
q = r.withPaging(q, limit, skip)
|
|
|
|
if err := q.Find(&items).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return items, nil
|
|
}
|
|
|
|
func (r *LeaderboardRepository) GetAggregatedByUserAndInterval(userId string, key *models.IntervalKey, by *uint8, limit, skip int) ([]*models.LeaderboardItemRanked, error) {
|
|
var items []*models.LeaderboardItemRanked
|
|
subq := r.db.
|
|
Table("leaderboard_items").
|
|
Select("*, rank() over (partition by \"key\" order by total desc) as \"rank\"").
|
|
Where("\"interval\" in ?", *key)
|
|
subq = utils.WhereNullable(subq, "\"by\"", by)
|
|
|
|
q := r.db.Table("(?) as ranked", subq).Where("user_id = ?", userId)
|
|
q = r.withPaging(q, limit, skip)
|
|
|
|
if err := q.Find(&items).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return items, nil
|
|
}
|
|
|
|
func (r *LeaderboardRepository) DeleteByUser(userId string) error {
|
|
if err := r.db.
|
|
Where("user_id = ?", userId).
|
|
Delete(models.LeaderboardItem{}).Error; err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (r *LeaderboardRepository) DeleteByUserAndInterval(userId string, key *models.IntervalKey) error {
|
|
if err := r.db.
|
|
Where("user_id = ?", userId).
|
|
Where("\"interval\" in ?", *key).
|
|
Delete(models.LeaderboardItem{}).Error; err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (r *LeaderboardRepository) withPaging(q *gorm.DB, limit, skip int) *gorm.DB {
|
|
if limit > 0 {
|
|
q = q.Where("\"rank\" <= ?", skip+limit)
|
|
}
|
|
if skip > 0 {
|
|
q = q.Where("\"rank\" > ?", skip)
|
|
}
|
|
return q
|
|
}
|