From 2f0022f9123f179926ac1a3f8242a0fa2e9aacd4 Mon Sep 17 00:00:00 2001
From: Max Richter <jim-x@web.de>
Date: Fri, 26 Apr 2024 17:57:32 +0200
Subject: [PATCH] feat: refactor performance collection

---
 app/src/lib/performance/Monitor.svelte        |  37 ++-
 .../lib/performance/PerformanceViewer.svelte  | 286 ++++++++++++------
 app/src/lib/performance/store.ts              |  40 +--
 app/src/lib/result-viewer/Viewer.svelte       |  55 ++--
 app/src/lib/runtime-executor.ts               |  18 +-
 app/src/routes/+page.svelte                   |  30 +-
 packages/types/src/components.ts              |  13 +-
 7 files changed, 306 insertions(+), 173 deletions(-)

diff --git a/app/src/lib/performance/Monitor.svelte b/app/src/lib/performance/Monitor.svelte
index 0ba767b..c942f0c 100644
--- a/app/src/lib/performance/Monitor.svelte
+++ b/app/src/lib/performance/Monitor.svelte
@@ -1,9 +1,32 @@
 <script lang="ts">
   export let points: number[];
 
-  $: max = Math.max(...points);
-  $: min = Math.min(...points);
+  export let type = "ms";
+  export let title = "Performance";
+  export let max: number | undefined = undefined;
+  export let min: number | undefined = undefined;
+
+  function getMax(m?: number) {
+    if (type === "%") {
+      return 100;
+    }
+
+    if (m !== undefined) {
+      if (m < 1) {
+        return Math.floor(m * 100) / 100;
+      }
+      if (m < 10) {
+        return Math.floor(m * 10) / 10;
+      }
+      return Math.floor(m);
+    }
+
+    return 1;
+  }
+
   function constructPath() {
+    max = max !== undefined ? max : Math.max(...points);
+    min = min !== undefined ? min : Math.min(...points);
     return points
       .map((point, i) => {
         const x = (i / (points.length - 1)) * 100;
@@ -15,11 +38,13 @@
 </script>
 
 <div class="wrapper">
-  <p>Runtime Execution</p>
-  <span class="min">{min}ms</span>
-  <span class="max">{max}ms</span>
+  <p>{title}</p>
+  <span class="min">{Math.floor(min || 0)}{type}</span>
+  <span class="max">{getMax(max)}{type}</span>
   <svg preserveAspectRatio="none" viewBox="0 0 100 100">
-    <polyline vector-effect="non-scaling-stroke" points={constructPath()} />
+    {#key points}
+      <polyline vector-effect="non-scaling-stroke" points={constructPath()} />
+    {/key}
   </svg>
 </div>
 
diff --git a/app/src/lib/performance/PerformanceViewer.svelte b/app/src/lib/performance/PerformanceViewer.svelte
index 29364d0..e2248fb 100644
--- a/app/src/lib/performance/PerformanceViewer.svelte
+++ b/app/src/lib/performance/PerformanceViewer.svelte
@@ -1,168 +1,265 @@
 <script lang="ts">
-  import { browser } from "$app/environment";
   import Monitor from "./Monitor.svelte";
   import { humanizeNumber } from "$lib/helpers";
   import { Checkbox } from "@nodes/ui";
-
-  type PerformanceData = {
-    total: Record<string, number>;
-    runs: Record<string, number[]>[];
-  };
+  import localStore from "$lib/helpers/localStore";
+  import { type PerformanceData } from "./store";
 
   export let data: PerformanceData;
-  export let viewer: PerformanceData;
-  let lastRunOnly = true;
+
+  let activeType = localStore<string>("nodes.performance.active-type", "total");
+  let showAverage = true;
+
+  function getAverage(key: string) {
+    return (
+      data
+        .map((run) => run[key]?.[0])
+        .filter((v) => v !== undefined)
+        .reduce((acc, run) => acc + run, 0) / data.length
+    );
+  }
+
+  function round(v: number) {
+    if (v < 1) {
+      return Math.floor(v * 100) / 100;
+    }
+    if (v < 10) {
+      return Math.floor(v * 10) / 10;
+    }
+    return Math.floor(v);
+  }
+
+  function getAverages() {
+    let lastRun = data.at(-1);
+    if (!lastRun) return {};
+    return Object.keys(lastRun).reduce(
+      (acc, key) => {
+        acc[key] = getAverage(key);
+        return acc;
+      },
+      {} as Record<string, number>,
+    );
+  }
+
+  function getLast(key: string) {
+    return data.at(-1)?.[key][0] || 0;
+  }
+
+  function getLasts() {
+    return data.at(-1) || {};
+  }
 
   function getTotalPerformance(onlyLast = false) {
     if (onlyLast) {
       return (
-        data.runs.at(-1).runtime[0] + viewer.runs.at(-1)["create-geometries"][0]
+        getLast("runtime") +
+        getLast("create-geometries") +
+        getLast("worker-transfer")
       );
     }
-    return data.total.runtime + viewer.total["create-geometries"];
-  }
-
-  function count(input?: number | number[]) {
-    if (!input) return 0;
-    if (Array.isArray(input))
-      return input.reduce((acc, val) => acc + val, 0) / input.length;
-    return input;
+    return (
+      getAverage("runtime") +
+      getAverage("create-geometries") +
+      getAverage("worker-transfer")
+    );
   }
 
   function getCacheRatio(onlyLast = false) {
-    let ratio = onlyLast
-      ? count(data.runs.at(-1)?.["cache-hit"])
-      : count(data.total["cache-hit"]);
+    let ratio = onlyLast ? getLast("cache-hit") : getAverage("cache-hit");
 
-    return `${Math.floor(ratio * 100)}%`;
+    return Math.floor(ratio * 100);
   }
 
+  const viewerKeys = [
+    "total-vertices",
+    "total-faces",
+    "create-geometries",
+    "split-result",
+  ];
+
   function getPerformanceData(onlyLast: boolean = false) {
-    if (onlyLast) {
-      return Object.entries(data.runs.at(-1))
-        .map(([key, value]) => [key, value[0]])
-        .filter(
-          ([key]) =>
-            !key.startsWith("node/") &&
-            key !== "total" &&
-            !key.includes("cache"),
-        )
-        .sort((a, b) => b[1] - a[1]);
-    }
-    return Object.entries(data.total)
-      .sort((a, b) => b[1] - a[1])
+    let data = onlyLast ? getLasts() : getAverages();
+
+    return Object.entries(data)
       .filter(
         ([key]) =>
-          !key.startsWith("node/") && key !== "total" && !key.includes("cache"),
-      );
+          !key.startsWith("node/") &&
+          key !== "total" &&
+          !key.includes("cache") &&
+          !viewerKeys.includes(key),
+      )
+      .sort((a, b) => b[1] - a[1]);
   }
 
   function getNodePerformanceData(onlyLast: boolean = false) {
-    if (onlyLast) {
-      return Object.entries(data.runs.at(-1))
-        .map(([key, value]) => [key, value[0]])
-        .filter(([key]) => key.startsWith("node/"))
-        .sort((a, b) => b[1] - a[1]);
-    }
-    return Object.entries(data.total)
+    let data = onlyLast ? getLasts() : getAverages();
+
+    return Object.entries(data)
       .filter(([key]) => key.startsWith("node/"))
       .sort((a, b) => b[1] - a[1]);
   }
 
   function getViewerPerformanceData(onlyLast: boolean = false) {
-    if (onlyLast) {
-      return Object.entries(viewer.runs.at(-1))
-        .map(([key, value]) => [key, value[0]])
-        .filter(([key]) => key !== "total-vertices" && key !== "total-faces")
-        .sort((a, b) => b[1] - a[1]);
-    }
-    return Object.entries(viewer.total)
-      .filter(([key]) => key !== "total-vertices" && key !== "total-faces")
+    let data = onlyLast ? getLasts() : getAverages();
+    return Object.entries(data)
+      .filter(
+        ([key]) =>
+          key !== "total-vertices" &&
+          key !== "total-faces" &&
+          viewerKeys.includes(key),
+      )
       .sort((a, b) => b[1] - a[1]);
   }
 
-  function constructPoints(key: keyof (typeof data.runs)[0]) {
-    return data.runs.map((run, i) => {
-      return run[key][0];
+  function getTotalPoints() {
+    if (showAverage) {
+      return data.map((run) => {
+        return (
+          run["runtime"].reduce((acc, v) => acc + v, 0) +
+          run["create-geometries"].reduce((acc, v) => acc + v, 0) +
+          run["worker-transfer"].reduce((acc, v) => acc + v, 0)
+        );
+      });
+    }
+
+    return data.map((run) => {
+      return (
+        run["runtime"][0] +
+        run["create-geometries"][0] +
+        run["worker-transfer"][0]
+      );
     });
   }
+
+  function constructPoints(key: string) {
+    if (key === "total") {
+      return getTotalPoints();
+    }
+    return data.map((run) => {
+      if (key in run) {
+        if (showAverage) {
+          return run[key].reduce((acc, v) => acc + v, 0) / run[key].length;
+        } else {
+          return run[key][0];
+        }
+      }
+      return 0;
+    });
+  }
+
+  function getTitle(t: string) {
+    if (t.includes("/")) {
+      return `Node ${t.split("/").slice(-1).join("/")}`;
+    }
+
+    return t
+      .split("-")
+      .map((v) => v[0].toUpperCase() + v.slice(1))
+      .join(" ");
+  }
 </script>
 
-{#key data}
-  {#if browser}
-    <Monitor points={constructPoints("runtime")} />
+{#key $activeType && data}
+  {#if $activeType === "cache-hit"}
+    <Monitor
+      title="Cache Hits"
+      points={constructPoints($activeType)}
+      min={0}
+      max={1}
+      type="%"
+    />
+  {:else}
+    <Monitor
+      title={getTitle($activeType)}
+      points={constructPoints($activeType)}
+    />
   {/if}
 
   <div class="p-4">
     <div class="flex items-center gap-2">
-      <Checkbox id="show-total" bind:value={lastRunOnly} />
+      <Checkbox id="show-total" bind:value={showAverage} />
       <label for="show-total">Show Average</label>
     </div>
-    {#if data.runs.length !== 0}
+    {#if data.length !== 0}
       <h3>General</h3>
       <table>
         <tr>
           <td>
-            {Math.floor(getTotalPerformance(!lastRunOnly) * 100) / 100}<span
-              >ms</span
+            {round(getTotalPerformance(!showAverage))}<span>ms</span>
+          </td>
+          <td
+            class:active={$activeType === "total"}
+            on:click={() => ($activeType = "total")}
+          >
+            total<span
+              >({Math.floor(1000 / getTotalPerformance(showAverage))}fps)</span
             >
           </td>
-          <td>total</td>
         </tr>
-        {#each getPerformanceData(!lastRunOnly) as [key, value]}
+        {#each getPerformanceData(!showAverage) as [key, value]}
           <tr>
             <td>
-              {Math.floor(value * 100) / 100}<span>ms</span>
+              {round(value)}<span>ms</span>
             </td>
-            <td>
+            <td
+              class:active={$activeType === key}
+              on:click={() => ($activeType = key)}
+            >
               {key}
             </td>
           </tr>
         {/each}
 
         <tr>
-          <td> {getCacheRatio(!lastRunOnly)} </td>
-          <td>cache hit</td>
-        </tr>
-
-        <tr>
-          <td>{data.runs.length}</td>
+          <td>{data.length}</td>
           <td>Samples</td>
         </tr>
 
         <h3>Nodes</h3>
-        {#each getNodePerformanceData(!lastRunOnly) as [key, value]}
+        <tr>
+          <td> {getCacheRatio(!showAverage)}<span>%</span> </td>
+          <td
+            class:active={$activeType === "cache-hit"}
+            on:click={() => ($activeType = "cache-hit")}>cache hits</td
+          >
+        </tr>
+        {#each getNodePerformanceData(!showAverage) as [key, value]}
           <tr>
             <td>
-              {Math.floor(value * 100) / 100}<span>ms</span>
+              {round(value)}<span>ms</span>
             </td>
-            <td>
+
+            <td
+              class:active={$activeType === key}
+              on:click={() => ($activeType = key)}
+            >
               {key.split("/").slice(-1).join("/")}
             </td>
           </tr>
         {/each}
 
-        {#if viewer.runs.length}
-          <h3>Viewer</h3>
+        <h3>Viewer</h3>
+        <tr>
+          <td>{humanizeNumber(getLast("total-vertices"))}</td>
+          <td>Vertices</td>
+        </tr>
+        <tr>
+          <td>{humanizeNumber(getLast("total-faces"))}</td>
+          <td>Faces</td>
+        </tr>
+        {#each getViewerPerformanceData(!showAverage) as [key, value]}
           <tr>
-            <td>{humanizeNumber(viewer.runs.at(-1)?.["total-vertices"])}</td>
-            <td>Vertices</td>
+            <td>
+              {round(value)}<span>ms</span>
+            </td>
+            <td
+              class:active={$activeType === key}
+              on:click={() => ($activeType = key)}
+            >
+              {key.split("/").slice(-1).join("/")}
+            </td>
           </tr>
-          <tr>
-            <td>{humanizeNumber(viewer.runs.at(-1)?.["total-faces"])}</td>
-            <td>Faces</td>
-          </tr>
-          {#each getViewerPerformanceData(!lastRunOnly) as [key, value]}
-            <tr>
-              <td>
-                {Math.floor(value * 100) / 100}<span>ms</span>
-              </td>
-              <td>
-                {key.split("/").slice(-1).join("/")}
-              </td>
-            </tr>
-          {/each}
-        {/if}
+        {/each}
       </table>
     {:else}
       <p>No runs available</p>
@@ -185,6 +282,9 @@
     padding-right: 10px;
     padding-block: 5px;
   }
+  td.active {
+    font-weight: bold;
+  }
   tr > td:nth-child(1) {
     text-align: right;
   }
diff --git a/app/src/lib/performance/store.ts b/app/src/lib/performance/store.ts
index 6dbbe48..63f9019 100644
--- a/app/src/lib/performance/store.ts
+++ b/app/src/lib/performance/store.ts
@@ -1,26 +1,24 @@
 import { readable, type Readable } from "svelte/store";
 
-export type PerformanceData = {
-  total: Record<string, number>;
-  runs: Record<string, number[]>[];
-}
+export type PerformanceData = Record<string, number[]>[];
 
 export interface PerformanceStore extends Readable<PerformanceData> {
   startRun(): void;
   stopRun(): void;
   addPoint(name: string, value?: number): void;
+  mergeData(data: PerformanceData[number]): void;
   get: () => PerformanceData;
 }
 
 export function createPerformanceStore(): PerformanceStore {
 
-  let data: PerformanceData = { total: {}, runs: [] };
+  let data: PerformanceData = [];
 
   let currentRun: Record<string, number[]> | undefined;
 
   let set: (v: PerformanceData) => void;
 
-  const { subscribe } = readable<PerformanceData>({ total: {}, runs: [] }, (_set) => {
+  const { subscribe } = readable<PerformanceData>([], (_set) => {
     set = _set;
   });
 
@@ -30,19 +28,8 @@ export function createPerformanceStore(): PerformanceStore {
 
   function stopRun() {
     if (currentRun) {
-      // Calculate total
-      Object.keys(currentRun).forEach((name) => {
-        if (!currentRun?.[name]?.length) return;
-        let runTotal = currentRun[name].reduce((a, b) => a + b, 0) / currentRun[name].length;
-        if (!data.total[name]) {
-          data.total[name] = runTotal;
-        } else {
-          data.total[name] = (data.total[name] + runTotal) / 2;
-        }
-      });
-
-      data.runs.push(currentRun);
-      data.runs = data.runs.slice(-100);
+      data.push(currentRun);
+      data = data.slice(-100);
       currentRun = undefined;
       if (set) set(data);
     }
@@ -58,11 +45,26 @@ export function createPerformanceStore(): PerformanceStore {
     return data;
   }
 
+  function mergeData(newData: PerformanceData[number]) {
+
+    let r = currentRun;
+    if (!r) return;
+
+    Object.keys(newData).forEach((name) => {
+      if (name in r) {
+        r[name].push(...newData[name]);
+      } else {
+        r[name] = newData[name];
+      }
+    });
+  }
+
   return {
     subscribe,
     startRun,
     stopRun,
     addPoint,
+    mergeData,
     get
   }
 }
diff --git a/app/src/lib/result-viewer/Viewer.svelte b/app/src/lib/result-viewer/Viewer.svelte
index 9faff43..824fde4 100644
--- a/app/src/lib/result-viewer/Viewer.svelte
+++ b/app/src/lib/result-viewer/Viewer.svelte
@@ -17,9 +17,6 @@
   let geometries: BufferGeometry[] = [];
   let lines: Vector3[][] = [];
 
-  let totalVertices = 0;
-  let totalFaces = 0;
-
   function fastArrayHash(arr: ArrayBuffer) {
     let ints = new Uint8Array(arr);
 
@@ -41,14 +38,11 @@
     geometry = new BufferGeometry(),
   ): BufferGeometry {
     // Extract data from the encoded array
-    let index = 0;
-    const geometryType = encodedData[index++];
+    let index = 1;
+    // const geometryType = encodedData[index++];
     const vertexCount = encodedData[index++];
     const faceCount = encodedData[index++];
 
-    totalVertices += vertexCount;
-    totalFaces += faceCount;
-
     // Indices
     const indicesEnd = index + faceCount * 3;
     const indices = encodedData.subarray(index, indicesEnd);
@@ -62,10 +56,24 @@
     );
     index = index + vertexCount * 3;
     let hash = fastArrayHash(vertices);
+    let posAttribute = geometry.getAttribute(
+      "position",
+    ) as BufferAttribute | null;
+
     if (geometry.userData?.hash === hash) {
       return geometry;
     }
 
+    if (posAttribute && posAttribute.count === vertexCount) {
+      posAttribute.set(vertices, 0);
+      posAttribute.needsUpdate = true;
+    } else {
+      geometry.setAttribute(
+        "position",
+        new Float32BufferAttribute(vertices, 3),
+      );
+    }
+
     const normals = new Float32Array(
       encodedData.buffer,
       index * 4,
@@ -81,20 +89,6 @@
       geometry.setIndex([...indices]);
     }
 
-    let posAttribute = geometry.getAttribute(
-      "position",
-    ) as BufferAttribute | null;
-
-    if (posAttribute && posAttribute.count === vertexCount) {
-      posAttribute.set(vertices, 0);
-      posAttribute.needsUpdate = true;
-    } else {
-      geometry.setAttribute(
-        "position",
-        new Float32BufferAttribute(vertices, 3),
-      );
-    }
-
     const normalsAttribute = geometry.getAttribute(
       "normal",
     ) as BufferAttribute | null;
@@ -172,15 +166,14 @@
     return positions;
   }
 
-  export const update = function updateGeometries(result: Int32Array) {
+  export let result: Int32Array;
+  $: result && updateGeometries();
+  function updateGeometries() {
     let a = performance.now();
     const inputs = parse_args(result);
     let b = performance.now();
     perf?.addPoint("split-result", b - a);
 
-    totalVertices = 0;
-    totalFaces = 0;
-
     if ($AppSettings.showStemLines) {
       a = performance.now();
       lines = inputs
@@ -194,11 +187,17 @@
       perf?.addPoint("create-lines", b - a);
     }
 
+    let totalVertices = 0;
+    let totalFaces = 0;
+
     a = performance.now();
     geometries = inputs
       .map((input, i) => {
         if (input[0] === 1) {
-          return createGeometryFromEncodedData(input, geometries[i]);
+          let geo = createGeometryFromEncodedData(input);
+          totalVertices += geo.userData.vertexCount;
+          totalFaces += geo.userData.faceCount;
+          return geo;
         }
       })
       .filter(Boolean) as BufferGeometry[];
@@ -206,7 +205,7 @@
     perf?.addPoint("create-geometries", b - a);
     perf?.addPoint("total-vertices", totalVertices);
     perf?.addPoint("total-faces", totalFaces);
-  };
+  }
 </script>
 
 <Canvas>
diff --git a/app/src/lib/runtime-executor.ts b/app/src/lib/runtime-executor.ts
index 9a2dee2..a0e0ce2 100644
--- a/app/src/lib/runtime-executor.ts
+++ b/app/src/lib/runtime-executor.ts
@@ -45,7 +45,7 @@ export class MemoryRuntimeExecutor implements RuntimeExecutor {
 
   perf?: PerformanceStore;
 
-  constructor(private registry: NodeRegistry, private cache?: RuntimeCache) { }
+  constructor(private registry: NodeRegistry, private cache?: RuntimeCache<Int32Array>) { }
 
   private async getNodeDefinitions(graph: Graph) {
 
@@ -215,6 +215,7 @@ export class MemoryRuntimeExecutor implements RuntimeExecutor {
         let inputHash = `node-${node.id}-${fastHashArrayBuffer(encoded_inputs)}`;
         b = performance.now();
         this.perf?.addPoint("hash-inputs", b - a);
+
         let cachedValue = this.cache?.get(inputHash);
         if (cachedValue !== undefined) {
           log.log(`Using cached value for ${node_type.id || node.id}`);
@@ -253,6 +254,10 @@ export class MemoryRuntimeExecutor implements RuntimeExecutor {
 
     this.perf?.stopRun();
 
+    if (this.cache) {
+      this.cache.size = sortedNodes.length * 2;
+    }
+
     return res as unknown as Int32Array;
 
   }
@@ -261,15 +266,18 @@ export class MemoryRuntimeExecutor implements RuntimeExecutor {
 
 export class MemoryRuntimeCache implements RuntimeCache {
 
-  private cache: Record<string, unknown> = {};
+  private cache: [string, unknown][] = [];
+  size = 50;
+
   get<T>(key: string): T | undefined {
-    return this.cache[key] as T;
+    return this.cache.find(([k]) => k === key)?.[1] as T;
   }
   set<T>(key: string, value: T): void {
-    this.cache[key] = value;
+    this.cache.push([key, value]);
+    this.cache = this.cache.slice(-this.size);
   }
   clear(): void {
-    this.cache = {};
+    this.cache = [];
   }
 
 }
diff --git a/app/src/routes/+page.svelte b/app/src/routes/+page.svelte
index 70891f9..b07fd65 100644
--- a/app/src/routes/+page.svelte
+++ b/app/src/routes/+page.svelte
@@ -20,18 +20,15 @@
   import GraphSettings from "$lib/settings/panels/GraphSettings.svelte";
   import NestedSettings from "$lib/settings/panels/NestedSettings.svelte";
   import { createPerformanceStore } from "$lib/performance";
-  import { type PerformanceData } from "$lib/performance/store";
 
   const nodeRegistry = new RemoteNodeRegistry("");
   const workerRuntime = new WorkerRuntimeExecutor();
 
-  let performanceData: PerformanceData;
-  let viewerPerformance = createPerformanceStore();
+  let performanceStore = createPerformanceStore();
 
-  let res: Int32Array;
   let activeNode: Node | undefined;
 
-  let updateViewer: (arg: Int32Array) => void;
+  let graphResult: Int32Array;
 
   let graph = localStorage.getItem("graph")
     ? JSON.parse(localStorage.getItem("graph")!)
@@ -69,23 +66,21 @@
     try {
       let a = performance.now();
       // res = await remoteRuntime.execute(_graph, _settings);
-      let res = await workerRuntime.execute(_graph, _settings);
+      graphResult = await workerRuntime.execute(_graph, _settings);
       let b = performance.now();
-      updateViewer(res);
       let perfData = await workerRuntime.getPerformanceData();
-      let lastRun = perfData.runs?.at(-1);
+      let lastRun = perfData.at(-1);
       if (lastRun) {
-        perfData.total["worker-transfer"] = b - a - lastRun.runtime[0];
         lastRun["worker-transfer"] = [b - a - lastRun.runtime[0]];
+        performanceStore.mergeData(lastRun);
       }
-      performanceData = perfData;
       isWorking = false;
     } catch (error) {
       console.log("errors", error);
     }
 
-    viewerPerformance.stopRun();
-    viewerPerformance.startRun();
+    performanceStore.stopRun();
+    performanceStore.startRun();
 
     if (unfinished) {
       let d = unfinished;
@@ -115,8 +110,8 @@
   <Grid.Row>
     <Grid.Cell>
       <Viewer
-        bind:update={updateViewer}
-        perf={viewerPerformance}
+        result={graphResult}
+        perf={performanceStore}
         centerCamera={$AppSettings.centerCamera}
       />
     </Grid.Cell>
@@ -168,11 +163,8 @@
             hidden={!$AppSettings.showPerformancePanel}
             icon="i-tabler-brand-speedtest"
           >
-            {#if performanceData}
-              <PerformanceViewer
-                data={performanceData}
-                viewer={$viewerPerformance}
-              />
+            {#if $performanceStore}
+              <PerformanceViewer data={$performanceStore} />
             {/if}
           </Panel>
           <Panel
diff --git a/packages/types/src/components.ts b/packages/types/src/components.ts
index 8b2a324..bc02296 100644
--- a/packages/types/src/components.ts
+++ b/packages/types/src/components.ts
@@ -36,19 +36,26 @@ export interface RuntimeExecutor {
   execute: (graph: Graph, settings: Record<string, unknown>) => Promise<Int32Array>;
 }
 
-export interface RuntimeCache {
+export interface RuntimeCache<T = unknown> {
+
+  /**
+  * The maximum number of items that can be stored in the cache
+  * @remarks When the cache size exceeds this value, the oldest items should be removed
+  */
+  size: number;
+
   /**
    * Get the value for the given key
    * @param key - The key to get the value for
    * @returns The value for the given key, or undefined if no such value exists
    */
-  get: (key: string) => unknown | undefined;
+  get: (key: string) => T | undefined;
   /**
    * Set the value for the given key
    * @param key - The key to set the value for
    * @param value - The value to set
    */
-  set: (key: string, value: unknown) => void;
+  set: (key: string, value: T) => void;
   /**
    * Clear the cache
    */