[fix] fix FavoriteImage update logic (#239)

Team-Hankki · Dec 27, 2024 · 14c225d · 14c225d
1 parent 3c8d981
commit 14c225d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/main/java/org/hankki/hankkiserver/api/favorite/service/FavoriteCommandService.java b/src/main/java/org/hankki/hankkiserver/api/favorite/service/FavoriteCommandService.java
@@ -100,7 +100,9 @@ private void checkTitleExists(final String title, final User user){
   }
 
   private void copySharedFavoriteStore(final long sharedId, final Favorite myFavorite) {
-    findStoresById(sharedId).forEach(it -> favoriteStoreUpdater.save(FavoriteStore.create(it, myFavorite)));
+    List<Store> storesInFavoriteStore = findStoresById(sharedId);
+    storesInFavoriteStore.forEach(it -> favoriteStoreUpdater.save(FavoriteStore.create(it, myFavorite)));
+    myFavorite.updateImageByFavoriteStoreCount(storesInFavoriteStore.size());
   }
 
   private List<Store> findStoresById(final long id) {