overleaf/services/web/scripts/back_fill_doc_rev.js

59 lines
1.2 KiB
JavaScript
Raw Normal View History

const { batchedUpdate } = require('./helpers/batchedUpdate')
const DRY_RUN = !process.argv.includes('--dry-run=false')
const LOG_EVERY_IN_S = parseInt(process.env.LOG_EVERY_IN_S, 10) || 5
async function main(DRY_RUN) {
let processed = 0
let deleted = 0
let lastLog = 0
function logProgress() {
console.log(`rev missing ${processed} | deleted=true ${deleted}`)
}
await batchedUpdate(
'docs',
{ rev: { $exists: false } },
async (docsCollection, docs) => {
if (!DRY_RUN) {
await docsCollection.updateMany(
{
_id: { $in: docs.map(doc => doc._id) },
rev: { $exists: false },
},
{ $set: { rev: 1 } }
)
}
processed += docs.length
deleted += docs.filter(doc => doc.deleted).length
if (Date.now() - lastLog >= LOG_EVERY_IN_S * 1000) {
logProgress()
lastLog = Date.now()
}
},
{
_id: 1,
deleted: true,
}
)
logProgress()
}
module.exports = main
if (require.main === module) {
main(DRY_RUN)
.then(() => {
console.log('Done.')
process.exit(0)
})
.catch(error => {
console.error({ error })
process.exit(1)
})
}