useApolloCache

useApolloCache.ts

import { DocumentNode, useApolloClient } from '@apollo/client';

interface IApolloCacheReturn {
  deleteApolloCacheForDocuments: (
    documents: Array<DocumentNode>
  ) => Promise<void>;
}
export const useApolloCache = (): IApolloCacheReturn => {
  const client = useApolloClient();
  const getQueryNameFromDocument = (document: DocumentNode) => {
    const definition = document.definitions[0];
    if (definition && 'name' in definition) {
      return definition.name?.value;
    }
  };
  const deleteApolloCacheForDocuments = async (
    documents: Array<DocumentNode>
  ) => {
    const documentNames = documents.map((doc) => getQueryNameFromDocument(doc));

    documentNames.forEach((docName) => {
      if (docName) {
        client.cache.evict({ fieldName: docName });
      }
    });
    client.cache.gc();
  };
  return { deleteApolloCacheForDocuments };
};