overleaf/services/project-history/app/js/FlushManager.js

// TODO: This file was created by bulk-decaffeinate.
// Fix any style issues and re-enable lint.
/*
 * decaffeinate suggestions:
 * DS101: Remove unnecessary use of Array.from
 * DS102: Remove unnecessary code created because of implicit returns
 * DS207: Consider shorter variations of null checks
 * Full docs: https://github.com/decaffeinate/decaffeinate/blob/master/docs/suggestions.md
 */
import async from 'async'
import logger from '@overleaf/logger'
import OError from '@overleaf/o-error'
import _ from 'lodash'
import * as RedisManager from './RedisManager.js'
import * as UpdatesProcessor from './UpdatesProcessor.js'
import * as ErrorRecorder from './ErrorRecorder.js'

export function flushIfOld(projectId, cutoffTime, callback) {
  if (callback == null) {
    callback = function () {}
  }
  return RedisManager.getFirstOpTimestamp(
    projectId,
    function (err, firstOpTimestamp) {
      if (err != null) {
        return callback(OError.tag(err))
      }
      // In the normal case, the flush marker will be set with the
      // timestamp of the oldest operation in the queue by docupdater.
      // If the marker is not set for any reason, we flush it anyway
      // for safety.
      if (!firstOpTimestamp || firstOpTimestamp < cutoffTime) {
        logger.debug(
          { projectId, firstOpTimestamp, cutoffTime },
          'flushing old project'
        )
        return UpdatesProcessor.processUpdatesForProject(projectId, callback)
      } else {
        return callback()
      }
    }
  )
}

export function flushOldOps(options, callback) {
  if (callback == null) {
    callback = function () {}
  }
  logger.debug({ options }, 'starting flush of old ops')
  // allow running flush in background for cron jobs
  if (options.background) {
    // return immediate response to client, then discard callback
    callback(null, { message: 'running flush in background' })
    callback = function () {}
  }
  return RedisManager.getProjectIdsWithHistoryOps(
    null,
    function (error, projectIds) {
      if (error != null) {
        return callback(OError.tag(error))
      }
      return ErrorRecorder.getFailedProjects(function (
        error,
        projectHistoryFailures
      ) {
        if (error != null) {
          return callback(OError.tag(error))
        }
        // exclude failed projects already in projectHistoryFailures
        const failedProjects = new Set()
        for (const entry of Array.from(projectHistoryFailures)) {
          failedProjects.add(entry.project_id)
        }
        // randomise order so we get different projects if there is a limit
        projectIds = _.shuffle(projectIds)
        const maxAge = options.maxAge || 6 * 3600 // default to 6 hours
        const cutoffTime = new Date(Date.now() - maxAge * 1000)
        const startTime = new Date()
        let count = 0
        const jobs = projectIds.map(
          projectId =>
            function (cb) {
              const timeTaken = new Date() - startTime
              count++
              if (
                (options != null ? options.timeout : undefined) &&
                timeTaken > options.timeout
              ) {
                // finish early due to timeout, return an error to bail out of the async iteration
                logger.debug('background retries timed out')
                return cb(new OError('retries timed out'))
              }
              if (
                (options != null ? options.limit : undefined) &&
                count > options.limit
              ) {
                // finish early due to reaching limit, return an error to bail out of the async iteration
                logger.debug({ count }, 'background retries hit limit')
                return cb(new OError('hit limit'))
              }
              if (failedProjects.has(projectId)) {
                // skip failed projects
                return setTimeout(cb, options.queueDelay || 100) // pause between flushes
              }
              return flushIfOld(projectId, cutoffTime, function (err) {
                if (err != null) {
                  logger.warn(
                    { projectId, flushErr: err },
                    'error flushing old project'
                  )
                }
                return setTimeout(cb, options.queueDelay || 100)
              })
            }
        ) // pause between flushes
        return async.series(async.reflectAll(jobs), function (error, results) {
          const success = []
          const failure = []
          results.forEach((result, i) => {
            if (
              result.error != null &&
              !['retries timed out', 'hit limit'].includes(
                result?.error?.message
              )
            ) {
              // ignore expected errors
              return failure.push(projectIds[i])
            } else {
              return success.push(projectIds[i])
            }
          })
          return callback(error, { success, failure, failedProjects })
        })
      })
    }
  )
}