server/lib/runners/runner.ts


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49

import express from 'express'
import { retryTransactionWrapper } from '@server/helpers/database-utils'
import { logger, loggerTagsFactory } from '@server/helpers/logger'
import { sequelizeTypescript } from '@server/initializers/database'
import { MRunner, MRunnerJob } from '@server/types/models/runners'
import { RUNNER_JOBS } from '@server/initializers/constants'
import { RunnerJobState } from '@shared/models'

const lTags = loggerTagsFactory('runner')

const updatingRunner = new Set<number>()

function updateLastRunnerContact (req: express.Request, runner: MRunner) {
  const now = new Date()

  // Don't update last runner contact too often
  if (now.getTime() - runner.lastContact.getTime() < RUNNER_JOBS.LAST_CONTACT_UPDATE_INTERVAL) return
  if (updatingRunner.has(runner.id)) return

  updatingRunner.add(runner.id)

  runner.lastContact = now
  runner.ip = req.ip

  logger.debug('Updating last runner contact for %s', runner.name, lTags(runner.name))

  retryTransactionWrapper(() => {
    return sequelizeTypescript.transaction(async transaction => {
      return runner.save({ transaction })
    })
  })
  .catch(err => logger.error('Cannot update last runner contact for %s', runner.name, { err, ...lTags(runner.name) }))
  .finally(() => updatingRunner.delete(runner.id))
}

function runnerJobCanBeCancelled (runnerJob: MRunnerJob) {
  const allowedStates = new Set<RunnerJobState>([
    RunnerJobState.PENDING,
    RunnerJobState.PROCESSING,
    RunnerJobState.WAITING_FOR_PARENT_JOB
  ])

  return allowedStates.has(runnerJob.state)
}

export {
  updateLastRunnerContact,
  runnerJobCanBeCancelled
}