add option to query multiple locations by ids

afuh · Aug 21, 2020 · c99738d · c99738d
1 parent 164051e
commit c99738d
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/graphql/resolvers.js b/graphql/resolvers.js
@@ -33,6 +33,9 @@ const resolvers = {
       const info = handleInfo({ stats })
       return { results, info }
     },
+    locationsByIds: async (_, { ids }, { dataSources }) => {
+      return dataSources.location.locationsByIds({ ids })
+    },
     location: async (_, { id }, { dataSources }) => {
       return dataSources.location.location({ id })
     },

diff --git a/graphql/sources.js b/graphql/sources.js
@@ -28,6 +28,10 @@ class Location extends RESTDataSource {
     const data = await this.get('/', { ...filter, page })
     return data
   }
+  async locationsByIds({ ids }) {
+    const data = await this.get('/' + ids)
+    return Array.isArray(data) ? data : [data]
+  }
   async location({ id }) {
     const data = await this.get('/' + id)
     return data

diff --git a/graphql/typeDefs.js b/graphql/typeDefs.js
@@ -22,6 +22,11 @@ const typeDefs = gql`
     """
     locations(page: Int, filter: FilterLocation): Locations
 
+    """
+    Get a list of locations selected by ids
+    """
+    locationsByIds(ids: [ID]): [Location]
+
     """
     Get a specific episode by ID
     """