wakapi/repositories/leaderboard.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
}