Add handler to vscode

apollographql · Dec 20, 2024 · 887418d · 887418d
1 parent 6d00546
commit 887418d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/extension/vscode/client.ts b/src/extension/vscode/client.ts
@@ -110,8 +110,12 @@ function registerClient(
   function getMutationsForClient() {
     return getMutations(getClient()?.queryManager.mutationStore ?? {});
   }
+  function getMemoryInternalsForClient() {
+    return getClient()?.getMemoryInternals?.();
+  }
   wsRpcHandler("getQueries", getQueriesForClient, { signal });
   wsRpcHandler("getMutations", getMutationsForClient, { signal });
+  wsRpcHandler("getMemoryInternals", getMemoryInternalsForClient, { signal });
   wsRpcHandler("getCache", () => getClient()?.cache.extract(true) ?? {}, {
     signal,
   });

diff --git a/src/extension/vscode/server.ts b/src/extension/vscode/server.ts
@@ -45,6 +45,10 @@ export function runServer(
     clients.get(id)!.rpcClient.request("getCache", id)
   );
 
+  handleRpc("getMemoryInternals", (id) =>
+    clients.get(id)!.rpcClient.request("getMemoryInternals", id)
+  );
+
   server.on("connection", function connection(ws) {
     let id: string | undefined;
     const wsAdapter = createSocketMessageAdapter(ws);