feat: improve graphql endpoint to accept limit param list queries

jgiovaresco · Jan 29, 2024 · 1d7e81d · 1d7e81d
1 parent b8affd5
commit 1d7e81d
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 6 deletions.
diff --git a/app-quarkus/src/main/kotlin/io/apim/samples/core/starwars/StarWarsQueryService.kt b/app-quarkus/src/main/kotlin/io/apim/samples/core/starwars/StarWarsQueryService.kt
@@ -44,11 +44,17 @@ class StarWarsQueryService {
     people.add(Person(4, "Darth Vader", "41.9BBY", planets[0], listOf(films[0], films[1], films[2], films[5]), listOf(), listOf(starships[2])))
   }
 
-  fun getFilms(): List<Film> = films
+  fun getFilms(limit: Int?): List<Film> {
+    return films.subList(0, limit ?: films.size)
+  }
   fun getFilm(id: Int): Film? = films.find { it.id == id }
-  fun getPlanets(): List<Planet> = planets
+  fun getPlanets(limit: Int?): List<Planet> {
+    return planets.subList(0, limit ?: planets.size)
+  }
   fun getPlanet(id: Int): Planet? = planets.find { it.id == id }
-  fun getPeople(): List<Person> = people
+  fun getPeople(limit: Int?): List<Person> {
+    return people.subList(0, limit ?: people.size)
+  }
   fun getPerson(id: Int): Person? = people.find { it.id == id }
   fun getPersonByFilm(film: Film): List<Person> = people.filter { p -> p.films.any { f -> f.id == film.id } }
   fun getPersonByPlanet(planet: Planet): List<Person> = people.filter { p -> p.homeWorld.id == planet.id }

diff --git a/app-quarkus/src/main/kotlin/io/apim/samples/ports/graphql/StarWarsResource.kt b/app-quarkus/src/main/kotlin/io/apim/samples/ports/graphql/StarWarsResource.kt
@@ -14,23 +14,25 @@ class StarWarsResource {
 
   @Query("allFilms")
   @Description("Get all films")
-  fun getAllFilms(): List<Film> = starWarsQueryService.getFilms()
+  fun getAllFilms(@Name("limit") limit: Int?): List<Film> {
+    return starWarsQueryService.getFilms(limit)
+  }
 
   @Query
   @Description("Get a film by id")
   fun getFilm(@Name("filmId") id: Int): Film? = starWarsQueryService.getFilm(id)
 
   @Query("allPlanets")
   @Description("Get all planets")
-  fun getAllPlanets(): List<Planet> = starWarsQueryService.getPlanets()
+  fun getAllPlanets(@Name("limit") limit: Int?): List<Planet> = starWarsQueryService.getPlanets(limit)
 
   @Query
   @Description("Get a planet by id")
   fun getPlanet(@Name("planetId") id: Int): Planet? = starWarsQueryService.getPlanet(id)
 
   @Query("allPeople")
   @Description("Get all people")
-  fun getAllPeople(): List<Person> = starWarsQueryService.getPeople()
+  fun getAllPeople(@Name("limit") limit: Int?): List<Person> = starWarsQueryService.getPeople(limit)
 
   @Query
   @Description("Get a person by id")