[fix]: Fixed bug with friends photos

RapDog64 · Jul 14, 2023 · a4801b2 · a4801b2
1 parent b0bb1df
commit a4801b2
Showing 1 changed file with 8 additions and 11 deletions.
diff --git a/rangiffler-gateway/src/main/java/com/rangiffler/service/PhotoService.java b/rangiffler-gateway/src/main/java/com/rangiffler/service/PhotoService.java
@@ -1,10 +1,12 @@
 package com.rangiffler.service;
 
 
-import com.rangiffler.model.*;
+import com.rangiffler.model.CountryJson;
+import com.rangiffler.model.FriendStatus;
+import com.rangiffler.model.PhotoJson;
+import com.rangiffler.model.PhotoServiceJson;
 import com.rangiffler.service.configuration.PhotoServiceClient;
 import lombok.RequiredArgsConstructor;
-import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
 import java.util.ArrayList;
@@ -42,17 +44,12 @@ public PhotoJson editPhoto(PhotoJson photoJson, UUID id) {
     }
 
     public List<PhotoJson> getAllFriendsPhotos(String username) {
-        List<PhotoJson> friendsPhoto = new ArrayList<>();
-        List<UserJson> friends = userService.receivePeopleAround(username)
+        return userService.receivePeopleAround(username)
                 .stream()
                 .filter(userJson -> userJson.getFriendStatus() == FriendStatus.FRIEND)
-                .toList();
-
-        for (UserJson userJson : friends) {
-            friendsPhoto.addAll(getAllUserPhotos(userJson.getUsername()));
-        }
-
-        return friendsPhoto;
+                .map(userJson -> getAllUserPhotos(userJson.getUsername()))
+                .findFirst()
+                .orElseGet(ArrayList::new);
     }
 
     public void deletePhoto(UUID photoId) {