overleaf/services/real-time/app/js/DeploymentManager.js
2021-03-22 10:46:25 +00:00

53 lines
1.7 KiB
JavaScript

const logger = require('logger-sharelatex')
const settings = require('settings-sharelatex')
const fs = require('fs')
// Monitor a status file (e.g. /etc/real_time_status) periodically and close the
// service if the file contents don't contain the matching deployment colour.
const FILE_CHECK_INTERVAL = 5000
const statusFile = settings.deploymentFile
const deploymentColour = settings.deploymentColour
function updateDeploymentStatus(fileContent) {
const closed = fileContent && fileContent.indexOf(deploymentColour) === -1
if (closed && !settings.serviceIsClosed) {
settings.serviceIsClosed = true
logger.warn({ fileContent }, 'closing service')
} else if (!closed && settings.serviceIsClosed) {
settings.serviceIsClosed = false
logger.warn({ fileContent }, 'opening service')
}
}
function pollStatusFile() {
fs.readFile(statusFile, { encoding: 'utf8' }, (err, fileContent) => {
if (err) {
logger.error(
{ file: statusFile, fsErr: err },
'error reading service status file'
)
return
}
updateDeploymentStatus(fileContent)
})
}
function checkStatusFileSync() {
// crash on start up if file does not exist
const content = fs.readFileSync(statusFile, { encoding: 'utf8' })
updateDeploymentStatus(content)
}
module.exports = {
initialise() {
if (statusFile && deploymentColour) {
logger.log(
{ statusFile, deploymentColour, interval: FILE_CHECK_INTERVAL },
'monitoring deployment status file'
)
checkStatusFileSync() // perform an initial synchronous check at start up
setInterval(pollStatusFile, FILE_CHECK_INTERVAL) // continue checking periodically
}
}
}