@optave/codegraph 3.1.2 → 3.1.4

package/src/db/repository/edges.js

@@ -1,3 +1,21 @@
+import { cachedStmt } from './cached-stmt.js';
+
+// ─── Prepared-statement caches (one per db instance) ────────────────────
+const _findCalleesStmt = new WeakMap();
+const _findCallersStmt = new WeakMap();
+const _findDistinctCallersStmt = new WeakMap();
+const _findAllOutgoingStmt = new WeakMap();
+const _findAllIncomingStmt = new WeakMap();
+const _findCalleeNamesStmt = new WeakMap();
+const _findCallerNamesStmt = new WeakMap();
+const _findImportTargetsStmt = new WeakMap();
+const _findImportSourcesStmt = new WeakMap();
+const _findImportDependentsStmt = new WeakMap();
+const _findCrossFileCallTargetsStmt = new WeakMap();
+const _countCrossFileCallersStmt = new WeakMap();
+const _getClassAncestorsStmt = new WeakMap();
+const _findIntraFileCallEdgesStmt = new WeakMap();
+
 // ─── Call-edge queries ──────────────────────────────────────────────────
 
 /**
@@ -8,13 +26,13 @@
  * @returns {{ id: number, name: string, kind: string, file: string, line: number, end_line: number|null }[]}
  */
 export function findCallees(db, nodeId) {
-  return db
-    .prepare(
-      `SELECT DISTINCT n.id, n.name, n.kind, n.file, n.line, n.end_line
-       FROM edges e JOIN nodes n ON e.target_id = n.id
-       WHERE e.source_id = ? AND e.kind = 'calls'`,
-    )
-    .all(nodeId);
+  return cachedStmt(
+    _findCalleesStmt,
+    db,
+    `SELECT DISTINCT n.id, n.name, n.kind, n.file, n.line, n.end_line
+     FROM edges e JOIN nodes n ON e.target_id = n.id
+     WHERE e.source_id = ? AND e.kind = 'calls'`,
+  ).all(nodeId);
 }
 
 /**
@@ -24,13 +42,13 @@ export function findCallees(db, nodeId) {
  * @returns {{ id: number, name: string, kind: string, file: string, line: number }[]}
  */
 export function findCallers(db, nodeId) {
-  return db
-    .prepare(
-      `SELECT n.id, n.name, n.kind, n.file, n.line
-       FROM edges e JOIN nodes n ON e.source_id = n.id
-       WHERE e.target_id = ? AND e.kind = 'calls'`,
-    )
-    .all(nodeId);
+  return cachedStmt(
+    _findCallersStmt,
+    db,
+    `SELECT n.id, n.name, n.kind, n.file, n.line
+     FROM edges e JOIN nodes n ON e.source_id = n.id
+     WHERE e.target_id = ? AND e.kind = 'calls'`,
+  ).all(nodeId);
 }
 
 /**
@@ -40,13 +58,13 @@ export function findCallers(db, nodeId) {
  * @returns {{ id: number, name: string, kind: string, file: string, line: number }[]}
  */
 export function findDistinctCallers(db, nodeId) {
-  return db
-    .prepare(
-      `SELECT DISTINCT n.id, n.name, n.kind, n.file, n.line
-       FROM edges e JOIN nodes n ON e.source_id = n.id
-       WHERE e.target_id = ? AND e.kind = 'calls'`,
-    )
-    .all(nodeId);
+  return cachedStmt(
+    _findDistinctCallersStmt,
+    db,
+    `SELECT DISTINCT n.id, n.name, n.kind, n.file, n.line
+     FROM edges e JOIN nodes n ON e.source_id = n.id
+     WHERE e.target_id = ? AND e.kind = 'calls'`,
+  ).all(nodeId);
 }
 
 // ─── All-edge queries (no kind filter) ─────────────────────────────────
@@ -58,13 +76,13 @@ export function findDistinctCallers(db, nodeId) {
  * @returns {{ name: string, kind: string, file: string, line: number, edge_kind: string }[]}
  */
 export function findAllOutgoingEdges(db, nodeId) {
-  return db
-    .prepare(
-      `SELECT n.name, n.kind, n.file, n.line, e.kind AS edge_kind
-       FROM edges e JOIN nodes n ON e.target_id = n.id
-       WHERE e.source_id = ?`,
-    )
-    .all(nodeId);
+  return cachedStmt(
+    _findAllOutgoingStmt,
+    db,
+    `SELECT n.name, n.kind, n.file, n.line, e.kind AS edge_kind
+     FROM edges e JOIN nodes n ON e.target_id = n.id
+     WHERE e.source_id = ?`,
+  ).all(nodeId);
 }
 
 /**
@@ -74,13 +92,13 @@ export function findAllOutgoingEdges(db, nodeId) {
  * @returns {{ name: string, kind: string, file: string, line: number, edge_kind: string }[]}
  */
 export function findAllIncomingEdges(db, nodeId) {
-  return db
-    .prepare(
-      `SELECT n.name, n.kind, n.file, n.line, e.kind AS edge_kind
-       FROM edges e JOIN nodes n ON e.source_id = n.id
-       WHERE e.target_id = ?`,
-    )
-    .all(nodeId);
+  return cachedStmt(
+    _findAllIncomingStmt,
+    db,
+    `SELECT n.name, n.kind, n.file, n.line, e.kind AS edge_kind
+     FROM edges e JOIN nodes n ON e.source_id = n.id
+     WHERE e.target_id = ?`,
+  ).all(nodeId);
 }
 
 // ─── Name-only callee/caller lookups (for embedder) ────────────────────
@@ -92,13 +110,14 @@ export function findAllIncomingEdges(db, nodeId) {
  * @returns {string[]}
  */
 export function findCalleeNames(db, nodeId) {
-  return db
-    .prepare(
-      `SELECT DISTINCT n.name
-       FROM edges e JOIN nodes n ON e.target_id = n.id
-       WHERE e.source_id = ? AND e.kind = 'calls'
-       ORDER BY n.name`,
-    )
+  return cachedStmt(
+    _findCalleeNamesStmt,
+    db,
+    `SELECT DISTINCT n.name
+     FROM edges e JOIN nodes n ON e.target_id = n.id
+     WHERE e.source_id = ? AND e.kind = 'calls'
+     ORDER BY n.name`,
+  )
     .all(nodeId)
     .map((r) => r.name);
 }
@@ -110,13 +129,14 @@ export function findCalleeNames(db, nodeId) {
  * @returns {string[]}
  */
 export function findCallerNames(db, nodeId) {
-  return db
-    .prepare(
-      `SELECT DISTINCT n.name
-       FROM edges e JOIN nodes n ON e.source_id = n.id
-       WHERE e.target_id = ? AND e.kind = 'calls'
-       ORDER BY n.name`,
-    )
+  return cachedStmt(
+    _findCallerNamesStmt,
+    db,
+    `SELECT DISTINCT n.name
+     FROM edges e JOIN nodes n ON e.source_id = n.id
+     WHERE e.target_id = ? AND e.kind = 'calls'
+     ORDER BY n.name`,
+  )
     .all(nodeId)
     .map((r) => r.name);
 }
@@ -130,13 +150,13 @@ export function findCallerNames(db, nodeId) {
  * @returns {{ file: string, edge_kind: string }[]}
  */
 export function findImportTargets(db, nodeId) {
-  return db
-    .prepare(
-      `SELECT n.file, e.kind AS edge_kind
-       FROM edges e JOIN nodes n ON e.target_id = n.id
-       WHERE e.source_id = ? AND e.kind IN ('imports', 'imports-type')`,
-    )
-    .all(nodeId);
+  return cachedStmt(
+    _findImportTargetsStmt,
+    db,
+    `SELECT n.file, e.kind AS edge_kind
+     FROM edges e JOIN nodes n ON e.target_id = n.id
+     WHERE e.source_id = ? AND e.kind IN ('imports', 'imports-type')`,
+  ).all(nodeId);
 }
 
 /**
@@ -146,13 +166,13 @@ export function findImportTargets(db, nodeId) {
  * @returns {{ file: string, edge_kind: string }[]}
  */
 export function findImportSources(db, nodeId) {
-  return db
-    .prepare(
-      `SELECT n.file, e.kind AS edge_kind
-       FROM edges e JOIN nodes n ON e.source_id = n.id
-       WHERE e.target_id = ? AND e.kind IN ('imports', 'imports-type')`,
-    )
-    .all(nodeId);
+  return cachedStmt(
+    _findImportSourcesStmt,
+    db,
+    `SELECT n.file, e.kind AS edge_kind
+     FROM edges e JOIN nodes n ON e.source_id = n.id
+     WHERE e.target_id = ? AND e.kind IN ('imports', 'imports-type')`,
+  ).all(nodeId);
 }
 
 /**
@@ -163,12 +183,12 @@ export function findImportSources(db, nodeId) {
  * @returns {object[]}
  */
 export function findImportDependents(db, nodeId) {
-  return db
-    .prepare(
-      `SELECT n.* FROM edges e JOIN nodes n ON e.source_id = n.id
-       WHERE e.target_id = ? AND e.kind IN ('imports', 'imports-type')`,
-    )
-    .all(nodeId);
+  return cachedStmt(
+    _findImportDependentsStmt,
+    db,
+    `SELECT n.* FROM edges e JOIN nodes n ON e.source_id = n.id
+     WHERE e.target_id = ? AND e.kind IN ('imports', 'imports-type')`,
+  ).all(nodeId);
 }
 
 // ─── Cross-file and hierarchy queries ──────────────────────────────────
@@ -182,13 +202,14 @@ export function findImportDependents(db, nodeId) {
  */
 export function findCrossFileCallTargets(db, file) {
   return new Set(
-    db
-      .prepare(
-        `SELECT DISTINCT e.target_id FROM edges e
-         JOIN nodes caller ON e.source_id = caller.id
-         JOIN nodes target ON e.target_id = target.id
-         WHERE target.file = ? AND caller.file != ? AND e.kind = 'calls'`,
-      )
+    cachedStmt(
+      _findCrossFileCallTargetsStmt,
+      db,
+      `SELECT DISTINCT e.target_id FROM edges e
+       JOIN nodes caller ON e.source_id = caller.id
+       JOIN nodes target ON e.target_id = target.id
+       WHERE target.file = ? AND caller.file != ? AND e.kind = 'calls'`,
+    )
       .all(file, file)
       .map((r) => r.target_id),
   );
@@ -203,12 +224,12 @@ export function findCrossFileCallTargets(db, file) {
  * @returns {number}
  */
 export function countCrossFileCallers(db, nodeId, file) {
-  return db
-    .prepare(
-      `SELECT COUNT(*) AS cnt FROM edges e JOIN nodes n ON e.source_id = n.id
-       WHERE e.target_id = ? AND e.kind = 'calls' AND n.file != ?`,
-    )
-    .get(nodeId, file).cnt;
+  return cachedStmt(
+    _countCrossFileCallersStmt,
+    db,
+    `SELECT COUNT(*) AS cnt FROM edges e JOIN nodes n ON e.source_id = n.id
+     WHERE e.target_id = ? AND e.kind = 'calls' AND n.file != ?`,
+  ).get(nodeId, file).cnt;
 }
 
 /**
@@ -220,14 +241,15 @@ export function countCrossFileCallers(db, nodeId, file) {
 export function getClassHierarchy(db, classNodeId) {
   const ancestors = new Set();
   const queue = [classNodeId];
+  const stmt = cachedStmt(
+    _getClassAncestorsStmt,
+    db,
+    `SELECT n.id, n.name FROM edges e JOIN nodes n ON e.target_id = n.id
+     WHERE e.source_id = ? AND e.kind = 'extends'`,
+  );
   while (queue.length > 0) {
     const current = queue.shift();
-    const parents = db
-      .prepare(
-        `SELECT n.id, n.name FROM edges e JOIN nodes n ON e.target_id = n.id
-         WHERE e.source_id = ? AND e.kind = 'extends'`,
-      )
-      .all(current);
+    const parents = stmt.all(current);
     for (const p of parents) {
       if (!ancestors.has(p.id)) {
         ancestors.add(p.id);
@@ -246,14 +268,14 @@ export function getClassHierarchy(db, classNodeId) {
  * @returns {{ caller_name: string, callee_name: string }[]}
  */
 export function findIntraFileCallEdges(db, file) {
-  return db
-    .prepare(
-      `SELECT caller.name AS caller_name, callee.name AS callee_name
-       FROM edges e
-       JOIN nodes caller ON e.source_id = caller.id
-       JOIN nodes callee ON e.target_id = callee.id
-       WHERE caller.file = ? AND callee.file = ? AND e.kind = 'calls'
-       ORDER BY caller.line`,
-    )
-    .all(file, file);
+  return cachedStmt(
+    _findIntraFileCallEdgesStmt,
+    db,
+    `SELECT caller.name AS caller_name, callee.name AS callee_name
+     FROM edges e
+     JOIN nodes caller ON e.source_id = caller.id
+     JOIN nodes callee ON e.target_id = callee.id
+     WHERE caller.file = ? AND callee.file = ? AND e.kind = 'calls'
+     ORDER BY caller.line`,
+  ).all(file, file);
 }

package/src/db/repository/embeddings.js

@@ -1,3 +1,10 @@
+import { cachedStmt } from './cached-stmt.js';
+
+// ─── Statement caches (one prepared statement per db instance) ────────────
+const _hasEmbeddingsStmt = new WeakMap();
+const _getEmbeddingCountStmt = new WeakMap();
+const _getEmbeddingMetaStmt = new WeakMap();
+
 /**
  * Check whether the embeddings table has data.
  * @param {object} db
@@ -5,7 +12,7 @@
  */
 export function hasEmbeddings(db) {
   try {
-    return !!db.prepare('SELECT 1 FROM embeddings LIMIT 1').get();
+    return !!cachedStmt(_hasEmbeddingsStmt, db, 'SELECT 1 FROM embeddings LIMIT 1').get();
   } catch {
     return false;
   }
@@ -18,7 +25,7 @@ export function hasEmbeddings(db) {
  */
 export function getEmbeddingCount(db) {
   try {
-    return db.prepare('SELECT COUNT(*) AS c FROM embeddings').get().c;
+    return cachedStmt(_getEmbeddingCountStmt, db, 'SELECT COUNT(*) AS c FROM embeddings').get().c;
   } catch {
     return 0;
   }
@@ -32,7 +39,11 @@ export function getEmbeddingCount(db) {
  */
 export function getEmbeddingMeta(db, key) {
   try {
-    const row = db.prepare('SELECT value FROM embedding_meta WHERE key = ?').get(key);
+    const row = cachedStmt(
+      _getEmbeddingMetaStmt,
+      db,
+      'SELECT value FROM embedding_meta WHERE key = ?',
+    ).get(key);
     return row?.value;
   } catch {
     return undefined;

package/src/db/repository/graph-read.js

@@ -1,12 +1,25 @@
+import { CORE_SYMBOL_KINDS } from '../../kinds.js';
+import { cachedStmt } from './cached-stmt.js';
+
+// ─── Statement caches (one prepared statement per db instance) ────────────
+const _getCallableNodesStmt = new WeakMap();
+const _getCallEdgesStmt = new WeakMap();
+const _getFileNodesAllStmt = new WeakMap();
+const _getImportEdgesStmt = new WeakMap();
+
+const CALLABLE_KINDS_SQL = CORE_SYMBOL_KINDS.map((k) => `'${k}'`).join(',');
+
 /**
- * Get callable nodes (function/method/class) for community detection.
+ * Get callable nodes (all core symbol kinds) for graph construction.
  * @param {object} db
  * @returns {{ id: number, name: string, kind: string, file: string }[]}
  */
 export function getCallableNodes(db) {
-  return db
-    .prepare("SELECT id, name, kind, file FROM nodes WHERE kind IN ('function','method','class')")
-    .all();
+  return cachedStmt(
+    _getCallableNodesStmt,
+    db,
+    `SELECT id, name, kind, file FROM nodes WHERE kind IN (${CALLABLE_KINDS_SQL})`,
+  ).all();
 }
 
 /**
@@ -15,7 +28,11 @@ export function getCallableNodes(db) {
  * @returns {{ source_id: number, target_id: number }[]}
  */
 export function getCallEdges(db) {
-  return db.prepare("SELECT source_id, target_id FROM edges WHERE kind = 'calls'").all();
+  return cachedStmt(
+    _getCallEdgesStmt,
+    db,
+    "SELECT source_id, target_id FROM edges WHERE kind = 'calls'",
+  ).all();
 }
 
 /**
@@ -24,7 +41,11 @@ export function getCallEdges(db) {
  * @returns {{ id: number, name: string, file: string }[]}
  */
 export function getFileNodesAll(db) {
-  return db.prepare("SELECT id, name, file FROM nodes WHERE kind = 'file'").all();
+  return cachedStmt(
+    _getFileNodesAllStmt,
+    db,
+    "SELECT id, name, file FROM nodes WHERE kind = 'file'",
+  ).all();
 }
 
 /**
@@ -33,7 +54,9 @@ export function getFileNodesAll(db) {
  * @returns {{ source_id: number, target_id: number }[]}
  */
 export function getImportEdges(db) {
-  return db
-    .prepare("SELECT source_id, target_id FROM edges WHERE kind IN ('imports','imports-type')")
-    .all();
+  return cachedStmt(
+    _getImportEdgesStmt,
+    db,
+    "SELECT source_id, target_id FROM edges WHERE kind IN ('imports','imports-type')",
+  ).all();
 }