quintodrome/repositories/index_repository.go

package repositories

import (
	"github.com/deluan/gosonic/models"
	"errors"
	"sort"
	"github.com/deluan/gosonic/utils"
)

type ArtistIndex interface {
	Put(m *models.ArtistIndex) error
	Get(id string) (*models.ArtistIndex, error)
	GetAll() ([]models.ArtistIndex, error)
}

type artistIndex struct {
	BaseRepository
}

func NewArtistIndexRepository() ArtistIndex {
	r := &artistIndex{}
	r.init("index", &models.ArtistIndex{})
	return r
}

func (r *artistIndex) Put(m *models.ArtistIndex) error {
	if m.Id == "" {
		return errors.New("Id is not set")
	}
	return r.saveOrUpdate(m.Id, m)
}

func (r *artistIndex) Get(id string) (*models.ArtistIndex, error) {
	var rec interface{}
	rec, err := r.readEntity(id)
	return rec.(*models.ArtistIndex), err
}

func (r *artistIndex) GetAll() ([]models.ArtistIndex, error) {
	var indices = make([]models.ArtistIndex, 0)
	err := r.loadAll(&indices, "")
	if err == nil {
		for _, idx := range indices {
			sort.Sort(byArtistName(idx.Artists))
		}
	}
	return indices, err
}

type byArtistName []models.ArtistInfo

func (a byArtistName) Len() int           { return len(a) }
func (a byArtistName) Swap(i, j int)      { a[i], a[j] = a[j], a[i] }
func (a byArtistName) Less(i, j int) bool { return utils.NoArticle(a[i].Artist) < utils.NoArticle(a[j].Artist) }