From b39de5538a8634e4ab5ddeed921f99b2351ac83b Mon Sep 17 00:00:00 2001
From: Dmitry Lyukov <d.lyukov@ya.ru>
Date: Mon, 28 Mar 2016 23:10:27 +0300
Subject: [PATCH] Added benchmark for saving to JSON

---
 src/saving_test.go | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/src/saving_test.go b/src/saving_test.go
index 3c0f40b..86a9378 100644
--- a/src/saving_test.go
+++ b/src/saving_test.go
@@ -57,3 +57,35 @@ func BenchmarkCsvSaving(b *testing.B) {
 
 	SaveToCsvWriter(data, keyMap, tmpFile, true)
 }
+
+func BenchmarkJSONSavingOnlySum(b *testing.B) {
+	data := GenerateRandStatsForTime(b.N)
+	keyMap := GetKeymap()
+
+	tmpFile, err := ioutil.TempFile(os.TempDir(), "benchmark")
+	if err != nil {
+		b.Fatal(err)
+	}
+
+	defer os.Remove(tmpFile.Name())
+
+	b.ResetTimer()
+
+	SaveToJSONWriter(data, keyMap, tmpFile, false)
+}
+
+func BenchmarkJSONSaving(b *testing.B) {
+	data := GenerateRandStatsForTime(b.N)
+	keyMap := GetKeymap()
+
+	tmpFile, err := ioutil.TempFile(os.TempDir(), "benchmark")
+	if err != nil {
+		b.Fatal(err)
+	}
+
+	defer os.Remove(tmpFile.Name())
+
+	b.ResetTimer()
+
+	SaveToJSONWriter(data, keyMap, tmpFile, true)
+}