batter cosmos batch query format

Author: andrejpk

src/cache.ts

@@ -68,9 +68,9 @@ export const createCachingMethods = <DType extends { id: string }>({
     // const response = await container.items.bulk(operations);
     const idList = ids.map((id) => `'${id}'`).join(",");
     const querySpec = {
-      query: `select * from c where c.id in (${idList})`,
-      //   query: "select * from c where c.id in (@ids)",
-      //   parameters: [{ name: "@ids", value: ids }],
+      //   query: `select * from c where c.id in (${idList})`,
+      query: "select * from c where ARRAY_CONTAINS(@ids, c.id)",
+      parameters: [{ name: "@ids", value: ids }],
     };
     const response = await container.items.query(querySpec).fetchAll();
 

src/datasource.ts

@@ -30,8 +30,7 @@ export class CosmosDataSource<TData extends { id: string }, TContext = any>
   dataLoader: CachedMethods<TData>["dataLoader"];
   primeCache: CachedMethods<TData>["primeCache"] = placeholderHandler;
 
-  // base SQL operations aren't cached so directly declared here
-  async findManyByQuery(query: SqlQuerySpec | string, { ttl }: FindArgs) {
+  async findManyByQuery(query: SqlQuerySpec | string, { ttl }: FindArgs = {}) {
     const results = await this.container.items.query<TData>(query).fetchAll();
     // prime these into the dataloader and maybe the cache
     if (this.dataLoader && results.resources) {