Bladeren bron

Merge pull request #7302 from overleaf/ho-session-purge

mvp to purge session keys from redis

GitOrigin-RevId: d1298d4b042164d9c94a7a1d1b7560b6531ae093
Henry Oswald 4 jaren geleden
bovenliggende
commit
9b71561814
1 gewijzigde bestanden met toevoegingen van 60 en 0 verwijderingen
  1. 60 0
      services/web/scripts/purge_non_logged_in_sessions.js

+ 60 - 0
services/web/scripts/purge_non_logged_in_sessions.js

@@ -0,0 +1,60 @@
+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()
+})