From 9b8f39107063a0d93871af3cdbbebe2364faa4c9 Mon Sep 17 00:00:00 2001
From: ahasparian <a.hasparian@gmail.com>
Date: Wed, 24 Jan 2024 14:39:08 +0200
Subject: [PATCH] refactor: Change method getUsserAdvertisement

---
 .../marketplace/ad/AdvertisementRepository.java     |  2 ++
 .../marketplace/ad/AdvertisementServiceImpl.java    | 13 +++++--------
 2 files changed, 7 insertions(+), 8 deletions(-)

diff --git a/src/main/java/com/sellbycar/marketplace/ad/AdvertisementRepository.java b/src/main/java/com/sellbycar/marketplace/ad/AdvertisementRepository.java
index aea36aa..807359a 100644
--- a/src/main/java/com/sellbycar/marketplace/ad/AdvertisementRepository.java
+++ b/src/main/java/com/sellbycar/marketplace/ad/AdvertisementRepository.java
@@ -9,4 +9,6 @@
 public interface AdvertisementRepository extends JpaRepository<AdvertisementDAO, Long>, JpaSpecificationExecutor<AdvertisementDAO> {
 
     @NonNull List<AdvertisementDAO> findAll();
+
+    @NonNull List<AdvertisementDAO> findByUserId(Long userId);
 }
diff --git a/src/main/java/com/sellbycar/marketplace/ad/AdvertisementServiceImpl.java b/src/main/java/com/sellbycar/marketplace/ad/AdvertisementServiceImpl.java
index 0a2c16a..a4e7a10 100644
--- a/src/main/java/com/sellbycar/marketplace/ad/AdvertisementServiceImpl.java
+++ b/src/main/java/com/sellbycar/marketplace/ad/AdvertisementServiceImpl.java
@@ -148,16 +148,13 @@ public void removeAdvertisement(Long id) {
     @Transactional
     public List<AdvertisementDTO> getUserAdvertisement() {
         UserDAO user = userService.getUserFromSecurityContextHolder();
-        List<AdvertisementDAO> advertisements = advertisementRepository.findAll();
-        if (advertisements.stream().anyMatch(adv -> adv.getUser().getId().equals(user.getId()))) {
-            return advertisements.stream()
-                    .map(advertisementMapper::toDTO)
-                    .collect(Collectors.toList());
-        } else {
-            throw new RequestException("Insufficient privileges", HttpStatus.FORBIDDEN);
-        }
+        List<AdvertisementDAO> advertisements = advertisementRepository.findByUserId(user.getId());
+        return advertisements.stream()
+                .map(advertisementMapper::toDTO)
+                .collect(Collectors.toList());
     }
 
+
     @Transactional
     public void enableAdvertisement(Long id) {
         UserDAO existingUser = userService.getUserFromSecurityContextHolder();