mirror of
https://github.com/navidrome/navidrome.git
synced 2025-04-03 04:27:37 +03:00
83 lines
2 KiB
Go
83 lines
2 KiB
Go
package persistence
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
. "github.com/Masterminds/squirrel"
|
|
"github.com/astaxie/beego/orm"
|
|
"github.com/navidrome/navidrome/model"
|
|
)
|
|
|
|
type scrobbleBufferRepository struct {
|
|
sqlRepository
|
|
}
|
|
|
|
func NewScrobbleBufferRepository(ctx context.Context, o orm.Ormer) model.ScrobbleBufferRepository {
|
|
r := &scrobbleBufferRepository{}
|
|
r.ctx = ctx
|
|
r.ormer = o
|
|
r.tableName = "scrobble_buffer"
|
|
return r
|
|
}
|
|
|
|
func (r *scrobbleBufferRepository) UserIDs(service string) ([]string, error) {
|
|
sql := Select().Columns("user_id").
|
|
From(r.tableName).
|
|
Where(And{
|
|
Eq{"service": service},
|
|
}).
|
|
GroupBy("user_id").
|
|
OrderBy("count(*)")
|
|
var userIds []string
|
|
err := r.queryAll(sql, &userIds)
|
|
return userIds, err
|
|
}
|
|
|
|
func (r *scrobbleBufferRepository) Enqueue(service, userId, mediaFileId string, playTime time.Time) error {
|
|
ins := Insert(r.tableName).SetMap(map[string]interface{}{
|
|
"user_id": userId,
|
|
"service": service,
|
|
"media_file_id": mediaFileId,
|
|
"play_time": playTime,
|
|
"enqueue_time": time.Now(),
|
|
})
|
|
_, err := r.executeSQL(ins)
|
|
return err
|
|
}
|
|
|
|
func (r *scrobbleBufferRepository) Next(service string, userId string) (*model.ScrobbleEntry, error) {
|
|
sql := Select().Columns("s.*, m.*").
|
|
From(r.tableName+" s").
|
|
LeftJoin("media_file m on m.id = s.media_file_id").
|
|
Where(And{
|
|
Eq{"service": service},
|
|
Eq{"user_id": userId},
|
|
}).
|
|
OrderBy("play_time", "s.rowid").Limit(1)
|
|
|
|
res := model.ScrobbleEntries{}
|
|
// TODO Rewrite queryOne to use QueryRows, to workaround the recursive embedded structs issue
|
|
err := r.queryAll(sql, &res)
|
|
if err == model.ErrNotFound || len(res) == 0 {
|
|
return nil, nil
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &res[0], nil
|
|
}
|
|
|
|
func (r *scrobbleBufferRepository) Dequeue(entry *model.ScrobbleEntry) error {
|
|
return r.delete(And{
|
|
Eq{"service": entry.Service},
|
|
Eq{"media_file_id": entry.MediaFile.ID},
|
|
Eq{"play_time": entry.PlayTime},
|
|
})
|
|
}
|
|
|
|
func (r *scrobbleBufferRepository) Length() (int64, error) {
|
|
return r.count(Select())
|
|
}
|
|
|
|
var _ model.ScrobbleBufferRepository = (*scrobbleBufferRepository)(nil)
|