overleaf/services/filestore/app/coffee/HealthCheckController.coffee

60 lines
1.5 KiB
CoffeeScript
Raw Normal View History

2016-12-14 12:14:15 -05:00
fs = require("fs-extra")
path = require("path")
async = require("async")
fileConverter = require("./FileConverter")
keyBuilder = require("./KeyBuilder")
fileController = require("./FileController")
logger = require('logger-sharelatex')
settings = require("settings-sharelatex")
streamBuffers = require("stream-buffers")
checkCanStoreFiles = (callback)->
req = {params:{}, query:{}, headers:{}}
res = {}
req.params.project_id = settings.health_check.project_id
req.params.file_id = settings.health_check.file_id
myWritableStreamBuffer = new streamBuffers.WritableStreamBuffer(initialSize: 100)
keyBuilder.userFileKey req, res, ->
fileController.getFile req, myWritableStreamBuffer
myWritableStreamBuffer.on "close", ->
if myWritableStreamBuffer.size() > 0
callback()
else
logger.err "no data in write stream buffer for health check"
callback()
checkFileConvert = (callback)->
imgPath = path.join(settings.path.uploadFolder, "/tiny.pdf")
async.waterfall [
(cb)->
fs.copy("./tiny.pdf", imgPath, cb)
2016-12-14 13:11:28 -05:00
(cb)-> fileConverter.thumbnail imgPath, cb
2016-12-14 12:14:15 -05:00
(resultPath, cb)-> fs.unlink resultPath, cb
2016-12-14 13:11:28 -05:00
(cb)-> fs.unlink imgPath, cb
2016-12-14 12:14:15 -05:00
], callback
2016-12-19 07:29:08 -05:00
isOk = true
2016-12-19 08:26:21 -05:00
q = async.queue (task, callback)->
task(callback)
2016-12-19 07:29:08 -05:00
2016-12-19 08:26:21 -05:00
runChecks = (callback)->
2016-12-19 07:29:08 -05:00
async.parallel [checkFileConvert, checkCanStoreFiles], (err)->
if err?
logger.err err:err, "Health check: error running"
isOk = false
else
isOk = true
2016-12-19 08:26:21 -05:00
callback()
2016-12-19 07:29:08 -05:00
2016-12-14 12:14:15 -05:00
module.exports =
check: (req, res)->
2016-12-19 07:29:08 -05:00
if isOk
res.send 200
else
res.send 500
2016-12-19 08:26:21 -05:00
q.push runChecks # run in background 1 at a time