@secondlayer/subgraphs 3.2.1 → 3.3.0

package/dist/src/runtime/catchup.js

@@ -18,12 +18,14 @@ class SubgraphContext {
   pgSchemaName;
   subgraphSchema;
   ops = [];
-  constructor(db, pgSchemaName, subgraphSchema, block, tx) {
+  byo;
+  constructor(db, pgSchemaName, subgraphSchema, block, tx, byo = false) {
     this.db = db;
     this.pgSchemaName = pgSchemaName;
     this.subgraphSchema = subgraphSchema;
     this.block = block;
     this._tx = tx;
+    this.byo = byo;
   }
   get tx() {
     return this._tx;
@@ -218,6 +220,15 @@ class SubgraphContext {
   }
   buildStatements(ops) {
     const statements = [];
+    if (this.byo) {
+      const insertTables = new Set;
+      for (const op of ops)
+        if (op.kind === "insert")
+          insertTables.add(op.table);
+      for (const table of insertTables) {
+        statements.push(`DELETE FROM "${this.pgSchemaName}"."${table}" WHERE "_block_height" = ${this.block.height}`);
+      }
+    }
     let currentBatch = null;
     let currentBatchKey = "";
     const flushInsertBatch = () => {
@@ -711,7 +722,19 @@ function matchPattern(value, pattern) {
   }
   return re.test(value);
 }
-function matchFilter(filter, transactions, eventsByTx) {
+var EMPTY_SET = new Set;
+function traitAllows(filter, contractId, traitContracts) {
+  const trait = filter.trait;
+  if (!trait)
+    return true;
+  if (!contractId)
+    return false;
+  return (traitContracts.get(trait) ?? EMPTY_SET).has(contractId);
+}
+function assetContract(assetId) {
+  return assetId?.split("::")[0];
+}
+function matchFilter(filter, transactions, eventsByTx, traitContracts) {
   const results = [];
   switch (filter.type) {
     case "stx_transfer":
@@ -775,6 +798,8 @@ function matchFilter(filter, transactions, eventsByTx) {
             if (!assetId || !matchPattern(assetId, filter.assetIdentifier))
               return false;
           }
+          if (!traitAllows(filter, assetContract(data.asset_identifier), traitContracts))
+            return false;
           if ("sender" in filter && filter.sender) {
             if (!matchPattern(data.sender, filter.sender))
               return false;
@@ -813,6 +838,8 @@ function matchFilter(filter, transactions, eventsByTx) {
             if (!assetId || !matchPattern(assetId, filter.assetIdentifier))
               return false;
           }
+          if (!traitAllows(filter, assetContract(data.asset_identifier), traitContracts))
+            return false;
           if ("sender" in filter && filter.sender) {
             if (!matchPattern(data.sender, filter.sender))
               return false;
@@ -845,6 +872,8 @@ function matchFilter(filter, transactions, eventsByTx) {
           if (!matchPattern(tx.sender, filter.caller))
             continue;
         }
+        if (!traitAllows(filter, tx.contract_id, traitContracts))
+          continue;
         const txEvents = eventsByTx.get(tx.tx_id) ?? [];
         results.push({ tx, events: txEvents });
       }
@@ -879,11 +908,13 @@ function matchFilter(filter, transactions, eventsByTx) {
             return false;
           if (data.topic !== "print")
             return false;
+          const printContractId = data.contract_identifier ?? data.contract_id;
           if (filter.contractId) {
-            const contractId = data.contract_identifier ?? data.contract_id;
-            if (!contractId || !matchPattern(contractId, filter.contractId))
+            if (!printContractId || !matchPattern(printContractId, filter.contractId))
               return false;
           }
+          if (!traitAllows(filter, printContractId, traitContracts))
+            return false;
           return true;
         });
         if (matched.length > 0) {
@@ -895,7 +926,7 @@ function matchFilter(filter, transactions, eventsByTx) {
   }
   return results;
 }
-function matchSources(sources, transactions, events) {
+function matchSources(sources, transactions, events, traitContracts = new Map) {
   const eventsByTx = new Map;
   for (const event of events) {
     const list = eventsByTx.get(event.tx_id) ?? [];
@@ -905,7 +936,7 @@ function matchSources(sources, transactions, events) {
   const seen = new Set;
   const results = [];
   for (const [sourceName, filter] of Object.entries(sources)) {
-    const matches = matchFilter(filter, transactions, eventsByTx);
+    const matches = matchFilter(filter, transactions, eventsByTx, traitContracts);
     for (const match of matches) {
       const dedupeKey = `${match.tx.tx_id}:${sourceName}`;
       if (!seen.has(dedupeKey)) {
@@ -922,8 +953,11 @@ import {
   getSourceDb,
   getTargetDb
 } from "@secondlayer/shared/db";
+import { resolveTraitContractIds } from "@secondlayer/shared/db/queries/contracts";
 import {
+  isByoSubgraph,
   recordSubgraphProcessed,
+  resolveSubgraphDb,
   updateSubgraphStatus
 } from "@secondlayer/shared/db/queries/subgraphs";
 import { logger as logger4 } from "@secondlayer/shared/logger";
@@ -1161,7 +1195,38 @@ async function refreshMatcher(db) {
 }
 
 // src/runtime/block-processor.ts
-var schemaNameCache = new Map;
+var routeCache = new Map;
+async function resolveRoute(subgraphName, targetDb) {
+  const cached = routeCache.get(subgraphName);
+  if (cached)
+    return cached;
+  const row = await targetDb.selectFrom("subgraphs").selectAll().where("name", "=", subgraphName).executeTakeFirst();
+  const byo = row ? isByoSubgraph(row) : false;
+  const route = {
+    schemaName: row?.schema_name ?? pgSchemaName(subgraphName),
+    dataDb: row && byo ? resolveSubgraphDb(row) : targetDb,
+    byo
+  };
+  routeCache.set(subgraphName, route);
+  return route;
+}
+function invalidateSubgraphRoute(subgraphName) {
+  routeCache.delete(subgraphName);
+}
+async function resolveTraitContracts(subgraph, blockHeight, db) {
+  const traits = new Set;
+  for (const source of Object.values(subgraph.sources)) {
+    const trait = source.trait;
+    if (trait)
+      traits.add(trait);
+  }
+  const resolved = new Map;
+  for (const trait of traits) {
+    const ids = await resolveTraitContractIds(db, trait, blockHeight);
+    resolved.set(trait, new Set(ids));
+  }
+  return resolved;
+}
 async function processBlock(subgraph, subgraphName, blockHeight, opts) {
   const sourceDb = getSourceDb();
   const targetDb = getTargetDb();
@@ -1201,7 +1266,8 @@ async function processBlock(subgraph, subgraphName, blockHeight, opts) {
     txs = await sourceDb.selectFrom("transactions").selectAll().where("block_height", "=", blockHeight).execute();
     evts = await sourceDb.selectFrom("events").selectAll().where("block_height", "=", blockHeight).execute();
   }
-  const matched = matchSources(subgraph.sources, txs, evts);
+  const traitContracts = await resolveTraitContracts(subgraph, blockHeight, targetDb);
+  const matched = matchSources(subgraph.sources, txs, evts, traitContracts);
   result.matched = matched.length;
   if (matched.length === 0) {
     if (!opts?.skipProgressUpdate) {
@@ -1209,12 +1275,8 @@ async function processBlock(subgraph, subgraphName, blockHeight, opts) {
     }
     return result;
   }
-  let schemaName = schemaNameCache.get(subgraphName);
-  if (!schemaName) {
-    const subgraphRecord = await targetDb.selectFrom("subgraphs").select("schema_name").where("name", "=", subgraphName).executeTakeFirst();
-    schemaName = subgraphRecord?.schema_name ?? pgSchemaName(subgraphName);
-    schemaNameCache.set(subgraphName, schemaName);
-  }
+  const route = await resolveRoute(subgraphName, targetDb);
+  const schemaName = route.schemaName;
   const blockMeta = {
     height: block.height,
     hash: block.hash,
@@ -1229,30 +1291,57 @@ async function processBlock(subgraph, subgraphName, blockHeight, opts) {
   };
   let handlerMs = 0;
   let flushMs = 0;
-  await targetDb.transaction().execute(async (tx) => {
-    const ctx = new SubgraphContext(tx, schemaName, subgraph.schema, blockMeta, initialTx);
-    const handlerStart = performance.now();
-    const runResult = await runHandlers(subgraph, matched, ctx);
-    handlerMs = performance.now() - handlerStart;
+  const applyProgress = async (tx, rr) => {
+    if (opts?.skipProgressUpdate)
+      return;
+    const status = rr.errors > 0 && rr.processed === 0 ? "error" : "active";
+    await updateSubgraphStatus(tx, subgraphName, status, blockHeight);
+    if (rr.processed > 0 || rr.errors > 0) {
+      const lastError = rr.errors > 0 ? `${rr.errors} error(s) at block ${blockHeight}` : undefined;
+      await recordSubgraphProcessed(tx, subgraphName, rr.processed, rr.errors, lastError);
+    }
+  };
+  if (route.byo) {
+    let runResult = { processed: 0, errors: 0 };
+    let manifest;
+    await route.dataDb.transaction().execute(async (tx) => {
+      const ctx = new SubgraphContext(tx, schemaName, subgraph.schema, blockMeta, initialTx, true);
+      const handlerStart = performance.now();
+      runResult = await runHandlers(subgraph, matched, ctx);
+      handlerMs = performance.now() - handlerStart;
+      if (ctx.pendingOps > 0) {
+        const flushStart = performance.now();
+        manifest = await ctx.flush();
+        flushMs = performance.now() - flushStart;
+      }
+    });
     result.processed = runResult.processed;
     result.errors = runResult.errors;
-    if (ctx.pendingOps > 0) {
-      const flushStart = performance.now();
-      const manifest = await ctx.flush();
-      if (manifest.count > 0) {
+    await targetDb.transaction().execute(async (tx) => {
+      if (manifest && manifest.count > 0) {
         await emitSubscriptionOutbox(tx, subgraphName, manifest, matcher, block.height);
       }
-      flushMs = performance.now() - flushStart;
-    }
-    if (!opts?.skipProgressUpdate) {
-      const status = runResult.errors > 0 && runResult.processed === 0 ? "error" : "active";
-      await updateSubgraphStatus(tx, subgraphName, status, blockHeight);
-    }
-    if (!opts?.skipProgressUpdate && (runResult.processed > 0 || runResult.errors > 0)) {
-      const lastError = runResult.errors > 0 ? `${runResult.errors} error(s) at block ${blockHeight}` : undefined;
-      await recordSubgraphProcessed(tx, subgraphName, runResult.processed, runResult.errors, lastError);
-    }
-  });
+      await applyProgress(tx, runResult);
+    });
+  } else {
+    await targetDb.transaction().execute(async (tx) => {
+      const ctx = new SubgraphContext(tx, schemaName, subgraph.schema, blockMeta, initialTx);
+      const handlerStart = performance.now();
+      const runResult = await runHandlers(subgraph, matched, ctx);
+      handlerMs = performance.now() - handlerStart;
+      result.processed = runResult.processed;
+      result.errors = runResult.errors;
+      if (ctx.pendingOps > 0) {
+        const flushStart = performance.now();
+        const manifest = await ctx.flush();
+        if (manifest.count > 0) {
+          await emitSubscriptionOutbox(tx, subgraphName, manifest, matcher, block.height);
+        }
+        flushMs = performance.now() - flushStart;
+      }
+      await applyProgress(tx, runResult);
+    });
+  }
   const totalMs = performance.now() - blockStart;
   result.timing = {
     totalMs: Math.round(totalMs),
@@ -1263,7 +1352,7 @@ async function processBlock(subgraph, subgraphName, blockHeight, opts) {
     try {
       const tables = Object.keys(subgraph.schema);
       for (const table of tables) {
-        const { rows } = await sql3.raw(`SELECT n_live_tup AS count FROM pg_stat_user_tables WHERE schemaname = '${schemaName}' AND relname = '${table}'`).execute(targetDb);
+        const { rows } = await sql3.raw(`SELECT n_live_tup AS count FROM pg_stat_user_tables WHERE schemaname = '${schemaName}' AND relname = '${table}'`).execute(route.dataDb);
         const count = Number(rows[0]?.count ?? 0);
         if (count >= 1e7) {
           logger4.warn("Subgraph table exceeds 10M rows (estimate)", {
@@ -1590,5 +1679,5 @@ export {
   catchUpSubgraph
 };
 
-//# debugId=BC0A838C52409B1064756E2164756E21
+//# debugId=3947491E2917E90B64756E2164756E21
 //# sourceMappingURL=catchup.js.map