Changeset ceaed42 for trip-planner/src/main/java/finki/diplomska/tripplanner/service/impl/LocationServiceImpl.java

Timestamp:

11/11/21 12:59:26 (3 years ago)

Author:

Ema <ema_spirova@…>

Branches:

master

Children:

59329aa

Parents:

6c1585f

Message:

add location to planner

File:

• trip-planner/src/main/java/finki/diplomska/tripplanner/service/impl/LocationServiceImpl.java (modified) (3 diffs)

trip-planner/src/main/java/finki/diplomska/tripplanner/service/impl/LocationServiceImpl.java

@@ -2 +2 @@
 
 import finki.diplomska.tripplanner.models.Location;
+import finki.diplomska.tripplanner.models.Planner;
+import finki.diplomska.tripplanner.models.dto.PlannerLocationDto;
 import finki.diplomska.tripplanner.models.exceptions.LocationNotFoundException;
 import finki.diplomska.tripplanner.repository.jpa.JpaLocationRepository;
+import finki.diplomska.tripplanner.repository.jpa.JpaPlannerRepository;
 import finki.diplomska.tripplanner.service.LocationService;
 import org.springframework.stereotype.Service;
@@ -15 +18 @@
 
     private final JpaLocationRepository locationRepository;
+    private final JpaPlannerRepository plannerRepository;
 
-    public LocationServiceImpl(JpaLocationRepository locationRepository) {
+    public LocationServiceImpl(JpaLocationRepository locationRepository, JpaPlannerRepository plannerRepository) {
         this.locationRepository = locationRepository;
+        this.plannerRepository = plannerRepository;
     }
 
@@ -150 +155 @@
     }
 
+    @Override
+    public Location addLocationToPlanner(PlannerLocationDto plannerLocationDto) {
+        Location location = this.locationRepository.findById(plannerLocationDto.getLocationId())
+                .orElseThrow(() -> new LocationNotFoundException(plannerLocationDto.getLocationId()));
+        Planner planner = this.plannerRepository.getById(plannerLocationDto.getPlannerId());
+        planner.getLocationList().add(location);
+        return this.locationRepository.save(location);
+    }
+
+    @Override
+    public List<Location> getAllLocationsForPlanner(Long plannerId) {
+        return this.locationRepository.getAllLocationsForPlanner(plannerId);
+    }
+
+
 }