diff --git a/api/funkwhale_api/favorites/views.py b/api/funkwhale_api/favorites/views.py index 5f4e1cd42e34393ecf8545c62278a2097cc56deb..e02d1a3e40f41a33cb72d843f0453f1bcc3c9e68 100644 --- a/api/funkwhale_api/favorites/views.py +++ b/api/funkwhale_api/favorites/views.py @@ -71,3 +71,19 @@ class TrackFavoriteViewSet( return Response({}, status=400) favorite.delete() return Response([], status=status.HTTP_204_NO_CONTENT) + + @list_route(methods=["get"]) + def all(self, request, *args, **kwargs): + """ + Return all the favorites of the current user, with only limited data + to have a performant endpoint and avoid lots of queries just to display + favorites status in the UI + """ + if not request.user.is_authenticated: + return Response({"results": [], "count": 0}, status=200) + + favorites = list( + request.user.track_favorites.values("id", "track").order_by("id") + ) + payload = {"results": favorites, "count": len(favorites)} + return Response(payload, status=200) diff --git a/api/tests/favorites/test_favorites.py b/api/tests/favorites/test_favorites.py index 0b99c93409a7ca1eb542fe2607d29a3b54c1293c..7e8d1d3fdd14f5e082329873bfe25a6c7c550bfc 100644 --- a/api/tests/favorites/test_favorites.py +++ b/api/tests/favorites/test_favorites.py @@ -39,6 +39,18 @@ def test_user_can_get_his_favorites(api_request, factories, logged_in_client, cl assert response.data["results"] == expected +def test_user_can_retrieve_all_favorites_at_once( + api_request, factories, logged_in_client, client +): + favorite = factories["favorites.TrackFavorite"](user=logged_in_client.user) + factories["favorites.TrackFavorite"]() + url = reverse("api:v1:favorites:tracks-all") + response = logged_in_client.get(url, {"user": logged_in_client.user.pk}) + expected = [{"track": favorite.track.id, "id": favorite.id}] + assert response.status_code == 200 + assert response.data["results"] == expected + + def test_user_can_add_favorite_via_api(factories, logged_in_client, activity_muted): track = factories["music.Track"]() url = reverse("api:v1:favorites:tracks-list") diff --git a/changes/changelog.d/favorites-performance.enhancement b/changes/changelog.d/favorites-performance.enhancement new file mode 100644 index 0000000000000000000000000000000000000000..c35828fc2b809a03aa89c84650b8abec675a8fc5 --- /dev/null +++ b/changes/changelog.d/favorites-performance.enhancement @@ -0,0 +1 @@ +Performance improvement when fetching favorites, down to a single, small http request diff --git a/front/src/store/favorites.js b/front/src/store/favorites.js index 131db24bd90cbe334d40bb2c7eaf359afe5e073c..1d1302eb6942e6697111b35774207d9cfe49c142 100644 --- a/front/src/store/favorites.js +++ b/front/src/store/favorites.js @@ -60,20 +60,12 @@ export default { page_size: 50, ordering: '-creation_date' } - let promise - if (url) { - promise = axios.get(url) - } else { - promise = axios.get('favorites/tracks/', {params: params}) - } + let promise = axios.get('favorites/tracks/all/', {params: params}) return promise.then((response) => { logger.default.info('Fetched a batch of ' + response.data.results.length + ' favorites') response.data.results.forEach(result => { - commit('track', {id: result.track.id, value: true}) + commit('track', {id: result.track, value: true}) }) - if (response.data.next) { - dispatch('fetch', response.data.next) - } }) } }