feat: Support for Elasticsearch "Collapse" query

* add collapse field * Update SearchBody.ts * Update SearchHitItem.ts
graphql-compose · Jan 15, 2022 · a06664f · a06664f
1 parent 4fdfcec
commit a06664f
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 2 deletions.
diff --git a/src/elasticDSL/SearchBody.ts b/src/elasticDSL/SearchBody.ts
@@ -20,6 +20,7 @@ export function getSearchBodyITC<TContext>(
  description,
  fields: {
  query: { type: () => getQueryITC(opts) },
+ collapse: 'JSON',
  aggs: { type: () => getAggsITC(opts) },
  size: 'Int',
  from: 'Int',

diff --git a/src/resolvers/search.ts b/src/resolvers/search.ts
@@ -66,9 +66,10 @@ export default function createSearchResolver<TSource, TContext>(
  argsConfigMap.query = searchITC.getField('query');
  argsConfigMap.aggs = searchITC.getField('aggs');
  argsConfigMap.sort = searchITC.getField('sort');
+ argsConfigMap.collapse = searchITC.getField('collapse');
  argsConfigMap.highlight = searchITC.getField('highlight');
 
- const topLevelArgs = ['q', 'query', 'sort', 'limit', 'skip', 'aggs', 'highlight', 'opts'];
+ const topLevelArgs = ['q', 'query', 'collapse', 'sort', 'limit', 'skip', 'aggs', 'highlight', 'opts'];
  argsConfigMap.opts = schemaComposer
  .createInputTC({
  name: `${sourceTC.getTypeName()}Opts`,
@@ -145,6 +146,11 @@ export default function createSearchResolver<TSource, TContext>(
  args.body.query = args.query;
  delete args.query;
  }
+
+ if (args.collapse) {
+ args.body.collapse = args.collapse;
+ delete args.collapse;
+ }
 
  if (args.aggs) {
  args.body.aggs = args.aggs;
@@ -191,7 +197,7 @@ export default function createSearchResolver<TSource, TContext>(
  return res;
  },
  })
- .reorderArgs(['q', 'query', 'sort', 'limit', 'skip', 'aggs']);
+ .reorderArgs(['q', 'query', 'collapse', 'sort', 'limit', 'skip', 'aggs']);
 }
 
 export function toDottedList(projection: ProjectionType, prev?: string[]): string[] | boolean {

diff --git a/src/types/SearchHitItem.ts b/src/types/SearchHitItem.ts
@@ -31,6 +31,11 @@ export function getSearchHitItemTC<TContext>(
 
  _version: 'Int',
 
+ inner_hits: {
+ type: 'JSON',
+ description: 'Returns data only if `args.collapse` is provided',
+ },
+
  highlight: {
  type: 'JSON',
  description: 'Returns data only if `args.highlight` is provided',