index.js

import assert from 'node:assert'
import * as Sentry from '@sentry/node'
import { preprocess } from './lib/preprocess.js'
import { evaluate } from './lib/evaluate.js'
import { RoundData } from './lib/round.js'
import timers from 'node:timers/promises'

// Tweak this value to improve the chances of the data being available
const PREPROCESS_DELAY = 60_000

const EVALUATE_DELAY = PREPROCESS_DELAY + 60_000

export const startEvaluate = async ({
  ieContract,
  ieContractWithSigner,
  fetchMeasurements,
  fetchRoundDetails,
  recordTelemetry,
  createPgClient,
  logger
}) => {
  assert(typeof createPgClient === 'function', 'createPgClient must be a function')

  const rounds = {
    current: null,
    previous: null
  }
  const cidsSeen = []
  const roundsSeen = []

  const onMeasurementsAdded = async (cid, _roundIndex) => {
    const roundIndex = BigInt(_roundIndex)
    if (cidsSeen.includes(cid)) return
    cidsSeen.push(cid)
    if (cidsSeen.length > 1000) cidsSeen.shift()

    if (!rounds.current) {
      rounds.current = new RoundData(roundIndex)
    } else if (rounds.current.index !== roundIndex) {
      // This occassionally happens because of a race condition between onMeasurementsAdded
      // and onRoundStart event handlers.
      // See https://github.com/filecoin-station/spark-evaluate/issues/233
      const msg = 'Round index mismatch when processing MeasurementsAdded event'
      const details = {
        currentRoundIndex: rounds.current.index,
        eventRoundIndex: roundIndex,
        measurementsCid: cid
      }
      console.error(msg, details)
      Sentry.captureException(new Error(msg), { extra: details })
      return
    }

    console.log('Event: MeasurementsAdded', { roundIndex })
    console.log(`Sleeping for ${PREPROCESS_DELAY}ms before preprocessing to improve chances of the data being available`)
    await timers.setTimeout(PREPROCESS_DELAY)
    console.log(`Now preprocessing measurements for CID ${cid} in round ${roundIndex}`)

    // Preprocess
    try {
      await preprocess({
        round: rounds.current,
        cid,
        roundIndex,
        fetchMeasurements,
        recordTelemetry,
        logger
      })
    } catch (err) {
      console.error(err)
      Sentry.captureException(err, {
        extra: {
          roundIndex,
          measurementsCid: cid
        }
      })
    }
  }

  const onRoundStart = async (_roundIndex) => {
    const roundIndex = BigInt(_roundIndex)
    if (roundsSeen.includes(roundIndex)) return
    roundsSeen.push(roundIndex)
    if (roundsSeen.length > 1000) roundsSeen.shift()

    console.log('Event: RoundStart', { roundIndex })

    if (!rounds.current) {
      console.error('No current round data available, skipping evaluation')
      return
    }

    rounds.previous = rounds.current
    rounds.current = new RoundData(roundIndex)
    console.log('Advanced the current round to %s', roundIndex)

    // TODO: Fix this properly and implement a signalling mechanism allowing the "preprocess" step
    // to notify the "evaluate" when the preprocessing is done, so that we don't have to use a timer
    // here. See also https://github.com/filecoin-station/spark-evaluate/issues/64
    console.log(`Sleeping for ${EVALUATE_DELAY}ms before evaluating the round to let the preprocess step finish for the last batch of measurements`)
    await timers.setTimeout(EVALUATE_DELAY)
    console.log(`Now evaluating the round ${roundIndex}`)

    // Evaluate previous round
    evaluate({
      round: rounds.previous,
      roundIndex: roundIndex - 1n,
      ieContractWithSigner,
      fetchRoundDetails,
      recordTelemetry,
      createPgClient,
      logger
    }).catch(err => {
      console.error(err)
      Sentry.captureException(err, {
        extra: {
          roundIndex
        }
      })
    })
  }

  // Listen for events
  ieContract.on('MeasurementsAdded', (...args) => {
    onMeasurementsAdded(...args).catch(err => {
      console.error(err)
      Sentry.captureException(err)
    })
  })

  ieContract.on('RoundStart', (...args) => {
    onRoundStart(...args).catch(err => {
      console.error(err)
      Sentry.captureException(err)
    })
  })
}