SQL/Orm MediaFileRepository complete

2020-01-12 18:36:19 -05:00
parent d70af2c39d
commit 56273dd4d9
6 changed files with 182 additions and 6 deletions
@@ -0,0 +1,115 @@
+package db_sql
+
+import (
+	"time"
+
+	"github.com/astaxie/beego/orm"
+	"github.com/cloudsonic/sonic-server/domain"
+	"github.com/cloudsonic/sonic-server/persistence"
+)
+
+type MediaFile struct {
+	ID          string    `orm:"pk;column(id)"`
+	Path        string    ``
+	Title       string    `orm:"index"`
+	Album       string    ``
+	Artist      string    ``
+	ArtistID    string    `orm:"column(artist_id)"`
+	AlbumArtist string    ``
+	AlbumID     string    `orm:"column(album_id);index"`
+	HasCoverArt bool      ``
+	TrackNumber int       ``
+	DiscNumber  int       ``
+	Year        int       ``
+	Size        string    ``
+	Suffix      string    ``
+	Duration    int       ``
+	BitRate     int       ``
+	Genre       string    ``
+	Compilation bool      ``
+	PlayCount   int       ``
+	PlayDate    time.Time `orm:"null"`
+	Rating      int       ``
+	Starred     bool      `orm:"index"`
+	StarredAt   time.Time `orm:"null"`
+	CreatedAt   time.Time `orm:"null"`
+	UpdatedAt   time.Time `orm:"null"`
+}
+
+type mediaFileRepository struct {
+	sqlRepository
+}
+
+func NewMediaFileRepository() domain.MediaFileRepository {
+	r := &mediaFileRepository{}
+	r.entityName = "media_file"
+	return r
+}
+
+func (r *mediaFileRepository) Put(m *domain.MediaFile) error {
+	tm := MediaFile(*m)
+	return r.put(m.ID, &tm)
+}
+
+func (r *mediaFileRepository) Get(id string) (*domain.MediaFile, error) {
+	tm := MediaFile{ID: id}
+	err := Db().Read(&tm)
+	if err == orm.ErrNoRows {
+		return nil, domain.ErrNotFound
+	}
+	if err != nil {
+		return nil, err
+	}
+	a := domain.MediaFile(tm)
+	return &a, nil
+}
+
+func (r *mediaFileRepository) toMediaFiles(all []MediaFile) (domain.MediaFiles, error) {
+	result := make(domain.MediaFiles, len(all))
+	for i, m := range all {
+		result[i] = domain.MediaFile(m)
+	}
+	return result, nil
+}
+
+func (r *mediaFileRepository) FindByAlbum(albumId string) (domain.MediaFiles, error) {
+	var mfs []MediaFile
+	_, err := r.newQuery(Db()).Filter("album_id", albumId).OrderBy("disc_number", "track_number").All(&mfs)
+	if err != nil {
+		return nil, err
+	}
+	return r.toMediaFiles(mfs)
+}
+
+func (r *mediaFileRepository) GetStarred(options ...domain.QueryOptions) (domain.MediaFiles, error) {
+	var starred []MediaFile
+	_, err := r.newQuery(Db(), options...).Filter("starred", true).All(&starred)
+	if err != nil {
+		return nil, err
+	}
+	return r.toMediaFiles(starred)
+}
+
+func (r *mediaFileRepository) GetAllIds() ([]string, error) {
+	qs := r.newQuery(Db())
+	var values []orm.Params
+	num, err := qs.Values(&values, "id")
+	if num == 0 {
+		return nil, err
+	}
+
+	result := persistence.CollectValue(values, func(item interface{}) string {
+		return item.(orm.Params)["ID"].(string)
+	})
+
+	return result, nil
+}
+
+func (r *mediaFileRepository) PurgeInactive(activeList domain.MediaFiles) ([]string, error) {
+	return r.purgeInactive(activeList, func(item interface{}) string {
+		return item.(domain.MediaFile).ID
+	})
+}
+
+var _ domain.MediaFileRepository = (*mediaFileRepository)(nil)
+var _ = domain.MediaFile(MediaFile{})