Fix album lists, to use annotations

2020-01-22 01:00:00 -05:00
parent d7116eebd4
commit c8a74881d8
6 changed files with 75 additions and 19 deletions
@@ -75,6 +75,19 @@ func (r *albumRepository) GetAll(options ...model.QueryOptions) (model.Albums, e
 	return r.toAlbums(all), nil
 }

+func (r *albumRepository) GetMap(ids []string) (map[string]model.Album, error) {
+	var all []album
+	_, err := r.newQuery().Filter("id__in", ids).All(&all)
+	if err != nil {
+		return nil, err
+	}
+	res := make(map[string]model.Album)
+	for _, a := range all {
+		res[a.ID] = model.Album(a)
+	}
+	return res, nil
+}
+
 // TODO Keep order when paginating
 func (r *albumRepository) GetRandom(options ...model.QueryOptions) (model.Albums, error) {
 	sq := r.newRawQuery(options...)