1 files changed, 83 insertions, 0 deletions

diff --git a/tests/bench.lua b/tests/bench.lua
new file mode 100755
index 0000000..2b5177b
--- /dev/null
+++ b/tests/bench.lua
@@ -0,0 +1,83 @@
+#!/usr/bin/env lua
+
+-- Simple JSON benchmark.
+--
+-- Your Mileage May Vary.
+--
+-- Mark Pulford <mark@kyne.com.au>
+
+require "socket"
+local json = require "cjson"
+local misc = require "cjson-misc"
+
+function benchmark(tests, seconds, rep)
+    local function bench(func, iter)
+        -- collectgarbage("stop")
+        collectgarbage("collect")
+        local t = socket.gettime()
+        for i = 1, iter do
+            func(i)
+        end
+        t = socket.gettime() - t
+        -- collectgarbage("restart")
+        return (iter / t)
+    end
+
+    -- Roughly calculate the number of interations required
+    -- to obtain a particular time period.
+    local function calc_iter(func, seconds)
+        local base_iter = 10
+        local rate = (bench(func, base_iter) + bench(func, base_iter)) / 2
+        return math.ceil(seconds * rate)
+    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 iter = calc_iter(func, seconds)
+        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 = misc.file_load(filename)
+    local data_obj = json.decode(data_json)
+
+    local function test_encode ()
+        json.encode(data_obj)
+    end
+    local function test_decode ()
+        json.decode(data_json)
+    end
+
+    local tests = {
+        encode = test_encode,
+        decode = test_decode
+    }
+
+    return benchmark(tests, 0.1, 5)
+end
+
+json.encode_keep_buffer(true)
+
+for i = 1, #arg do
+    local results = bench_file(arg[i])
+    for k, v in pairs(results) do
+        print(string.format("%s: %s: %d", arg[i], k, v))
+    end
+end
+
+-- vi:ai et sw=4 ts=4: