server/lib/runners/runner.ts

   1 import express from 'express'
   2 import { retryTransactionWrapper } from '@server/helpers/database-utils'
   3 import { logger, loggerTagsFactory } from '@server/helpers/logger'
   4 import { sequelizeTypescript } from '@server/initializers/database'
   5 import { MRunner } from '@server/types/models/runners'
   6 import { RUNNER_JOBS } from '@server/initializers/constants'
   7
   8 const lTags = loggerTagsFactory('runner')
   9
  10 const updatingRunner = new Set<number>()
  11
  12 function updateLastRunnerContact (req: express.Request, runner: MRunner) {
  13   const now = new Date()
  14
  15   // Don't update last runner contact too often
  16   if (now.getTime() - runner.lastContact.getTime() < RUNNER_JOBS.LAST_CONTACT_UPDATE_INTERVAL) return
  17   if (updatingRunner.has(runner.id)) return
  18
  19   updatingRunner.add(runner.id)
  20
  21   runner.lastContact = now
  22   runner.ip = req.ip
  23
  24   logger.debug('Updating last runner contact for %s', runner.name, lTags(runner.name))
  25
  26   retryTransactionWrapper(() => {
  27     return sequelizeTypescript.transaction(async transaction => {
  28       return runner.save({ transaction })
  29     })
  30   })
  31   .catch(err => logger.error('Cannot update last runner contact for %s', runner.name, { err, ...lTags(runner.name) }))
  32   .finally(() => updatingRunner.delete(runner.id))
  33 }
  34
  35 export {
  36   updateLastRunnerContact
  37 }