fix(cu): in findresult, check for exact eval by message id

permaweb · Jan 7, 2025 · 51012d6 · 51012d6
1 parent 859af1e
commit 51012d6
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 9 deletions.
diff --git a/servers/cu/src/domain/lib/chainEvaluation.js b/servers/cu/src/domain/lib/chainEvaluation.js
@@ -21,9 +21,7 @@ function loadLatestEvaluationWith ({ findEvaluation, findLatestProcessMemory, lo
 
     return findEvaluation({
       processId: ctx.id,
-      to: ctx.to,
-      ordinate: ctx.ordinate,
-      cron: ctx.cron
+      messageId: ctx.messageId
     })
       .map((evaluation) => {
         logger(

diff --git a/servers/cu/src/domain/lib/loadProcess.js b/servers/cu/src/domain/lib/loadProcess.js
@@ -65,9 +65,7 @@ function loadLatestEvaluationWith ({ findEvaluation, findLatestProcessMemory, sa
 
     return findEvaluation({
       processId: ctx.id,
-      to: ctx.to,
-      ordinate: ctx.ordinate,
-      cron: ctx.cron
+      messageId: ctx.messageId
     })
       .map((evaluation) => {
         logger(

diff --git a/servers/cu/src/effects/ao-evaluation.js b/servers/cu/src/effects/ao-evaluation.js
@@ -70,18 +70,27 @@ const fromEvaluationDoc = pipe(
   toEvaluation
 )
 
+// TODO: Change the query to use process Id, message Id, order by id
 export function findEvaluationWith ({ db }) {
-  function createQuery ({ processId, timestamp, ordinate, cron }) {
+  function createQuery ({ processId, messageId }) {
     return {
       sql: `
         SELECT
           id, "processId", "messageId", "deepHash", nonce, epoch, timestamp,
           ordinate, "blockHeight", cron, "evaluatedAt", output
         FROM ${EVALUATIONS_TABLE}
         WHERE
-          id = ?;
+          id > ? AND id <= ?
+          messageId = ?
+        ORDER BY
+          id ASC
+        LIMIT 1;
       `,
-      parameters: [createEvaluationId({ processId, timestamp, ordinate, cron })]
+      parameters: [
+        createEvaluationId({ processId, timestamp: '' }),
+        createEvaluationId({ processId, timestamp: COLLATION_SEQUENCE_MAX_CHAR }),
+        messageId
+      ]
     }
   }
 

diff --git a/servers/cu/src/effects/sqlite.js b/servers/cu/src/effects/sqlite.js
@@ -109,6 +109,12 @@ const createCheckpointFilesIndexes = async (db) => db.prepare(
     (processId, timestamp);`
 ).run()
 
+const createEvaluationsIndexes = async (db) => db.prepare(
+  `CREATE INDEX IF NOT EXISTS idx_${EVALUATIONS_TABLE}_id_messageId
+    ON ${EVALUATIONS_TABLE}
+    (id, messageId);`
+).run()
+
 let internalSqliteDb
 export async function createSqliteClient ({ url, bootstrap = false, walLimit = bytes.parse('100mb') }) {
   if (internalSqliteDb) return internalSqliteDb
@@ -138,6 +144,7 @@ export async function createSqliteClient ({ url, bootstrap = false, walLimit = b
       .then(() => createMessagesIndexes(db))
       .then(() => createCheckpointsIndexes(db))
       .then(() => createCheckpointFilesIndexes(db))
+      .then(() => createEvaluationsIndexes(db))
   }
 
   return {