Added saving to system tempfile

This commit is contained in:
2016-03-27 02:48:33 +03:00
parent 64c02ca343
commit 719c3c2eea

View File

@@ -2,7 +2,10 @@
package main package main
import ( import (
"io/ioutil"
"log"
"math/rand" "math/rand"
"os"
"testing" "testing"
) )
@@ -20,8 +23,17 @@ func BenchmarkCsvSavingOnlySum(b *testing.B) {
} }
data = append(data, curStat) data = append(data, curStat)
} }
tmpFile, err := ioutil.TempFile(os.TempDir(), "benchmark")
if err != nil {
log.Fatal(err)
}
defer os.Remove(tmpFile.Name())
b.ResetTimer() b.ResetTimer()
SaveToCsvFile(data, keyMap, "/tmp/bla.csv", true)
SaveToCsvWriter(data, keyMap, tmpFile, true)
} }
func BenchmarkCsvSaving(b *testing.B) { func BenchmarkCsvSaving(b *testing.B) {
@@ -38,6 +50,14 @@ func BenchmarkCsvSaving(b *testing.B) {
} }
data = append(data, curStat) data = append(data, curStat)
} }
tmpFile, err := ioutil.TempFile(os.TempDir(), "benchmark")
if err != nil {
log.Fatal(err)
}
defer os.Remove(tmpFile.Name())
b.ResetTimer() b.ResetTimer()
SaveToCsvFile(data, keyMap, "/tmp/bla.csv", false)
SaveToCsvWriter(data, keyMap, tmpFile, false)
} }