kibana/examples/search_examples/server/fibonacci_strategy.ts

/*
 * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
 * or more contributor license agreements. Licensed under the Elastic License
 * 2.0 and the Server Side Public License, v 1; you may not use this file except
 * in compliance with, at your election, the Elastic License 2.0 or the Server
 * Side Public License, v 1.
 */

import uuid from 'uuid';
import { ISearchStrategy } from '@kbn/data-plugin/server';
import { FibonacciRequest, FibonacciResponse } from '../common/types';

export const fibonacciStrategyProvider = (): ISearchStrategy<
  FibonacciRequest,
  FibonacciResponse
> => {
  const responseMap = new Map<string, [number[], number, number]>();
  return {
    search: (request: FibonacciRequest) => {
      const id = request.id ?? uuid();
      const [sequence, total, started] = responseMap.get(id) ?? [
        [],
        request.params?.n ?? 0,
        Date.now(),
      ];
      if (sequence.length < 2) {
        if (total > 0) sequence.push(sequence.length);
      } else {
        const [a, b] = sequence.slice(-2);
        sequence.push(a + b);
      }
      const loaded = sequence.length;
      responseMap.set(id, [sequence, total, started]);
      if (loaded >= total) {
        responseMap.delete(id);
      }

      const isRunning = loaded < total;
      const isPartial = isRunning;
      const took = Date.now() - started;
      const values = sequence.slice(0, loaded);

      // Usually we'd do something like "of()" but for some reason it breaks in tests with the error
      // "You provided an invalid object where a stream was expected." which is why we have to cast
      // down below as well
      return [{ id, loaded, total, isRunning, isPartial, rawResponse: { took, values } }];
    },
    cancel: async (id: string) => {
      responseMap.delete(id);
    },
  } as unknown as ISearchStrategy<FibonacciRequest, FibonacciResponse>;
};