فهرست منبع

Don't run redis commands in parallel for easier consistency reasoning

James Allen 10 سال پیش
والد
کامیت
c823e06912

+ 23 - 26
services/document-updater/app/coffee/RedisManager.coffee

@@ -11,40 +11,37 @@ Errors = require "./Errors"
 minutes = 60 # seconds for Redis expire
 minutes = 60 # seconds for Redis expire
 
 
 module.exports = RedisManager =
 module.exports = RedisManager =
-	putDocInMemory : (project_id, doc_id, docLines, version, callback)->
+	putDocInMemory : (project_id, doc_id, docLines, version, _callback)->
 		timer = new metrics.Timer("redis.put-doc")
 		timer = new metrics.Timer("redis.put-doc")
-		logger.log project_id:project_id, doc_id:doc_id, version: version, "putting doc in redis"
-		async.parallel [
-			(cb) ->
-				multi = rclient.multi()
-				multi.set keys.docLines(doc_id:doc_id), JSON.stringify(docLines)
-				multi.set keys.projectKey({doc_id:doc_id}), project_id
-				multi.set keys.docVersion(doc_id:doc_id), version
-				multi.exec cb
-			(cb) ->
-				rclient.sadd keys.docsInProject(project_id:project_id), doc_id, cb
-		], (err) ->
+		callback = (error) ->
 			timer.done()
 			timer.done()
-			callback(err)
+			_callback(error)
+		logger.log project_id:project_id, doc_id:doc_id, version: version, "putting doc in redis"
+		multi = rclient.multi()
+		multi.set keys.docLines(doc_id:doc_id), JSON.stringify(docLines)
+		multi.set keys.projectKey({doc_id:doc_id}), project_id
+		multi.set keys.docVersion(doc_id:doc_id), version
+		multi.exec (error) ->
+			return callback(error) if error?
+			rclient.sadd keys.docsInProject(project_id:project_id), doc_id, callback
 
 
-	removeDocFromMemory : (project_id, doc_id, callback)->
+	removeDocFromMemory : (project_id, doc_id, _callback)->
 		logger.log project_id:project_id, doc_id:doc_id, "removing doc from redis"
 		logger.log project_id:project_id, doc_id:doc_id, "removing doc from redis"
-		async.parallel [
-			(cb) ->
-				multi = rclient.multi()
-				multi.del keys.docLines(doc_id:doc_id)
-				multi.del keys.projectKey(doc_id:doc_id)
-				multi.del keys.docVersion(doc_id:doc_id)
-				multi.exec cb
-			(cb) ->
-				rclient.srem keys.docsInProject(project_id:project_id), doc_id, cb
-		], (err) ->
+		callback = (err) ->
 			if err?
 			if err?
 				logger.err project_id:project_id, doc_id:doc_id, err:err, "error removing doc from redis"
 				logger.err project_id:project_id, doc_id:doc_id, err:err, "error removing doc from redis"
-				callback(err, null)
+				_callback(err)
 			else
 			else
 				logger.log project_id:project_id, doc_id:doc_id, "removed doc from redis"
 				logger.log project_id:project_id, doc_id:doc_id, "removed doc from redis"
-				callback()
+				_callback()
+
+		multi = rclient.multi()
+		multi.del keys.docLines(doc_id:doc_id)
+		multi.del keys.projectKey(doc_id:doc_id)
+		multi.del keys.docVersion(doc_id:doc_id)
+		multi.exec (error) ->
+			return callback(error) if error?
+			rclient.srem keys.docsInProject(project_id:project_id), doc_id, callback
 
 
 	getDoc : (doc_id, callback = (error, lines, version) ->)->
 	getDoc : (doc_id, callback = (error, lines, version) ->)->
 		timer = new metrics.Timer("redis.get-doc")
 		timer = new metrics.Timer("redis.get-doc")

+ 4 - 4
services/document-updater/app/coffee/TrackChangesManager.coffee

@@ -25,10 +25,10 @@ module.exports = TrackChangesManager =
 	FLUSH_EVERY_N_OPS: 50
 	FLUSH_EVERY_N_OPS: 50
 	pushUncompressedHistoryOp: (project_id, doc_id, op, callback = (error) ->) ->
 	pushUncompressedHistoryOp: (project_id, doc_id, op, callback = (error) ->) ->
 		jsonOp = JSON.stringify op
 		jsonOp = JSON.stringify op
-		async.parallel [
-			(cb) -> rclient.rpush "UncompressedHistoryOps:#{doc_id}", jsonOp, cb
-			(cb) -> rclient.sadd "DocsWithHistoryOps:#{project_id}", doc_id, cb
-		], (error, results) ->
+		multi = rclient.multi()
+		multi.rpush "UncompressedHistoryOps:#{doc_id}", jsonOp
+		multi.sadd "DocsWithHistoryOps:#{project_id}", doc_id
+		multi.exec (error, results) ->
 			return callback(error) if error?
 			return callback(error) if error?
 			[length, _] = results
 			[length, _] = results
 			if length > 0 and length % TrackChangesManager.FLUSH_EVERY_N_OPS == 0
 			if length > 0 and length % TrackChangesManager.FLUSH_EVERY_N_OPS == 0

+ 8 - 6
services/document-updater/test/unit/coffee/TrackChangesManager/TrackChangesManagerTests.coffee

@@ -43,8 +43,10 @@ describe "TrackChangesManager", ->
 	describe "pushUncompressedHistoryOp", ->
 	describe "pushUncompressedHistoryOp", ->
 		beforeEach ->
 		beforeEach ->
 			@op = { op: [{ i: "foo", p: 4 }] }
 			@op = { op: [{ i: "foo", p: 4 }] }
-			@rclient.rpush = sinon.stub().yields(null, @length = 42)
-			@rclient.sadd = sinon.stub().yields()
+			@rclient.multi = sinon.stub().returns(@multi = {})
+			@multi.rpush = sinon.stub()
+			@multi.sadd = sinon.stub()
+			@multi.exec = sinon.stub().yields(null, [@length = 42, "foo"])
 			@TrackChangesManager.flushDocChanges = sinon.stub().callsArg(2)
 			@TrackChangesManager.flushDocChanges = sinon.stub().callsArg(2)
 
 
 		describe "pushing the op", ->
 		describe "pushing the op", ->
@@ -52,10 +54,10 @@ describe "TrackChangesManager", ->
 				@TrackChangesManager.pushUncompressedHistoryOp @project_id, @doc_id, @op, @callback
 				@TrackChangesManager.pushUncompressedHistoryOp @project_id, @doc_id, @op, @callback
 
 
 			it "should push the op into redis", ->
 			it "should push the op into redis", ->
-				@rclient.rpush
+				@multi.rpush
 					.calledWith("UncompressedHistoryOps:#{@doc_id}", JSON.stringify @op)
 					.calledWith("UncompressedHistoryOps:#{@doc_id}", JSON.stringify @op)
 					.should.equal true
 					.should.equal true
-				@rclient.sadd
+				@multi.sadd
 					.calledWith("DocsWithHistoryOps:#{@project_id}", @doc_id)
 					.calledWith("DocsWithHistoryOps:#{@project_id}", @doc_id)
 					.should.equal true
 					.should.equal true
 
 
@@ -67,7 +69,7 @@ describe "TrackChangesManager", ->
 
 
 		describe "when there are a multiple of FLUSH_EVERY_N_OPS ops", ->
 		describe "when there are a multiple of FLUSH_EVERY_N_OPS ops", ->
 			beforeEach ->
 			beforeEach ->
-				@rclient.rpush = sinon.stub().yields(null, 2 * @TrackChangesManager.FLUSH_EVERY_N_OPS)
+				@multi.exec = sinon.stub().yields(null, [2 * @TrackChangesManager.FLUSH_EVERY_N_OPS, "foo"])
 				@TrackChangesManager.pushUncompressedHistoryOp @project_id, @doc_id, @op, @callback
 				@TrackChangesManager.pushUncompressedHistoryOp @project_id, @doc_id, @op, @callback
 
 
 			it "should tell the track changes api to flush", ->
 			it "should tell the track changes api to flush", ->
@@ -77,7 +79,7 @@ describe "TrackChangesManager", ->
 
 
 		describe "when TrackChangesManager errors", ->
 		describe "when TrackChangesManager errors", ->
 			beforeEach ->
 			beforeEach ->
-				@rclient.rpush = sinon.stub().yields(null, 2 * @TrackChangesManager.FLUSH_EVERY_N_OPS)
+				@multi.exec = sinon.stub().yields(null, [2 * @TrackChangesManager.FLUSH_EVERY_N_OPS, "foo"])
 				@TrackChangesManager.flushDocChanges = sinon.stub().callsArgWith(2, @error = new Error("oops"))
 				@TrackChangesManager.flushDocChanges = sinon.stub().callsArgWith(2, @error = new Error("oops"))
 				@TrackChangesManager.pushUncompressedHistoryOp @project_id, @doc_id, @op, @callback
 				@TrackChangesManager.pushUncompressedHistoryOp @project_id, @doc_id, @op, @callback