Use ternary to handle masking in QueryManager

apollographql · Aug 24, 2024 · 23a8080 · 23a8080
1 parent aa2ed74
commit 23a8080
Showing 1 changed file with 4 additions and 12 deletions.
diff --git a/src/core/QueryManager.ts b/src/core/QueryManager.ts
@@ -1534,24 +1534,16 @@ export class QueryManager<TStore> {
 
   public maskOperation<TData = unknown>(options: MaskOperationOptions<TData>) {
     const { document, data } = options;
-    const { cache } = this;
 
-    if (!this.dataMasking) {
-      return data;
-    }
-
-    return maskOperation(data, document, cache);
+    return this.dataMasking ? maskOperation(data, document, this.cache) : data;
   }
 
   public maskFragment<TData = unknown>(options: MaskFragmentOptions<TData>) {
     const { data, fragment, fragmentName } = options;
-    const { cache } = this;
-
-    if (!this.dataMasking) {
-      return data;
-    }
 
-    return maskFragment(data, fragment, cache, fragmentName);
+    return this.dataMasking ?
+        maskFragment(data, fragment, this.cache, fragmentName)
+      : data;
   }
 
   private fetchQueryByPolicy<TData, TVars extends OperationVariables>(