third-party-mirrors/navidrome
1
0
Fork 0
mirror of https://github.com/navidrome/navidrome.git synced 2025-04-24 07:30:57 +03:00
Code Issues Packages Projects Releases Wiki Activity

Using sorted sets

Browse Source
This commit is contained in:
Deluan 2016-03-05 19:40:36 -05:00
parent 3db18d8bd6
commit d2701ea67c
3 changed files with 13 additions and 11 deletions

2
api/get_music_directory.go

@ -97,7 +97,7 @@ func (c *GetMusicDirectoryController) buildAlbumDir(al *domain.Album, tracks []d
func (c *GetMusicDirectoryController) isArtist(id string) bool {
found, err := c.artistRepo.Exists(id)
if err != nil {
beego.Error("Error searching for Artist:", err)
beego.Error("Error searching for Artist", id, " - ", err)
c.SendError(responses.ERROR_GENERIC, "Internal Error")
}
return found

2
domain/base.go

@ -2,7 +2,7 @@ package domain
type BaseRepository interface {
NewId(fields ...string) string
CountAll() (int, error)
CountAll() (int64, error)
Exists(id string) (bool, error)
}

20
persistence/ledis_repository.go

@ -4,10 +4,11 @@ import (
"crypto/md5"
"encoding/json"
"fmt"
"github.com/deluan/gosonic/domain"
"github.com/deluan/gosonic/utils"
"github.com/siddontang/ledisdb/ledis"
"reflect"
"strings"
"github.com/deluan/gosonic/domain"
)
type ledisRepository struct {
@ -35,14 +36,14 @@ func (r *ledisRepository) NewId(fields ...string) string {
return fmt.Sprintf("%x", md5.Sum([]byte(s)))
}
func (r *ledisRepository) CountAll() (int, error) {
ids, err := db().SMembers([]byte(r.table + "s:all"))
return len(ids), err
func (r *ledisRepository) CountAll() (int64, error) {
size, err := db().ZCard([]byte(r.table + "s:all"))
return size, err
}
func (r *ledisRepository) Exists(id string) (bool, error) {
res, err := db().SIsMember([]byte(r.table+"s:all"), []byte(id))
return res != 0, err
res, _ := db().ZScore([]byte(r.table+"s:all"), []byte(id))
return res != ledis.InvalidScore, nil
}
func (r *ledisRepository) saveOrUpdate(id string, entity interface{}) error {
@ -62,13 +63,14 @@ func (r *ledisRepository) saveOrUpdate(id string, entity interface{}) error {
}
if _, err = db().SAdd([]byte(allKey), []byte(id)); err != nil {
sid := ledis.ScorePair{0, []byte(id)}
if _, err = db().ZAdd([]byte(allKey), sid); err != nil {
return err
}
if parentTable, parentId := r.getParent(entity); parentTable != "" {
parentCollectionKey := fmt.Sprintf("%s:%s:%ss", parentTable, parentId, r.table)
_, err = db().SAdd([]byte(parentCollectionKey), []byte(id))
_, err = db().ZAdd([]byte(parentCollectionKey), sid)
}
return nil
}
@ -150,7 +152,7 @@ func (r *ledisRepository) loadFromSet(setName string, entities interface{}, qo .
if o.SortBy != "" {
sortKey = []byte(fmt.Sprintf("%s:*:%s", r.table, o.SortBy))
}
response, err := db().XSSort([]byte(setName), o.Offset, o.Size, o.Alpha, o.Desc, sortKey, r.getFieldKeys("*"))
response, err := db().XZSort([]byte(setName), o.Offset, o.Size, o.Alpha, o.Desc, sortKey, r.getFieldKeys("*"))
if err != nil {
return err
}