Another big refactor: Back to a single folder for persistence implementation

2020-01-14 18:23:29 -05:00
parent 08e096c569
commit a99c3a8af3
27 changed files with 177 additions and 171 deletions
@@ -0,0 +1,117 @@
+package persistence
+
+import (
+	"time"
+
+	"github.com/astaxie/beego/orm"
+	"github.com/cloudsonic/sonic-server/domain"
+)
+
+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       `orm:"index"`
+	PlayDate    time.Time `orm:"null"`
+	Rating      int       `orm:"index"`
+	Starred     bool      `orm:"index"`
+	StarredAt   time.Time `orm:"null"`
+	CreatedAt   time.Time `orm:"null"`
+	UpdatedAt   time.Time `orm:"null"`
+}
+
+type mediaFileRepository struct {
+	searchableRepository
+}
+
+func NewMediaFileRepository() domain.MediaFileRepository {
+	r := &mediaFileRepository{}
+	r.tableName = "media_file"
+	return r
+}
+
+func (r *mediaFileRepository) Put(m *domain.MediaFile) error {
+	tm := MediaFile(*m)
+	return withTx(func(o orm.Ormer) error {
+		return r.put(o, m.ID, m.Title, &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 {
+	result := make(domain.MediaFiles, len(all))
+	for i, m := range all {
+		result[i] = domain.MediaFile(m)
+	}
+	return result
+}
+
+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), nil
+}
+
+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), nil
+}
+
+func (r *mediaFileRepository) PurgeInactive(activeList domain.MediaFiles) error {
+	return withTx(func(o orm.Ormer) error {
+		_, err := r.purgeInactive(o, activeList, func(item interface{}) string {
+			return item.(domain.MediaFile).ID
+		})
+		return err
+	})
+}
+
+func (r *mediaFileRepository) Search(q string, offset int, size int) (domain.MediaFiles, error) {
+	if len(q) <= 2 {
+		return nil, nil
+	}
+
+	var results []MediaFile
+	err := r.doSearch(r.tableName, q, offset, size, &results, "rating desc", "starred desc", "play_count desc", "title")
+	if err != nil {
+		return nil, err
+	}
+	return r.toMediaFiles(results), nil
+}
+
+var _ domain.MediaFileRepository = (*mediaFileRepository)(nil)
+var _ = domain.MediaFile(MediaFile{})