Adding update for Flight

core/src/main/java/cz/muni/fi/pa165/core/facade/flight/FlightFacade.java

@@ -19,4 +19,6 @@ public interface FlightFacade<K> {
     List<FlightDto> findAll();
 
     void deleteById(K id);
+
+    FlightDto update(Long id, NewFlightDtoRequest newFlightDtoRequest);
 }

core/src/main/java/cz/muni/fi/pa165/core/facade/flight/FlightFacadeImpl.java

@@ -71,4 +71,15 @@ public class FlightFacadeImpl implements FlightFacade<Long> {
     public void deleteById(Long id) {
         flightService.deleteById(id);
     }
+
+    @Override
+    public FlightDto update(Long id, NewFlightDtoRequest newFlightDtoRequest) {
+        var newFlightEntity = flightMapper.toEntityFromNewRequest(newFlightDtoRequest);
+        var updatedFlightEntity = flightService.update(id, newFlightEntity);
+
+        var flightDto = flightMapper.toDto(updatedFlightEntity);
+        flightDto.setStewards(mapStewardDtos(updatedFlightEntity));
+
+        return flightDto;
+    }
 }

core/src/main/java/cz/muni/fi/pa165/core/rest/FlightController.java

@@ -86,6 +86,6 @@ public class FlightController implements FlightApiDelegate {
      */
     @Override
     public ResponseEntity<FlightDto> updateFlight(Long id, NewFlightDtoRequest newFlightDtoRequest) {
-        return FlightApiDelegate.super.updateFlight(id, newFlightDtoRequest);
+        return ResponseEntity.ok(flightFacade.update(id, newFlightDtoRequest));
     }
 }

core/src/main/java/cz/muni/fi/pa165/core/service/flight/FlightService.java

@@ -8,4 +8,6 @@ import java.util.Optional;
 public interface FlightService extends BaseService<Flight, Long> {
 
     Optional<Flight> findByIdWithStewards(Long id);
+
+    Flight update(Long id, Flight newFlight);
 }

core/src/main/java/cz/muni/fi/pa165/core/service/flight/FlightServiceImpl.java

@@ -2,6 +2,7 @@ package cz.muni.fi.pa165.core.service.flight;
 
 import cz.muni.fi.pa165.core.data.domain.Flight;
 import cz.muni.fi.pa165.core.data.repository.flight.FlightRepository;
+import cz.muni.fi.pa165.core.exceptions.ResourceNotFoundException;
 import cz.muni.fi.pa165.core.service.common.BaseServiceImpl;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
@@ -23,4 +24,15 @@ public class FlightServiceImpl extends BaseServiceImpl<Flight, Long> implements
     public Optional<Flight> findByIdWithStewards(Long id) {
         return flightRepository.findByIdWithStewards(id);
     }
+
+    @Override
+    public Flight update(Long id, Flight newFlight) {
+        var entityToUpdate = flightRepository.findById(id)
+                .orElseThrow(() -> new ResourceNotFoundException("Not Found."));
+
+        entityToUpdate.setDepartureTime(newFlight.getDepartureTime());
+        entityToUpdate.setArrivalTime(newFlight.getArrivalTime());
+
+        return flightRepository.save(entityToUpdate);
+    }
 }