Merge pull request #4 from jean-leonco/chore/prime-cache

chore(prime-cache): add missing prime function to createLoader
woovibr · Jan 6, 2021 · 1d24dcb · 1d24dcb
2 parents 61aa78a + c93222d
commit 1d24dcb
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/createLoader.ts b/src/createLoader.ts
@@ -82,6 +82,12 @@ export const createLoader = <
 
   const clearCache = ({ dataloaders }: Context, id: string) => dataloaders[loaderName].clear(id.toString());
 
+  const primeCache = ({ dataloaders }: Context, id: string, data: Value) =>
+    dataloaders[loaderName].prime(id.toString(), data);
+
+  const clearAndPrimeCache = (context: Context, id: string, data: Value) =>
+    clearCache(context, id) && primeCache(context, id, data);
+
   const loadAll = withConnectionCursor(model, load, (context: Context, args: FilteredConnectionArguments) => {
     const builtMongoConditions = buildMongoConditionsFromFilters(context, args.filters, filterMapping as any);
 
@@ -105,6 +111,8 @@ export const createLoader = <
     },
     getLoader,
     clearCache,
+    primeCache,
+    clearAndPrimeCache,
     load,
     loadAll,
   };