98 lines
2.1 KiB
Go
98 lines
2.1 KiB
Go
package db_ledis
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"github.com/cloudsonic/sonic-server/domain"
|
|
)
|
|
|
|
var (
|
|
nowPlayingKeyPrefix = []byte("nowplaying")
|
|
)
|
|
|
|
type nowPlayingRepository struct {
|
|
ledisRepository
|
|
}
|
|
|
|
func NewNowPlayingRepository() domain.NowPlayingRepository {
|
|
r := &nowPlayingRepository{}
|
|
r.init("nowplaying", &domain.NowPlayingInfo{})
|
|
return r
|
|
}
|
|
|
|
func nowPlayingKeyName(playerId int) string {
|
|
return fmt.Sprintf("%s:%d", nowPlayingKeyPrefix, playerId)
|
|
}
|
|
|
|
func (r *nowPlayingRepository) Enqueue(info *domain.NowPlayingInfo) error {
|
|
h, err := json.Marshal(info)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
keyName := []byte(nowPlayingKeyName(info.PlayerId))
|
|
|
|
_, err = Db().LPush(keyName, []byte(h))
|
|
Db().LExpire(keyName, int64(domain.NowPlayingExpire.Seconds()))
|
|
return err
|
|
}
|
|
|
|
func (r *nowPlayingRepository) Dequeue(playerId int) (*domain.NowPlayingInfo, error) {
|
|
keyName := []byte(nowPlayingKeyName(playerId))
|
|
|
|
val, err := Db().RPop(keyName)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if val == nil {
|
|
return nil, nil
|
|
}
|
|
return r.parseInfo(val)
|
|
}
|
|
|
|
func (r *nowPlayingRepository) Head(playerId int) (*domain.NowPlayingInfo, error) {
|
|
keyName := []byte(nowPlayingKeyName(playerId))
|
|
|
|
val, err := Db().LIndex(keyName, 0)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return r.parseInfo(val)
|
|
}
|
|
|
|
func (r *nowPlayingRepository) Tail(playerId int) (*domain.NowPlayingInfo, error) {
|
|
keyName := []byte(nowPlayingKeyName(playerId))
|
|
|
|
val, err := Db().LIndex(keyName, -1)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return r.parseInfo(val)
|
|
}
|
|
|
|
func (r *nowPlayingRepository) Count(playerId int) (int64, error) {
|
|
keyName := []byte(nowPlayingKeyName(playerId))
|
|
return Db().LLen(keyName)
|
|
}
|
|
|
|
// TODO Will not work for multiple players
|
|
func (r *nowPlayingRepository) GetAll() ([]*domain.NowPlayingInfo, error) {
|
|
np, err := r.Head(1)
|
|
if np == nil || err != nil {
|
|
return nil, err
|
|
}
|
|
return []*domain.NowPlayingInfo{np}, err
|
|
}
|
|
|
|
func (r *nowPlayingRepository) parseInfo(val []byte) (*domain.NowPlayingInfo, error) {
|
|
info := &domain.NowPlayingInfo{}
|
|
err := json.Unmarshal(val, info)
|
|
if err != nil {
|
|
return nil, nil
|
|
}
|
|
return info, nil
|
|
}
|
|
|
|
var _ domain.NowPlayingRepository = (*nowPlayingRepository)(nil)
|