|
@@ -199,3 +199,50 @@ func TestJSONFileLoggerWithLabelsEnv(t *testing.T) {
|
|
t.Fatalf("Wrong log attrs: %q, expected %q", extra, expected)
|
|
t.Fatalf("Wrong log attrs: %q, expected %q", extra, expected)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
+
|
|
|
|
+func BenchmarkJSONFileLoggerWithReader(b *testing.B) {
|
|
|
|
+ b.StopTimer()
|
|
|
|
+ b.ResetTimer()
|
|
|
|
+ cid := "a7317399f3f857173c6179d44823594f8294678dea9999662e5c625b5a1c7657"
|
|
|
|
+ dir, err := ioutil.TempDir("", "json-logger-bench")
|
|
|
|
+ if err != nil {
|
|
|
|
+ b.Fatal(err)
|
|
|
|
+ }
|
|
|
|
+ defer os.RemoveAll(dir)
|
|
|
|
+
|
|
|
|
+ l, err := New(logger.Context{
|
|
|
|
+ ContainerID: cid,
|
|
|
|
+ LogPath: filepath.Join(dir, "container.log"),
|
|
|
|
+ })
|
|
|
|
+ if err != nil {
|
|
|
|
+ b.Fatal(err)
|
|
|
|
+ }
|
|
|
|
+ defer l.Close()
|
|
|
|
+ msg := &logger.Message{ContainerID: cid, Line: []byte("line"), Source: "src1"}
|
|
|
|
+ jsonlog, err := (&jsonlog.JSONLog{Log: string(msg.Line) + "\n", Stream: msg.Source, Created: msg.Timestamp}).MarshalJSON()
|
|
|
|
+ if err != nil {
|
|
|
|
+ b.Fatal(err)
|
|
|
|
+ }
|
|
|
|
+ b.SetBytes(int64(len(jsonlog)+1) * 30)
|
|
|
|
+
|
|
|
|
+ b.StartTimer()
|
|
|
|
+
|
|
|
|
+ go func() {
|
|
|
|
+ for i := 0; i < b.N; i++ {
|
|
|
|
+ for j := 0; j < 30; j++ {
|
|
|
|
+ l.Log(msg)
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+ l.Close()
|
|
|
|
+ }()
|
|
|
|
+
|
|
|
|
+ lw := l.(logger.LogReader).ReadLogs(logger.ReadConfig{Follow: true})
|
|
|
|
+ watchClose := lw.WatchClose()
|
|
|
|
+ for {
|
|
|
|
+ select {
|
|
|
|
+ case <-lw.Msg:
|
|
|
|
+ case <-watchClose:
|
|
|
|
+ return
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+}
|