]>
Commit | Line | Data |
---|---|---|
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 | } |