feat: add limit argument to relatedResources

newrelic · Aug 13, 2020 · cc22434 · cc22434
1 parent 489d004
commit cc22434
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/plugins/gatsby-source-swiftype/gatsby-node.js b/plugins/gatsby-source-swiftype/gatsby-node.js
@@ -65,9 +65,19 @@ exports.createResolvers = ({ createResolvers }) => {
   createResolvers({
     Mdx: {
       relatedResources: {
+        args: {
+          limit: {
+            type: 'Int',
+            defaultValue: 5,
+          },
+        },
         type: ['RelatedResource!'],
-        resolve(source, _args, context) {
-          return context.nodeModel.getNodesByIds({ ids: source.children });
+        resolve(source, args, context) {
+          const { limit } = args;
+
+          return context.nodeModel
+            .getNodesByIds({ ids: source.children })
+            .slice(0, Math.max(limit, 0));
         },
       },
     },