overleaf/services/track-changes/benchmark.coffee

57 lines
1.3 KiB
CoffeeScript
Raw Normal View History

2014-02-26 12:35:10 -05:00
request = require "request"
rclient = require("redis").createClient()
async = require "async"
{ObjectId} = require("./app/js/mongojs")
NO_OF_DOCS = 100
NO_OF_UPDATES = 200
user_id = ObjectId().toString()
updates = for i in [1..NO_OF_UPDATES]
{
op: { i: "a", p: 0 }
v: i
meta: ts: new Date(), user_id: user_id
}
jsonUpdates = (JSON.stringify(u) for u in updates)
doc_ids = (ObjectId().toString() for i in [1..NO_OF_DOCS])
populateRedis = (callback = (error) ->) ->
console.log "Populating Redis queues..."
jobs = []
for doc_id in doc_ids
do (doc_id) ->
jobs.push (callback) ->
rclient.rpush "UncompressedHistoryOps:#{doc_id}", jsonUpdates..., callback
async.series jobs, (error) ->
return callback(error) if error?
console.log "Done."
callback()
flushDocs = (callback = (error) ->) ->
console.log "Flushing docs..."
inProgress = 0
jobs = []
for doc_id in doc_ids
do (doc_id) ->
jobs.push (callback) ->
inProgress = inProgress + 1
request.post "http://localhost:3014/doc/#{doc_id}/flush", (error) ->
inProgress = inProgress - 1
console.log Date.now(), "In progress: #{inProgress}"
callback(error)
async.parallel jobs, (error) ->
return callback(error) if error?
console.log "Done."
callback()
populateRedis (error) ->
throw error if error?
flushDocs (error) ->
throw error if error?
process.exit(0)