61 lines
1.2 KiB
Go
61 lines
1.2 KiB
Go
package persistence
|
|
|
|
import (
|
|
"strconv"
|
|
|
|
"github.com/astaxie/beego/orm"
|
|
"github.com/deluan/navidrome/model"
|
|
)
|
|
|
|
type genreRepository struct {
|
|
ormer orm.Ormer
|
|
}
|
|
|
|
func NewGenreRepository(o orm.Ormer) model.GenreRepository {
|
|
return &genreRepository{ormer: o}
|
|
}
|
|
|
|
func (r genreRepository) GetAll() (model.Genres, error) {
|
|
genres := make(map[string]model.Genre)
|
|
|
|
// Collect SongCount
|
|
var res []orm.Params
|
|
_, err := r.ormer.Raw("select genre, count(*) as c from media_file group by genre").Values(&res)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for _, r := range res {
|
|
name := r["genre"].(string)
|
|
count := r["c"].(string)
|
|
g, ok := genres[name]
|
|
if !ok {
|
|
g = model.Genre{Name: name}
|
|
}
|
|
g.SongCount, _ = strconv.Atoi(count)
|
|
genres[name] = g
|
|
}
|
|
|
|
// Collect AlbumCount
|
|
_, err = r.ormer.Raw("select genre, count(*) as c from album group by genre").Values(&res)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for _, r := range res {
|
|
name := r["genre"].(string)
|
|
count := r["c"].(string)
|
|
g, ok := genres[name]
|
|
if !ok {
|
|
g = model.Genre{Name: name}
|
|
}
|
|
g.AlbumCount, _ = strconv.Atoi(count)
|
|
genres[name] = g
|
|
}
|
|
|
|
// Build response
|
|
result := model.Genres{}
|
|
for _, g := range genres {
|
|
result = append(result, g)
|
|
}
|
|
return result, err
|
|
}
|