query now returns cosmos result info

Author: andrejpk

src/datasource.ts

@@ -1,7 +1,7 @@
 import { DataSource } from "apollo-datasource";
 import { ApolloError } from "apollo-server-errors";
 import { InMemoryLRUCache, KeyValueCache } from "apollo-server-caching";
-import { Container, SqlQuerySpec } from "@azure/cosmos";
+import { Container, SqlQuerySpec, FeedOptions } from "@azure/cosmos";
 import { Logger } from "./helpers";
 
 import { isCosmosDbContainer } from "./helpers";
@@ -16,12 +16,21 @@ const placeholderHandler = () => {
   throw new Error("DataSource not initialized");
 };
 
+export interface CosmosQueryDbArgs {
+  /** Maps to CosmosDB feed/request options for parameters like maxItemCount
+   * See https://docs.microsoft.com/en-us/javascript/api/%40azure/cosmos/feedoptions?view=azure-node-latest
+   */
+  requestOptions?: FeedOptions;
+}
+
+export type QueryFindArgs = FindArgs & CosmosQueryDbArgs;
+
 export class CosmosDataSource<TData extends { id: string }, TContext = any>
   extends DataSource<TContext>
   implements CachedMethods<TData> {
   container: Container;
   context?: TContext;
-  private options: CosmosDataSourceOptions;
+  options: CosmosDataSourceOptions;
   // these get set by the initializer but they must be defined or nullable after the constructor
   // runs, so we guard against using them before init
   findOneById: CachedMethods<TData>["findOneById"] = placeholderHandler;
@@ -30,13 +39,26 @@ export class CosmosDataSource<TData extends { id: string }, TContext = any>
   dataLoader: CachedMethods<TData>["dataLoader"];
   primeLoader: CachedMethods<TData>["primeLoader"] = placeholderHandler;
 
-  async findManyByQuery(query: SqlQuerySpec | string, { ttl }: FindArgs = {}) {
-    const results = await this.container.items.query<TData>(query).fetchAll();
+  /**
+   * Same as findManyByQuery but returns the entire CosmosDB response which is sometimes useful
+   * @param query
+   * @param param1
+   */
+  async findManyByQuery(
+    query: string | SqlQuerySpec,
+    { ttl, requestOptions }: QueryFindArgs = {}
+  ) {
+    this.options?.logger?.debug(
+      `findManyByQuery: CosmosQuery: ${(query as any).query || query}`
+    );
+    const results = await this.container.items
+      .query<TData>(query, requestOptions)
+      .fetchAll();
     // prime these into the dataloader and maybe the cache
     if (this.dataLoader && results.resources) {
       this.primeLoader(results.resources, ttl);
     }
-    return results.resources;
+    return results;
   }
 
   constructor(container: Container, options: CosmosDataSourceOptions = {}) {

src/index.ts

@@ -1,4 +1,4 @@
-import { CosmosDataSource } from "./datasource";
+import { CosmosDataSource, QueryFindArgs } from "./datasource";
 import { FindArgs } from "./cache";
 
-export { CosmosDataSource, FindArgs };
+export { CosmosDataSource, FindArgs, QueryFindArgs };