From 43f51355b26d69be2cafc1fa14e06e08e3e84d4d Mon Sep 17 00:00:00 2001
From: Brian Gough <bjg@network-theory.co.uk>
Date: Thu, 20 Apr 2017 11:01:46 +0100
Subject: [PATCH] add limit parameter to flushAll

---
 .../track-changes/app/coffee/HttpController.coffee   | 12 ++++++++----
 .../track-changes/app/coffee/UpdatesManager.coffee   |  7 ++++---
 2 files changed, 12 insertions(+), 7 deletions(-)

diff --git a/services/track-changes/app/coffee/HttpController.coffee b/services/track-changes/app/coffee/HttpController.coffee
index 0b59a6e1ec..4f99cdfda3 100644
--- a/services/track-changes/app/coffee/HttpController.coffee
+++ b/services/track-changes/app/coffee/HttpController.coffee
@@ -23,12 +23,16 @@ module.exports = HttpController =
 			res.send 204
 
 	flushAll: (req, res, next = (error) ->) ->
-		logger.log "flushing all projects"
-		UpdatesManager.flushAll (error, result) ->
+		# limit on projects to flush or -1 for all (default)
+		limit = if req.query.limit? then parseInt(req.query.limit, 10) else -1
+		logger.log {limit: limit}, "flushing all projects"
+		UpdatesManager.flushAll limit, (error, result) ->
 			return next(error) if error?
-			{failed, succeeded} = result
+			{failed, succeeded, all} = result
 			status = "#{succeeded.length} succeeded, #{failed.length} failed"
-			if failed.length > 0
+			if limit == 0
+				res.status(200).send "#{status}\nwould flush:\n#{all.join('\n')}\n"
+			else if failed.length > 0
 				logger.log {failed: failed, succeeded: succeeded}, "error flushing projects"
 				res.status(500).send "#{status}\nfailed to flush:\n#{failed.join('\n')}\n"
 			else
diff --git a/services/track-changes/app/coffee/UpdatesManager.coffee b/services/track-changes/app/coffee/UpdatesManager.coffee
index 3c12b2a713..f3571dc16d 100644
--- a/services/track-changes/app/coffee/UpdatesManager.coffee
+++ b/services/track-changes/app/coffee/UpdatesManager.coffee
@@ -145,12 +145,13 @@ module.exports = UpdatesManager =
 				async.parallelLimit jobs, 5, callback
 
 	# flush all outstanding changes
-	flushAll: (callback = (error, result) ->) ->
+	flushAll: (limit, callback = (error, result) ->) ->
 		RedisManager.getProjectIdsWithHistoryOps (error, project_ids) ->
 			return callback(error) if error?
 			logger.log {count: project_ids?.length, project_ids: project_ids}, "found projects"
 			jobs = []
-			for project_id in project_ids
+			selectedProjects = if limit < 0 then project_ids else project_ids[0...limit]
+			for project_id in selectedProjects
 				do (project_id) ->
 					jobs.push (cb) ->
 						UpdatesManager.processUncompressedUpdatesForProject project_id, (err) ->
@@ -159,7 +160,7 @@ module.exports = UpdatesManager =
 				return callback(error) if error?
 				failedProjects = (x.project_id for x in result when x.failed)
 				succeededProjects = (x.project_id for x in result when not x.failed)
-				callback(null, {failed: failedProjects, succeeded: succeededProjects})
+				callback(null, {failed: failedProjects, succeeded: succeededProjects, all: project_ids})
 
 	getDanglingUpdates: (callback = (error, doc_ids) ->) ->
 		RedisManager.getAllDocIdsWithHistoryOps (error, all_doc_ids) ->