code cleanup

aws-amplify · Oct 21, 2021 · 475c409 · 475c409
1 parent 326c0d7
commit 475c409
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/packages/graphql-docs-generator/src/generator/getFields.ts b/packages/graphql-docs-generator/src/generator/getFields.ts
@@ -64,12 +64,19 @@ export default function getFields(
 }
 
 function adjustDepth(field, depth) {
+  if (isGraphQLAggregateField(field)) {
+    return depth + 1;
+  }
+  return depth - 1;
+}
+
+function isGraphQLAggregateField(field) {
   if (
     field &&
     field.name == 'aggregateItems' &&
     field.type?.ofType?.name == 'SearchableAggregateResult'
   ) {
-    return depth + 1;
+    return true;
   }
-  return depth - 1;
-}
+  return false;
+}