mirror of
https://github.com/overleaf/overleaf.git
synced 2024-12-30 21:03:46 +00:00
762 lines
24 KiB
JavaScript
762 lines
24 KiB
JavaScript
/* eslint-disable
|
|
camelcase,
|
|
handle-callback-err,
|
|
*/
|
|
// TODO: This file was created by bulk-decaffeinate.
|
|
// Fix any style issues and re-enable lint.
|
|
/*
|
|
* decaffeinate suggestions:
|
|
* DS101: Remove unnecessary use of Array.from
|
|
* DS102: Remove unnecessary code created because of implicit returns
|
|
* DS103: Rewrite code to no longer use __guard__
|
|
* DS201: Simplify complex destructure assignments
|
|
* DS207: Consider shorter variations of null checks
|
|
* Full docs: https://github.com/decaffeinate/decaffeinate/blob/master/docs/suggestions.md
|
|
*/
|
|
let RedisManager
|
|
const Settings = require('settings-sharelatex')
|
|
const rclient = require('redis-sharelatex').createClient(
|
|
Settings.redis.documentupdater
|
|
)
|
|
const logger = require('logger-sharelatex')
|
|
const metrics = require('./Metrics')
|
|
const Errors = require('./Errors')
|
|
const crypto = require('crypto')
|
|
const async = require('async')
|
|
const ProjectHistoryRedisManager = require('./ProjectHistoryRedisManager')
|
|
|
|
// Sometimes Redis calls take an unexpectedly long time. We have to be
|
|
// quick with Redis calls because we're holding a lock that expires
|
|
// after 30 seconds. We can't let any errors in the rest of the stack
|
|
// hold us up, and need to bail out quickly if there is a problem.
|
|
const MAX_REDIS_REQUEST_LENGTH = 5000 // 5 seconds
|
|
|
|
// Make times easy to read
|
|
const minutes = 60 // seconds for Redis expire
|
|
|
|
const logHashErrors =
|
|
Settings.documentupdater != null
|
|
? Settings.documentupdater.logHashErrors
|
|
: undefined
|
|
const logHashReadErrors = logHashErrors != null ? logHashErrors.read : undefined
|
|
|
|
const MEGABYTES = 1024 * 1024
|
|
const MAX_RANGES_SIZE = 3 * MEGABYTES
|
|
|
|
const keys = Settings.redis.documentupdater.key_schema
|
|
const historyKeys = Settings.redis.history.key_schema // note: this is track changes, not project-history
|
|
|
|
module.exports = RedisManager = {
|
|
rclient,
|
|
|
|
putDocInMemory(
|
|
project_id,
|
|
doc_id,
|
|
docLines,
|
|
version,
|
|
ranges,
|
|
pathname,
|
|
projectHistoryId,
|
|
_callback
|
|
) {
|
|
const timer = new metrics.Timer('redis.put-doc')
|
|
const callback = function (error) {
|
|
timer.done()
|
|
return _callback(error)
|
|
}
|
|
docLines = JSON.stringify(docLines)
|
|
if (docLines.indexOf('\u0000') !== -1) {
|
|
const error = new Error('null bytes found in doc lines')
|
|
// this check was added to catch memory corruption in JSON.stringify.
|
|
// It sometimes returned null bytes at the end of the string.
|
|
logger.error({ err: error, doc_id, docLines }, error.message)
|
|
return callback(error)
|
|
}
|
|
const docHash = RedisManager._computeHash(docLines)
|
|
// record bytes sent to redis
|
|
metrics.summary('redis.docLines', docLines.length, { status: 'set' })
|
|
logger.log(
|
|
{ project_id, doc_id, version, docHash, pathname, projectHistoryId },
|
|
'putting doc in redis'
|
|
)
|
|
return RedisManager._serializeRanges(ranges, function (error, ranges) {
|
|
if (error != null) {
|
|
logger.error({ err: error, doc_id, project_id }, error.message)
|
|
return callback(error)
|
|
}
|
|
const multi = rclient.multi()
|
|
multi.set(keys.docLines({ doc_id }), docLines)
|
|
multi.set(keys.projectKey({ doc_id }), project_id)
|
|
multi.set(keys.docVersion({ doc_id }), version)
|
|
multi.set(keys.docHash({ doc_id }), docHash)
|
|
if (ranges != null) {
|
|
multi.set(keys.ranges({ doc_id }), ranges)
|
|
} else {
|
|
multi.del(keys.ranges({ doc_id }))
|
|
}
|
|
multi.set(keys.pathname({ doc_id }), pathname)
|
|
multi.set(keys.projectHistoryId({ doc_id }), projectHistoryId)
|
|
return multi.exec(function (error, result) {
|
|
if (error != null) {
|
|
return callback(error)
|
|
}
|
|
// update docsInProject set
|
|
return rclient.sadd(
|
|
keys.docsInProject({ project_id }),
|
|
doc_id,
|
|
callback
|
|
)
|
|
})
|
|
})
|
|
},
|
|
|
|
removeDocFromMemory(project_id, doc_id, _callback) {
|
|
logger.log({ project_id, doc_id }, 'removing doc from redis')
|
|
const callback = function (err) {
|
|
if (err != null) {
|
|
logger.err({ project_id, doc_id, err }, 'error removing doc from redis')
|
|
return _callback(err)
|
|
} else {
|
|
logger.log({ project_id, doc_id }, 'removed doc from redis')
|
|
return _callback()
|
|
}
|
|
}
|
|
|
|
let multi = rclient.multi()
|
|
multi.strlen(keys.docLines({ doc_id }))
|
|
multi.del(keys.docLines({ doc_id }))
|
|
multi.del(keys.projectKey({ doc_id }))
|
|
multi.del(keys.docVersion({ doc_id }))
|
|
multi.del(keys.docHash({ doc_id }))
|
|
multi.del(keys.ranges({ doc_id }))
|
|
multi.del(keys.pathname({ doc_id }))
|
|
multi.del(keys.projectHistoryId({ doc_id }))
|
|
multi.del(keys.projectHistoryType({ doc_id }))
|
|
multi.del(keys.unflushedTime({ doc_id }))
|
|
multi.del(keys.lastUpdatedAt({ doc_id }))
|
|
multi.del(keys.lastUpdatedBy({ doc_id }))
|
|
return multi.exec(function (error, response) {
|
|
if (error != null) {
|
|
return callback(error)
|
|
}
|
|
const length = response != null ? response[0] : undefined
|
|
if (length > 0) {
|
|
// record bytes freed in redis
|
|
metrics.summary('redis.docLines', length, { status: 'del' })
|
|
}
|
|
multi = rclient.multi()
|
|
multi.srem(keys.docsInProject({ project_id }), doc_id)
|
|
multi.del(keys.projectState({ project_id }))
|
|
return multi.exec(callback)
|
|
})
|
|
},
|
|
|
|
checkOrSetProjectState(project_id, newState, callback) {
|
|
if (callback == null) {
|
|
callback = function (error, stateChanged) {}
|
|
}
|
|
const multi = rclient.multi()
|
|
multi.getset(keys.projectState({ project_id }), newState)
|
|
multi.expire(keys.projectState({ project_id }), 30 * minutes)
|
|
return multi.exec(function (error, response) {
|
|
if (error != null) {
|
|
return callback(error)
|
|
}
|
|
logger.log(
|
|
{ project_id, newState, oldState: response[0] },
|
|
'checking project state'
|
|
)
|
|
return callback(null, response[0] !== newState)
|
|
})
|
|
},
|
|
|
|
clearProjectState(project_id, callback) {
|
|
if (callback == null) {
|
|
callback = function (error) {}
|
|
}
|
|
return rclient.del(keys.projectState({ project_id }), callback)
|
|
},
|
|
|
|
getDoc(project_id, doc_id, callback) {
|
|
if (callback == null) {
|
|
callback = function (
|
|
error,
|
|
lines,
|
|
version,
|
|
ranges,
|
|
pathname,
|
|
projectHistoryId,
|
|
unflushedTime
|
|
) {}
|
|
}
|
|
const timer = new metrics.Timer('redis.get-doc')
|
|
const multi = rclient.multi()
|
|
multi.get(keys.docLines({ doc_id }))
|
|
multi.get(keys.docVersion({ doc_id }))
|
|
multi.get(keys.docHash({ doc_id }))
|
|
multi.get(keys.projectKey({ doc_id }))
|
|
multi.get(keys.ranges({ doc_id }))
|
|
multi.get(keys.pathname({ doc_id }))
|
|
multi.get(keys.projectHistoryId({ doc_id }))
|
|
multi.get(keys.unflushedTime({ doc_id }))
|
|
multi.get(keys.lastUpdatedAt({ doc_id }))
|
|
multi.get(keys.lastUpdatedBy({ doc_id }))
|
|
return multi.exec(function (error, ...rest) {
|
|
let [
|
|
docLines,
|
|
version,
|
|
storedHash,
|
|
doc_project_id,
|
|
ranges,
|
|
pathname,
|
|
projectHistoryId,
|
|
unflushedTime,
|
|
lastUpdatedAt,
|
|
lastUpdatedBy
|
|
] = Array.from(rest[0])
|
|
const timeSpan = timer.done()
|
|
if (error != null) {
|
|
return callback(error)
|
|
}
|
|
// check if request took too long and bail out. only do this for
|
|
// get, because it is the first call in each update, so if this
|
|
// passes we'll assume others have a reasonable chance to succeed.
|
|
if (timeSpan > MAX_REDIS_REQUEST_LENGTH) {
|
|
error = new Error('redis getDoc exceeded timeout')
|
|
return callback(error)
|
|
}
|
|
// record bytes loaded from redis
|
|
if (docLines != null) {
|
|
metrics.summary('redis.docLines', docLines.length, { status: 'get' })
|
|
}
|
|
// check sha1 hash value if present
|
|
if (docLines != null && storedHash != null) {
|
|
const computedHash = RedisManager._computeHash(docLines)
|
|
if (logHashReadErrors && computedHash !== storedHash) {
|
|
logger.error(
|
|
{
|
|
project_id,
|
|
doc_id,
|
|
doc_project_id,
|
|
computedHash,
|
|
storedHash,
|
|
docLines
|
|
},
|
|
'hash mismatch on retrieved document'
|
|
)
|
|
}
|
|
}
|
|
|
|
try {
|
|
docLines = JSON.parse(docLines)
|
|
ranges = RedisManager._deserializeRanges(ranges)
|
|
} catch (e) {
|
|
return callback(e)
|
|
}
|
|
|
|
version = parseInt(version || 0, 10)
|
|
// check doc is in requested project
|
|
if (doc_project_id != null && doc_project_id !== project_id) {
|
|
logger.error(
|
|
{ project_id, doc_id, doc_project_id },
|
|
'doc not in project'
|
|
)
|
|
return callback(new Errors.NotFoundError('document not found'))
|
|
}
|
|
|
|
if (projectHistoryId != null) {
|
|
projectHistoryId = parseInt(projectHistoryId)
|
|
}
|
|
|
|
// doc is not in redis, bail out
|
|
if (docLines == null) {
|
|
return callback(
|
|
null,
|
|
docLines,
|
|
version,
|
|
ranges,
|
|
pathname,
|
|
projectHistoryId,
|
|
unflushedTime,
|
|
lastUpdatedAt,
|
|
lastUpdatedBy
|
|
)
|
|
}
|
|
|
|
// doc should be in project set, check if missing (workaround for missing docs from putDoc)
|
|
return rclient.sadd(keys.docsInProject({ project_id }), doc_id, function (
|
|
error,
|
|
result
|
|
) {
|
|
if (error != null) {
|
|
return callback(error)
|
|
}
|
|
if (result !== 0) {
|
|
// doc should already be in set
|
|
logger.error(
|
|
{ project_id, doc_id, doc_project_id },
|
|
'doc missing from docsInProject set'
|
|
)
|
|
}
|
|
return callback(
|
|
null,
|
|
docLines,
|
|
version,
|
|
ranges,
|
|
pathname,
|
|
projectHistoryId,
|
|
unflushedTime,
|
|
lastUpdatedAt,
|
|
lastUpdatedBy
|
|
)
|
|
})
|
|
})
|
|
},
|
|
|
|
getDocVersion(doc_id, callback) {
|
|
if (callback == null) {
|
|
callback = function (error, version, projectHistoryType) {}
|
|
}
|
|
return rclient.mget(
|
|
keys.docVersion({ doc_id }),
|
|
keys.projectHistoryType({ doc_id }),
|
|
function (error, result) {
|
|
if (error != null) {
|
|
return callback(error)
|
|
}
|
|
let [version, projectHistoryType] = Array.from(result || [])
|
|
version = parseInt(version, 10)
|
|
return callback(null, version, projectHistoryType)
|
|
}
|
|
)
|
|
},
|
|
|
|
getDocLines(doc_id, callback) {
|
|
if (callback == null) {
|
|
callback = function (error, version) {}
|
|
}
|
|
return rclient.get(keys.docLines({ doc_id }), function (error, docLines) {
|
|
if (error != null) {
|
|
return callback(error)
|
|
}
|
|
return callback(null, docLines)
|
|
})
|
|
},
|
|
|
|
getPreviousDocOps(doc_id, start, end, callback) {
|
|
if (callback == null) {
|
|
callback = function (error, jsonOps) {}
|
|
}
|
|
const timer = new metrics.Timer('redis.get-prev-docops')
|
|
return rclient.llen(keys.docOps({ doc_id }), function (error, length) {
|
|
if (error != null) {
|
|
return callback(error)
|
|
}
|
|
return rclient.get(keys.docVersion({ doc_id }), function (
|
|
error,
|
|
version
|
|
) {
|
|
if (error != null) {
|
|
return callback(error)
|
|
}
|
|
version = parseInt(version, 10)
|
|
const first_version_in_redis = version - length
|
|
|
|
if (start < first_version_in_redis || end > version) {
|
|
error = new Errors.OpRangeNotAvailableError(
|
|
'doc ops range is not loaded in redis'
|
|
)
|
|
logger.warn(
|
|
{ err: error, doc_id, length, version, start, end },
|
|
'doc ops range is not loaded in redis'
|
|
)
|
|
return callback(error)
|
|
}
|
|
|
|
start = start - first_version_in_redis
|
|
if (end > -1) {
|
|
end = end - first_version_in_redis
|
|
}
|
|
|
|
if (isNaN(start) || isNaN(end)) {
|
|
error = new Error('inconsistent version or lengths')
|
|
logger.error(
|
|
{ err: error, doc_id, length, version, start, end },
|
|
'inconsistent version or length'
|
|
)
|
|
return callback(error)
|
|
}
|
|
|
|
return rclient.lrange(keys.docOps({ doc_id }), start, end, function (
|
|
error,
|
|
jsonOps
|
|
) {
|
|
let ops
|
|
if (error != null) {
|
|
return callback(error)
|
|
}
|
|
try {
|
|
ops = jsonOps.map((jsonOp) => JSON.parse(jsonOp))
|
|
} catch (e) {
|
|
return callback(e)
|
|
}
|
|
const timeSpan = timer.done()
|
|
if (timeSpan > MAX_REDIS_REQUEST_LENGTH) {
|
|
error = new Error('redis getPreviousDocOps exceeded timeout')
|
|
return callback(error)
|
|
}
|
|
return callback(null, ops)
|
|
})
|
|
})
|
|
})
|
|
},
|
|
|
|
getHistoryType(doc_id, callback) {
|
|
if (callback == null) {
|
|
callback = function (error, projectHistoryType) {}
|
|
}
|
|
return rclient.get(keys.projectHistoryType({ doc_id }), function (
|
|
error,
|
|
projectHistoryType
|
|
) {
|
|
if (error != null) {
|
|
return callback(error)
|
|
}
|
|
return callback(null, projectHistoryType)
|
|
})
|
|
},
|
|
|
|
setHistoryType(doc_id, projectHistoryType, callback) {
|
|
if (callback == null) {
|
|
callback = function (error) {}
|
|
}
|
|
return rclient.set(
|
|
keys.projectHistoryType({ doc_id }),
|
|
projectHistoryType,
|
|
callback
|
|
)
|
|
},
|
|
|
|
DOC_OPS_TTL: 60 * minutes,
|
|
DOC_OPS_MAX_LENGTH: 100,
|
|
updateDocument(
|
|
project_id,
|
|
doc_id,
|
|
docLines,
|
|
newVersion,
|
|
appliedOps,
|
|
ranges,
|
|
updateMeta,
|
|
callback
|
|
) {
|
|
if (appliedOps == null) {
|
|
appliedOps = []
|
|
}
|
|
if (callback == null) {
|
|
callback = function (error) {}
|
|
}
|
|
return RedisManager.getDocVersion(doc_id, function (
|
|
error,
|
|
currentVersion,
|
|
projectHistoryType
|
|
) {
|
|
if (error != null) {
|
|
return callback(error)
|
|
}
|
|
if (currentVersion + appliedOps.length !== newVersion) {
|
|
error = new Error(`Version mismatch. '${doc_id}' is corrupted.`)
|
|
logger.error(
|
|
{
|
|
err: error,
|
|
doc_id,
|
|
currentVersion,
|
|
newVersion,
|
|
opsLength: appliedOps.length
|
|
},
|
|
'version mismatch'
|
|
)
|
|
return callback(error)
|
|
}
|
|
|
|
const jsonOps = appliedOps.map((op) => JSON.stringify(op))
|
|
for (const op of Array.from(jsonOps)) {
|
|
if (op.indexOf('\u0000') !== -1) {
|
|
error = new Error('null bytes found in jsonOps')
|
|
// this check was added to catch memory corruption in JSON.stringify
|
|
logger.error({ err: error, doc_id, jsonOps }, error.message)
|
|
return callback(error)
|
|
}
|
|
}
|
|
|
|
const newDocLines = JSON.stringify(docLines)
|
|
if (newDocLines.indexOf('\u0000') !== -1) {
|
|
error = new Error('null bytes found in doc lines')
|
|
// this check was added to catch memory corruption in JSON.stringify
|
|
logger.error({ err: error, doc_id, newDocLines }, error.message)
|
|
return callback(error)
|
|
}
|
|
const newHash = RedisManager._computeHash(newDocLines)
|
|
|
|
const opVersions = appliedOps.map((op) => (op != null ? op.v : undefined))
|
|
logger.log(
|
|
{ doc_id, version: newVersion, hash: newHash, op_versions: opVersions },
|
|
'updating doc in redis'
|
|
)
|
|
// record bytes sent to redis in update
|
|
metrics.summary('redis.docLines', newDocLines.length, {
|
|
status: 'update'
|
|
})
|
|
return RedisManager._serializeRanges(ranges, function (error, ranges) {
|
|
if (error != null) {
|
|
logger.error({ err: error, doc_id }, error.message)
|
|
return callback(error)
|
|
}
|
|
if (ranges != null && ranges.indexOf('\u0000') !== -1) {
|
|
error = new Error('null bytes found in ranges')
|
|
// this check was added to catch memory corruption in JSON.stringify
|
|
logger.error({ err: error, doc_id, ranges }, error.message)
|
|
return callback(error)
|
|
}
|
|
const multi = rclient.multi()
|
|
multi.set(keys.docLines({ doc_id }), newDocLines) // index 0
|
|
multi.set(keys.docVersion({ doc_id }), newVersion) // index 1
|
|
multi.set(keys.docHash({ doc_id }), newHash) // index 2
|
|
multi.ltrim(
|
|
keys.docOps({ doc_id }),
|
|
-RedisManager.DOC_OPS_MAX_LENGTH,
|
|
-1
|
|
) // index 3
|
|
if (ranges != null) {
|
|
multi.set(keys.ranges({ doc_id }), ranges) // index 4
|
|
} else {
|
|
multi.del(keys.ranges({ doc_id })) // also index 4
|
|
}
|
|
// push the ops last so we can get the lengths at fixed index position 7
|
|
if (jsonOps.length > 0) {
|
|
multi.rpush(keys.docOps({ doc_id }), ...Array.from(jsonOps)) // index 5
|
|
// expire must come after rpush since before it will be a no-op if the list is empty
|
|
multi.expire(keys.docOps({ doc_id }), RedisManager.DOC_OPS_TTL) // index 6
|
|
if (projectHistoryType === 'project-history') {
|
|
metrics.inc('history-queue', 1, { status: 'skip-track-changes' })
|
|
logger.log(
|
|
{ doc_id },
|
|
'skipping push of uncompressed ops for project using project-history'
|
|
)
|
|
} else {
|
|
// project is using old track-changes history service
|
|
metrics.inc('history-queue', 1, { status: 'track-changes' })
|
|
multi.rpush(
|
|
historyKeys.uncompressedHistoryOps({ doc_id }),
|
|
...Array.from(jsonOps)
|
|
) // index 7
|
|
}
|
|
// Set the unflushed timestamp to the current time if the doc
|
|
// hasn't been modified before (the content in mongo has been
|
|
// valid up to this point). Otherwise leave it alone ("NX" flag).
|
|
multi.set(keys.unflushedTime({ doc_id }), Date.now(), 'NX')
|
|
multi.set(keys.lastUpdatedAt({ doc_id }), Date.now()) // index 8
|
|
if (updateMeta != null ? updateMeta.user_id : undefined) {
|
|
multi.set(keys.lastUpdatedBy({ doc_id }), updateMeta.user_id) // index 9
|
|
} else {
|
|
multi.del(keys.lastUpdatedBy({ doc_id })) // index 9
|
|
}
|
|
}
|
|
return multi.exec(function (error, result) {
|
|
let docUpdateCount
|
|
if (error != null) {
|
|
return callback(error)
|
|
}
|
|
|
|
if (projectHistoryType === 'project-history') {
|
|
docUpdateCount = undefined // only using project history, don't bother with track-changes
|
|
} else {
|
|
// project is using old track-changes history service
|
|
docUpdateCount = result[7] // length of uncompressedHistoryOps queue (index 7)
|
|
}
|
|
|
|
if (
|
|
jsonOps.length > 0 &&
|
|
__guard__(
|
|
Settings.apis != null ? Settings.apis.project_history : undefined,
|
|
(x) => x.enabled
|
|
)
|
|
) {
|
|
metrics.inc('history-queue', 1, { status: 'project-history' })
|
|
return ProjectHistoryRedisManager.queueOps(
|
|
project_id,
|
|
...Array.from(jsonOps),
|
|
(error, projectUpdateCount) =>
|
|
callback(null, docUpdateCount, projectUpdateCount)
|
|
)
|
|
} else {
|
|
return callback(null, docUpdateCount)
|
|
}
|
|
})
|
|
})
|
|
})
|
|
},
|
|
|
|
renameDoc(project_id, doc_id, user_id, update, projectHistoryId, callback) {
|
|
if (callback == null) {
|
|
callback = function (error) {}
|
|
}
|
|
return RedisManager.getDoc(project_id, doc_id, function (
|
|
error,
|
|
lines,
|
|
version
|
|
) {
|
|
if (error != null) {
|
|
return callback(error)
|
|
}
|
|
|
|
if (lines != null && version != null) {
|
|
return rclient.set(
|
|
keys.pathname({ doc_id }),
|
|
update.newPathname,
|
|
function (error) {
|
|
if (error != null) {
|
|
return callback(error)
|
|
}
|
|
return ProjectHistoryRedisManager.queueRenameEntity(
|
|
project_id,
|
|
projectHistoryId,
|
|
'doc',
|
|
doc_id,
|
|
user_id,
|
|
update,
|
|
callback
|
|
)
|
|
}
|
|
)
|
|
} else {
|
|
return ProjectHistoryRedisManager.queueRenameEntity(
|
|
project_id,
|
|
projectHistoryId,
|
|
'doc',
|
|
doc_id,
|
|
user_id,
|
|
update,
|
|
callback
|
|
)
|
|
}
|
|
})
|
|
},
|
|
|
|
clearUnflushedTime(doc_id, callback) {
|
|
if (callback == null) {
|
|
callback = function (error) {}
|
|
}
|
|
return rclient.del(keys.unflushedTime({ doc_id }), callback)
|
|
},
|
|
|
|
getDocIdsInProject(project_id, callback) {
|
|
if (callback == null) {
|
|
callback = function (error, doc_ids) {}
|
|
}
|
|
return rclient.smembers(keys.docsInProject({ project_id }), callback)
|
|
},
|
|
|
|
getDocTimestamps(doc_ids, callback) {
|
|
// get lastupdatedat timestamps for an array of doc_ids
|
|
if (callback == null) {
|
|
callback = function (error, result) {}
|
|
}
|
|
return async.mapSeries(
|
|
doc_ids,
|
|
(doc_id, cb) => rclient.get(keys.lastUpdatedAt({ doc_id }), cb),
|
|
callback
|
|
)
|
|
},
|
|
|
|
queueFlushAndDeleteProject(project_id, callback) {
|
|
// store the project id in a sorted set ordered by time with a random offset to smooth out spikes
|
|
const SMOOTHING_OFFSET =
|
|
Settings.smoothingOffset > 0
|
|
? Math.round(Settings.smoothingOffset * Math.random())
|
|
: 0
|
|
return rclient.zadd(
|
|
keys.flushAndDeleteQueue(),
|
|
Date.now() + SMOOTHING_OFFSET,
|
|
project_id,
|
|
callback
|
|
)
|
|
},
|
|
|
|
getNextProjectToFlushAndDelete(cutoffTime, callback) {
|
|
// find the oldest queued flush that is before the cutoff time
|
|
if (callback == null) {
|
|
callback = function (error, key, timestamp) {}
|
|
}
|
|
return rclient.zrangebyscore(
|
|
keys.flushAndDeleteQueue(),
|
|
0,
|
|
cutoffTime,
|
|
'WITHSCORES',
|
|
'LIMIT',
|
|
0,
|
|
1,
|
|
function (err, reply) {
|
|
if (err != null) {
|
|
return callback(err)
|
|
}
|
|
if (!(reply != null ? reply.length : undefined)) {
|
|
return callback()
|
|
} // return if no projects ready to be processed
|
|
// pop the oldest entry (get and remove in a multi)
|
|
const multi = rclient.multi()
|
|
// Poor man's version of ZPOPMIN, which is only available in Redis 5.
|
|
multi.zrange(keys.flushAndDeleteQueue(), 0, 0, 'WITHSCORES')
|
|
multi.zremrangebyrank(keys.flushAndDeleteQueue(), 0, 0)
|
|
multi.zcard(keys.flushAndDeleteQueue()) // the total length of the queue (for metrics)
|
|
return multi.exec(function (err, reply) {
|
|
if (err != null) {
|
|
return callback(err)
|
|
}
|
|
if (!(reply != null ? reply.length : undefined)) {
|
|
return callback()
|
|
}
|
|
const [key, timestamp] = Array.from(reply[0])
|
|
const queueLength = reply[2]
|
|
return callback(null, key, timestamp, queueLength)
|
|
})
|
|
}
|
|
)
|
|
},
|
|
|
|
_serializeRanges(ranges, callback) {
|
|
if (callback == null) {
|
|
callback = function (error, serializedRanges) {}
|
|
}
|
|
let jsonRanges = JSON.stringify(ranges)
|
|
if (jsonRanges != null && jsonRanges.length > MAX_RANGES_SIZE) {
|
|
return callback(new Error('ranges are too large'))
|
|
}
|
|
if (jsonRanges === '{}') {
|
|
// Most doc will have empty ranges so don't fill redis with lots of '{}' keys
|
|
jsonRanges = null
|
|
}
|
|
return callback(null, jsonRanges)
|
|
},
|
|
|
|
_deserializeRanges(ranges) {
|
|
if (ranges == null || ranges === '') {
|
|
return {}
|
|
} else {
|
|
return JSON.parse(ranges)
|
|
}
|
|
},
|
|
|
|
_computeHash(docLines) {
|
|
// use sha1 checksum of doclines to detect data corruption.
|
|
//
|
|
// note: must specify 'utf8' encoding explicitly, as the default is
|
|
// binary in node < v5
|
|
return crypto.createHash('sha1').update(docLines, 'utf8').digest('hex')
|
|
}
|
|
}
|
|
|
|
function __guard__(value, transform) {
|
|
return typeof value !== 'undefined' && value !== null
|
|
? transform(value)
|
|
: undefined
|
|
}
|