Add performance test for rapidjson.Document:parse() and stringify().

xpol · Oct 31, 2016 · 2840c85 · 2840c85
1 parent 0d869ad
commit 2840c85
Showing 1 changed file with 57 additions and 46 deletions.
diff --git a/performance/run.lua b/performance/run.lua
@@ -3,73 +3,84 @@
 
 
 local function time(f, times)
-  collectgarbage()
-  local gettime = os.clock
+	collectgarbage()
+	local gettime = os.clock
 
-  local ok, socket = pcall(require, 'socket')
-  if ok then
-    gettime = socket.gettime
-  end
+	local ok, socket = pcall(require, 'socket')
+	if ok then
+		gettime = socket.gettime
+	end
 
-  local start = gettime()
+	local start = gettime()
 
-  for _=0,times do f() end
+	for _=0,times do f() end
 
-  local stop = gettime()
+	local stop = gettime()
 
-  return stop - start
+	return stop - start
 end
 
 
 local function readfile(file)
-  local f = io.open(file)
-  if not f then return nil end
-  local d = f:read('*a')
-  f:close()
-  return d
+	local f = io.open(file)
+	if not f then return nil end
+	local d = f:read('*a')
+	f:close()
+	return d
 end
 
 
 local function profile(jsonfile, times)
-  times = times or 10000
-
-  print(jsonfile..': (x'..times..')')
-  print('', 'module', '  decoding', '  encoding')
-  local d = readfile(jsonfile)
-
-  local rapidjson = require('rapidjson')
-  local cjson = require('cjson')
-  local dkjson = require('dkjson')
-
-  local modules = {
-    {'dkjson', dkjson.decode, dkjson.encode},
-    {'cjson', cjson.decode, cjson.encode},
-    {'rapidjson', rapidjson.decode, rapidjson.encode},
-  }
-
-  for _, m in ipairs(modules) do
-    local name, dec, enc = m[1], m[2], m[3]
-    local td = time(function() dec(d) end, times)
-    local t = dec(d)
-    local te = time(function() enc(t) end, times)
-    print(string.format('\t%6s\t% 13.10f\t% 13.10f', name, td, te))
-  end
+	times = times or 10000
+
+	print(jsonfile..': (x'..times..')')
+	print('              module  decoding      encoding')
+	local d = readfile(jsonfile)
+
+	local rapidjson = require('rapidjson')
+	local cjson = require('cjson')
+	local dkjson = require('dkjson')
+
+	local function docParse(s)
+		local d = rapidjson.Document()
+		d:parse(s)
+		return d
+	end
+
+	local function docStringify(d)
+		return d:stringify()
+	end
+
+	local modules = {
+		{'            dkjson', dkjson.decode, dkjson.encode},
+		{'             cjson', cjson.decode, cjson.encode},
+		{'         rapidjson', rapidjson.decode, rapidjson.encode},
+		{'rapidjson.Document', docParse, docStringify},
+	}
+
+	for _, m in ipairs(modules) do
+		local name, dec, enc = m[1], m[2], m[3]
+		local td = time(function() dec(d) end, times)
+		local t = dec(d)
+		local te = time(function() enc(t) end, times)
+		print(string.format('% 20s % 13.10f % 13.10f', name, td, te))
+	end
 end
 
 local function main()
-  profile('performance/nulls.json')
-  profile('performance/booleans.json')
-  profile('performance/guids.json')
-  profile('performance/paragraphs.json')
-  profile('performance/floats.json')
-  profile('performance/integers.json')
-  profile('performance/mixed.json')
+	profile('performance/nulls.json')
+	profile('performance/booleans.json')
+	profile('performance/guids.json')
+	profile('performance/paragraphs.json')
+	profile('performance/floats.json')
+	profile('performance/integers.json')
+	profile('performance/mixed.json')
 end
 
 local r, m = pcall(main)
 
 if not r then
-  print(m)
+	print(m)
 end
 
 return 0