overleaf/services/web/scripts/purge_non_logged_in_sessions.js
Henry Oswald 9b71561814 Merge pull request #7302 from overleaf/ho-session-purge
mvp to purge session keys from redis

GitOrigin-RevId: d1298d4b042164d9c94a7a1d1b7560b6531ae093
2022-04-05 12:19:36 +00:00

60 lines
1.4 KiB
JavaScript

const RedisWrapper = require('@overleaf/redis-wrapper')
const Settings = require('@overleaf/settings')
const SessionManager = require('../app/src/Features/Authentication/SessionManager')
const async = require('async')
const _ = require('lodash')
const redis = RedisWrapper.createClient(Settings.redis.web)
let totalDeletedSessions = 0
const queue = async.queue(function (sessKey, callback) {
const cb = _.once(callback)
redis.get(sessKey, (_err, session) => {
try {
if (SessionManager.isUserLoggedIn(JSON.parse(session))) {
cb()
} else {
redis.del(sessKey, () => {
totalDeletedSessions++
cb()
})
}
} catch (err) {
console.log(`${sessKey} couldn't parse`)
cb()
}
})
}, 10)
function scanAndPurge(cb) {
const stream = redis.scanStream({
match: 'sess:*',
count: 100,
})
console.log('starting scan')
stream.on('data', resultKeys => {
console.log(`Keys found, count:${resultKeys.length}`)
console.log(`Keys deleted so far: ${totalDeletedSessions}`)
queue.push(resultKeys)
})
stream.on('end', () => {
queue.drain = () => {
console.log(
`All sessions have been checked, ${totalDeletedSessions} deleted`
)
cb()
}
})
stream.on('error', err => {
console.log(err)
})
}
scanAndPurge(err => {
console.error(err)
process.exit()
})