Move all benchmark code into bench.lua

Move benchmark() into bench.lua since it not used elsewhere.
Replace posix.gettimeofday() with socket.gettime() to improve portability.

1 files changed, 34 insertions, 0 deletions

diff --git a/tests/bench.lua b/tests/bench.lua
index 847365b..c1808cf 100755
--- a/tests/bench.lua
+++ b/tests/bench.lua
@@ -7,8 +7,40 @@
 -- Mark Pulford <mark@kyne.com.au>
 
 require "common"
+require "socket"
 local json = require "cjson"
 
+function benchmark(tests, iter, rep)
+    local function bench(func, iter)
+        collectgarbage("collect")
+        local t = socket.gettime()
+        for i = 1, iter do
+            func(i)
+        end
+        t = socket.gettime() - t
+        return (iter / t)
+    end
+
+    local test_results = {}
+    for name, func in pairs(tests) do
+        -- k(number), v(string)
+        -- k(string), v(function)
+        -- k(number), v(function)
+        if type(func) == "string" then
+            name = func
+            func = _G[name]
+        end
+        local result = {}
+        for i = 1, rep do
+            result[i] = bench(func, iter)
+        end
+        table.sort(result)
+        test_results[name] = result[rep]
+    end
+
+    return test_results
+end
+
 function bench_file(filename)
     local data_json = file_load(filename)
     local data_obj = json.decode(data_json)
@@ -28,6 +60,8 @@ function bench_file(filename)
     return benchmark(tests, 5000, 5)
 end
 
+cjson.encode_keep_buffer(true)
+
 for i = 1, #arg do
     local results = bench_file(arg[i])
     for k, v in pairs(results) do