add a benchmark test

Signed-off-by: Daniel Dao <[email protected]>
Graylog2 · Jan 20, 2016 · 8fa7145 · 8fa7145
1 parent 4f441fa
commit 8fa7145
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/gelf/writer_test.go b/gelf/writer_test.go
@@ -5,9 +5,12 @@
 package gelf
 
 import (
+ "compress/flate"
  "crypto/rand"
  "encoding/base64"
  "fmt"
+ "io"
+ "io/ioutil"
  "strings"
  "testing"
  "time"
@@ -228,3 +231,20 @@ func TestExtraData(t *testing.T) {
  }
  }
 }
+
+func BenchmarkWriteBestSpeed(b *testing.B) {
+ r, err := NewReader("127.0.0.1:0")
+ if err != nil {
+ b.Fatalf("NewReader: %s", err)
+ }
+ go io.Copy(ioutil.Discard, r)
+ w, err := NewWriter(r.Addr())
+ if err != nil {
+ b.Fatalf("NewWriter: %s", err)
+ }
+ w.CompressionType = flate.BestSpeed
+ b.ResetTimer()
+ for i := 0; i < b.N; i++ {
+ w.Write([]byte("This is a message"))
+ }
+}