From fb4212043f9e54bae86f13b021038b1b5cb9d246 Mon Sep 17 00:00:00 2001
From: Aleksey Lobanov <alekseylobanov1@gmail.com>
Date: Mon, 28 Mar 2016 23:59:29 +0300
Subject: [PATCH] Added benchmarks for saving in csv.gz and json.gz formats

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

diff --git a/src/saving_test.go b/src/saving_test.go
index 86a9378..0434b0d 100644
--- a/src/saving_test.go
+++ b/src/saving_test.go
@@ -89,3 +89,35 @@ func BenchmarkJSONSaving(b *testing.B) {
 
 	SaveToJSONWriter(data, keyMap, tmpFile, true)
 }
+
+func BenchmarkCsvGzSaving(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()
+
+	SaveToCsvGzWriter(data, keyMap, tmpFile, true)
+}
+
+func BenchmarkJSONGzSaving(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()
+
+	SaveToJSONGzWriter(data, keyMap, tmpFile, true)
+}