initial cheapest flight search version

Author: carousel

pom.xml

@@ -96,6 +96,13 @@
             <artifactId>commons-lang3</artifactId>
             <version>3.10</version>
         </dependency>
+        <!-- https://mvnrepository.com/artifact/com.google.guava/guava -->
+        <dependency>
+            <groupId>com.google.guava</groupId>
+            <artifactId>guava</artifactId>
+            <version>29.0-jre</version>
+        </dependency>
+
     </dependencies>
     <build>
         <plugins>

src/main/java/com/miro/flightadvisor/controllers/CityController.java

@@ -58,11 +58,6 @@ public Optional<List<Comment>> comments() {
         return cityService.allCommentsForCity();
     }
 
-    @GetMapping("/routes")
-    public Optional<List<Route>> getRoutes() {
-        return cityService.allRoutes();
-    }
-
     @PostMapping("/cities/{cityId}/comments")
     @ResponseStatus(HttpStatus.CREATED)
     public ResponseEntity<String> addCommentForCity(@PathVariable("cityId") String cityId, @RequestBody @Valid CommentBean commentBean) {

src/main/java/com/miro/flightadvisor/controllers/FlightRouteController.java

@@ -0,0 +1,36 @@
+package com.miro.flightadvisor.controllers;
+
+import com.miro.flightadvisor.beans.CommentBean;
+import com.miro.flightadvisor.entities.Comment;
+import com.miro.flightadvisor.entities.Route;
+import com.miro.flightadvisor.services.CityService;
+import com.miro.flightadvisor.services.RouteService;
+import com.sun.istack.NotNull;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.http.HttpStatus;
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.*;
+
+import javax.validation.Valid;
+import java.util.List;
+import java.util.Optional;
+
+@RestController
+public class FlightRouteController {
+
+    private RouteService routeService;
+
+
+    @Autowired
+    public FlightRouteController(RouteService routeService) {
+        this.routeService = routeService;
+    }
+
+
+    @GetMapping("/routes/find-cheapest/{src}/{dest}")
+    public Optional<?> getCity(@NotNull @PathVariable(value = "src") String src, @PathVariable(value = "dest") String dest) {
+        return routeService.getCheapestFlight(src, dest);
+    }
+
+
+}

src/main/java/com/miro/flightadvisor/entities/CheapestFlight.java

@@ -0,0 +1,24 @@
+package com.miro.flightadvisor.entities;
+
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+import java.math.BigDecimal;
+import java.util.ArrayList;
+import java.util.List;
+
+@Data
+@NoArgsConstructor
+public class CheapestFlight {
+    private BigDecimal cheapest;
+    private BigDecimal total;
+    private List<Route> routes = new ArrayList<>();
+
+    public CheapestFlight(BigDecimal cheapest, BigDecimal total, List<Route> routes) {
+        this.cheapest = cheapest;
+        this.total = total;
+        this.routes = routes;
+    }
+
+
+}

src/main/java/com/miro/flightadvisor/entities/Route.java

@@ -42,7 +42,7 @@ public class Route {
     @Column(name = "stops", length = 1)
     private Integer stops;
 
-    @Column(name = "equipment", length = 3)
+    @Column(name = "equipment", length = 24)
     private String equipment;
 
     @Column(name = "flight_cost")

src/main/java/com/miro/flightadvisor/repositories/AirportRepository.java

@@ -7,8 +7,14 @@
 import java.util.List;
 import java.util.Optional;
 
-public interface AirportRepository extends JpaRepository<Airport, Integer> {
+public interface AirportRepository extends JpaRepository<Airport, Long> {
+
+    @Query("select a.providedAirportId from Airport a where a.city=:city")
+    List<Integer> getOnlyAirportId(String city);
+
     Optional<Airport> findByName(String name);
 
+    Optional<List<Airport>> findByCity(String city);
+
     Optional<Airport> findByProvidedAirportId(Integer providedAirportId);
 }

src/main/java/com/miro/flightadvisor/services/CityService.java

@@ -37,16 +37,17 @@ public CityService(CityRepository cityRepository, CommentRepository commentRepos
         this.routeRepository = routeRepository;
     }
 
-    public Optional<List<Route>> allRoutes() {
-        return Optional.of(routeRepository.findAll());
-    }
-
     public void addCity(CityBean cityBean) {
-        City city = new City();
-        city.setCountry(cityBean.getCountry());
-        city.setDescription(cityBean.getDescription());
-        city.setName(cityBean.getName());
-        this.cityRepository.save(city);
+        City newCity = new City();
+        newCity.setCountry(cityBean.getCountry());
+        newCity.setDescription(cityBean.getDescription());
+        newCity.setName(cityBean.getName());
+        Optional<City> city = cityRepository.findByName(newCity.getName());
+        if (city.isPresent()) {
+            return;
+        } else {
+            this.cityRepository.save(newCity);
+        }
     }
 
     public Optional<List<City>> allCities() {

src/main/java/com/miro/flightadvisor/services/RouteService.java

@@ -1,28 +1,40 @@
 package com.miro.flightadvisor.services;
 
+import ch.qos.logback.core.db.BindDataSourceToJNDIAction;
+import com.google.common.collect.ImmutableList;
+import com.google.common.collect.Lists;
 import com.miro.flightadvisor.beans.AirportBean;
 import com.miro.flightadvisor.beans.RouteBean;
 import com.miro.flightadvisor.entities.Airport;
+import com.miro.flightadvisor.entities.CheapestFlight;
 import com.miro.flightadvisor.entities.Route;
+import com.miro.flightadvisor.exception.FlightAdvisorRuntimeException;
 import com.miro.flightadvisor.repositories.AirportRepository;
 import com.miro.flightadvisor.repositories.RouteRepository;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+import sun.corba.Bridge;
 
-import java.util.List;
-import java.util.Optional;
+import javax.persistence.criteria.CriteriaBuilder;
+import java.math.BigDecimal;
+import java.util.*;
+import java.util.stream.Collector;
+import java.util.stream.Collectors;
+import java.util.stream.IntStream;
+import java.util.stream.Stream;
 
 @Service
 public class RouteService {
 
     private final RouteRepository routeRepository;
+    private final AirportRepository airportRepository;
 
     @Autowired
-    public RouteService(RouteRepository routeRepository) {
+    public RouteService(RouteRepository routeRepository, AirportRepository airportRepository) {
         this.routeRepository = routeRepository;
+        this.airportRepository = airportRepository;
     }
 
-
     public void saveRoute(RouteBean routeBean) {
         Integer sourceAirportId = routeBean.getSourceAirportId();
         Integer destinationAirportId = routeBean.getDestinationAirportId();
@@ -34,4 +46,49 @@ public void saveRoute(RouteBean routeBean) {
             routeRepository.save(newRoute);
         }
     }
+
+    public Optional<List<Route>> allRoutes() {
+        return Optional.of(routeRepository.findAll());
+    }
+
+    public Optional<CheapestFlight> getCheapestFlight(String src, String dest) {
+
+        List<Integer> sources = airportRepository.getOnlyAirportId(src);
+        List<Integer> destinations = airportRepository.getOnlyAirportId(dest);
+        return calculateAndReturnData(sources, destinations);
+    }
+
+    public Optional<CheapestFlight> calculateAndReturnData(List<Integer> sources, List<Integer> destinations) {
+        List<List<Integer>> airportIds = Lists.cartesianProduct(ImmutableList.of(
+                sources,
+                destinations));
+        List<Route> routes = new ArrayList<>();
+        airportIds.stream().forEachOrdered(a -> {
+            Integer source = a.get(0);
+            Integer destination = a.get(1);
+            Optional<Route> route = routeRepository.findBySourceAirportIdAndDestinationAirportId(source, destination);
+            if (route.isPresent()) {
+                routes.add(route.get());
+            }
+        });
+
+        if (routes.size() > 0) {
+            List<BigDecimal> totalCosts = new ArrayList<>();
+            routes.stream().forEach(r -> {
+                totalCosts.add(r.getFlightCost());
+            });
+
+            BigDecimal total = totalCosts.stream().reduce((i, j) -> i.add(j)).get();
+            BigDecimal cheapest = totalCosts.stream().reduce((i, j) -> i.min(j)).get();
+
+            CheapestFlight cheapestFlight = new CheapestFlight(cheapest, total, routes);
+            return Optional.of(cheapestFlight);
+        }
+            BigDecimal total = new BigDecimal("0.0");
+            BigDecimal cheapest = new BigDecimal("0.0");
+            CheapestFlight cheapestFlight = new CheapestFlight(cheapest, total, routes);
+            return Optional.of(cheapestFlight);
+
+    }
+
 }