update cache when pdf is added

detroitledger · May 5, 2021 · 4483546 · 4483546
1 parent 6a879d5
commit 4483546
Showing 1 changed file with 22 additions and 1 deletion.
diff --git a/src/containers/ManagePdfs.js b/src/containers/ManagePdfs.js
@@ -14,11 +14,32 @@ const ADD_PDF = gql`
   }
 `;
 
+const ADD_PDF_OPTS = {
+  update(cache, { data: { addPdf } }) {
+    cache.modify({
+      fields: {
+        pdfs(existingPdfs = []) {
+          const newPdfRef = cache.writeFragment({
+            data: addPdf,
+            fragment: gql`
+              fragment NewPdf on Pdf {
+                id
+                type
+              }
+            `,
+          });
+          return [...existingPdfs, newPdfRef];
+        },
+      },
+    });
+  },
+};
+
 const ManagePdfs = () => {
   const [
     addPdf,
     { loading: addPdfLoading, error: addPdfError, data: addPdfData },
-  ] = useMutation(ADD_PDF);
+  ] = useMutation(ADD_PDF, ADD_PDF_OPTS);
 
   if (addPdfError) {
     console.log(