77 lines
1.7 KiB
Go
77 lines
1.7 KiB
Go
package db_ledis
|
|
|
|
import (
|
|
"errors"
|
|
"sort"
|
|
"time"
|
|
|
|
"github.com/cloudsonic/sonic-server/domain"
|
|
)
|
|
|
|
type mediaFileRepository struct {
|
|
ledisRepository
|
|
}
|
|
|
|
func NewMediaFileRepository() domain.MediaFileRepository {
|
|
r := &mediaFileRepository{}
|
|
r.init("mediafile", &domain.MediaFile{})
|
|
return r
|
|
}
|
|
|
|
func (r *mediaFileRepository) Put(m *domain.MediaFile) error {
|
|
if m.ID == "" {
|
|
return errors.New("mediaFile ID is not set")
|
|
}
|
|
return r.saveOrUpdate(m.ID, m)
|
|
}
|
|
|
|
func (r *mediaFileRepository) Get(id string) (*domain.MediaFile, error) {
|
|
m, err := r.readEntity(id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
mf := m.(*domain.MediaFile)
|
|
if mf.ID != id {
|
|
return nil, nil
|
|
}
|
|
return mf, nil
|
|
}
|
|
|
|
func (r *mediaFileRepository) FindByAlbum(albumId string) (domain.MediaFiles, error) {
|
|
var mfs = make(domain.MediaFiles, 0)
|
|
err := r.loadChildren("album", albumId, &mfs, domain.QueryOptions{SortBy: "TrackNumber"})
|
|
sort.Sort(mfs)
|
|
return mfs, err
|
|
}
|
|
|
|
func (r *mediaFileRepository) GetStarred(options domain.QueryOptions) (domain.MediaFiles, error) {
|
|
var mfs = make(domain.MediaFiles, 0)
|
|
start := time.Time{}.Add(1 * time.Hour)
|
|
err := r.loadRange("Starred", start, time.Now(), &mfs, options)
|
|
return mfs, err
|
|
}
|
|
|
|
func (r *mediaFileRepository) GetAllIds() ([]string, error) {
|
|
idMap, err := r.getAllIds()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
ids := make([]string, len(idMap))
|
|
|
|
i := 0
|
|
for id := range idMap {
|
|
ids[i] = id
|
|
i++
|
|
}
|
|
|
|
return ids, nil
|
|
}
|
|
|
|
func (r *mediaFileRepository) PurgeInactive(active domain.MediaFiles) ([]string, error) {
|
|
return r.purgeInactive(active, func(e interface{}) string {
|
|
return e.(domain.MediaFile).ID
|
|
})
|
|
}
|
|
|
|
var _ domain.MediaFileRepository = (*mediaFileRepository)(nil)
|