Update ipcLink.ts

Author: charlesvien

packages/electron-trpc/src/renderer/__tests__/ipcLink.test.ts

@@ -5,7 +5,7 @@ import superjson from "superjson";
 import { beforeEach, describe, expect, test, vi } from "vitest";
 import z from "zod";
 import type { RendererGlobalElectronTRPC } from "../../types";
-import { ipcLink } from "../ipcLink";
+import { _resetIPCClient, ipcLink } from "../ipcLink";
 
 const t = initTRPC.create({ transformer: superjson });
 const router = t.router({
@@ -38,6 +38,7 @@ const electronTRPC: RendererGlobalElectronTRPC = {} as any;
 let handlers: ((message: TRPCResponseMessage) => void)[] = [];
 beforeEach(() => {
   handlers = [];
+  _resetIPCClient();
   electronTRPC.sendMessage = vi.fn();
   electronTRPC.onMessage = vi.fn().mockImplementation((handler) => {
     handlers.push(handler);

packages/electron-trpc/src/renderer/ipcLink.ts

@@ -104,11 +104,25 @@ export type IPCLinkOptions<TRouter extends AnyTRPCRouter> = TransformerOptions<
   inferTRPCClientTypes<TRouter>
 >;
 
+let singletonClient: IPCClient | null = null;
+
+function getOrCreateClient(): IPCClient {
+  if (!singletonClient) {
+    singletonClient = new IPCClient();
+  }
+  return singletonClient;
+}
+
+/** @internal Reset singleton for testing only */
+export function _resetIPCClient() {
+  singletonClient = null;
+}
+
 export function ipcLink<TRouter extends AnyTRPCRouter>(
   opts?: IPCLinkOptions<TRouter>,
 ): TRPCLink<TRouter> {
   return () => {
-    const client = new IPCClient();
+    const client = getOrCreateClient();
     const transformer = getTransformer(opts?.transformer);
 
     return ({ op }) => {