Merge pull request #1180 from permaweb/jfrain99/ignore-local-checkpoints-cu

feat(cu): ignore local checkpoints in env

Author: jfrain99

servers/cu/src/bootstrap.js

@@ -406,6 +406,7 @@ export const createApis = async (ctx) => {
         CHECKPONT_VALIDATION_STEPS: ctx.CHECKPONT_VALIDATION_STEPS,
         CHECKPONT_VALIDATION_THRESH: ctx.CHECKPONT_VALIDATION_THRESH,
         CHECKPONT_VALIDATION_RETRIES: ctx.CHECKPONT_VALIDATION_RETRIES,
+        IGNORE_LOCAL_CHECKPOINTS: ctx.IGNORE_LOCAL_CHECKPOINTS,
         logger
       }),
       saveLatestProcessMemory: AoProcessClient.saveLatestProcessMemoryWith({

servers/cu/src/config.js

@@ -177,7 +177,8 @@ const CONFIG_ENVS = {
     ALLOW_OWNERS: process.env.ALLOW_OWNERS || [],
     CHECKPONT_VALIDATION_STEPS: process.env.CHECKPONT_VALIDATION_STEPS || 10,
     CHECKPONT_VALIDATION_THRESH: process.env.CHECKPONT_VALIDATION_THRESH || 0.75,
-    CHECKPONT_VALIDATION_RETRIES: process.env.CHECKPONT_VALIDATION_RETRIES || 5
+    CHECKPONT_VALIDATION_RETRIES: process.env.CHECKPONT_VALIDATION_RETRIES || 5,
+    IGNORE_LOCAL_CHECKPOINTS: process.env.IGNORE_LOCAL_CHECKPOINTS === 'true'
   },
   production: {
     MODE,
@@ -234,7 +235,8 @@ const CONFIG_ENVS = {
     ALLOW_OWNERS: process.env.ALLOW_OWNERS || [],
     CHECKPONT_VALIDATION_STEPS: process.env.CHECKPONT_VALIDATION_STEPS || 10,
     CHECKPONT_VALIDATION_THRESH: process.env.CHECKPONT_VALIDATION_THRESH || 0.75,
-    CHECKPONT_VALIDATION_RETRIES: process.env.CHECKPONT_VALIDATION_RETRIES || 5
+    CHECKPONT_VALIDATION_RETRIES: process.env.CHECKPONT_VALIDATION_RETRIES || 5,
+    IGNORE_LOCAL_CHECKPOINTS: process.env.IGNORE_LOCAL_CHECKPOINTS === 'true'
   }
 }
 

servers/cu/src/domain/model.js

@@ -241,7 +241,12 @@ export const domainConfigSchema = z.object({
    * How many times to jump backwards when failing to validate
    * a state from a checkpoint
    */
-  CHECKPONT_VALIDATION_RETRIES: positiveIntSchema
+  CHECKPONT_VALIDATION_RETRIES: positiveIntSchema,
+
+  /**
+   * Whether to ignore local checkpoints. Set to 'true' to ignore local checkpoints (file and record checkpoints).
+   */
+  IGNORE_LOCAL_CHECKPOINTS: z.preprocess((val) => !!val, z.boolean())
 })
 
 export const bufferSchema = z.any().refine(buffer => {

servers/cu/src/effects/ao-process.js

@@ -804,7 +804,8 @@ export function findLatestProcessMemoryWith ({
   hashWasmMemory,
   CHECKPONT_VALIDATION_STEPS,
   CHECKPONT_VALIDATION_THRESH,
-  CHECKPONT_VALIDATION_RETRIES
+  CHECKPONT_VALIDATION_RETRIES,
+  IGNORE_LOCAL_CHECKPOINTS
 }) {
   const logger = _logger.child('ao-process:findLatestProcessMemory')
   readProcessMemoryFile = fromPromise(readProcessMemoryFile)
@@ -1129,6 +1130,12 @@ export function findLatestProcessMemoryWith ({
    */
   function maybeFile (args) {
     const { processId, omitMemory } = args
+    
+    if (IGNORE_LOCAL_CHECKPOINTS) {
+      logger('Local Checkpoints are ignored for this CU. Not attempting to query file checkpoints...', processId)
+      return Rejected(args)
+    }
+
     /**
      * Attempt to find the latest checkpoint in a file
      */
@@ -1183,6 +1190,10 @@ export function findLatestProcessMemoryWith ({
 
   function maybeRecord (args) {
     const { processId, omitMemory } = args
+    if (IGNORE_LOCAL_CHECKPOINTS) {
+      logger('Local Checkpoints are ignored for this CU. Not attempting to query record checkpoints...', processId)
+      return Rejected(args)
+    }
 
     if (PROCESS_IGNORE_ARWEAVE_CHECKPOINTS.includes(processId)) {
       logger('Arweave Checkpoints are ignored for process "%s". Not attempting to query file checkpoints...', processId)