79 lines
2.1 KiB
Go
79 lines
2.1 KiB
Go
![]() |
package jsonfilelog
|
||
|
|
||
|
import (
|
||
|
"io/ioutil"
|
||
|
"os"
|
||
|
"path/filepath"
|
||
|
"testing"
|
||
|
"time"
|
||
|
|
||
|
"github.com/docker/docker/daemon/logger"
|
||
|
"github.com/docker/docker/pkg/jsonlog"
|
||
|
)
|
||
|
|
||
|
func TestJSONFileLogger(t *testing.T) {
|
||
|
tmp, err := ioutil.TempDir("", "docker-logger-")
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
defer os.RemoveAll(tmp)
|
||
|
filename := filepath.Join(tmp, "container.log")
|
||
|
l, err := New(filename)
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
defer l.Close()
|
||
|
cid := "a7317399f3f857173c6179d44823594f8294678dea9999662e5c625b5a1c7657"
|
||
|
if err := l.Log(&logger.Message{ContainerID: cid, Line: []byte("line1"), Source: "src1"}); err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
if err := l.Log(&logger.Message{ContainerID: cid, Line: []byte("line2"), Source: "src2"}); err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
if err := l.Log(&logger.Message{ContainerID: cid, Line: []byte("line3"), Source: "src3"}); err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
res, err := ioutil.ReadFile(filename)
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
expected := `{"log":"line1\n","stream":"src1","time":"0001-01-01T00:00:00Z"}
|
||
|
{"log":"line2\n","stream":"src2","time":"0001-01-01T00:00:00Z"}
|
||
|
{"log":"line3\n","stream":"src3","time":"0001-01-01T00:00:00Z"}
|
||
|
`
|
||
|
|
||
|
if string(res) != expected {
|
||
|
t.Fatalf("Wrong log content: %q, expected %q", res, expected)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func BenchmarkJSONFileLogger(b *testing.B) {
|
||
|
tmp, err := ioutil.TempDir("", "docker-logger-")
|
||
|
if err != nil {
|
||
|
b.Fatal(err)
|
||
|
}
|
||
|
defer os.RemoveAll(tmp)
|
||
|
filename := filepath.Join(tmp, "container.log")
|
||
|
l, err := New(filename)
|
||
|
if err != nil {
|
||
|
b.Fatal(err)
|
||
|
}
|
||
|
defer l.Close()
|
||
|
cid := "a7317399f3f857173c6179d44823594f8294678dea9999662e5c625b5a1c7657"
|
||
|
testLine := "Line that thinks that it is log line from docker\n"
|
||
|
msg := &logger.Message{ContainerID: cid, Line: []byte(testLine), Source: "stderr", Timestamp: time.Now().UTC()}
|
||
|
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.ResetTimer()
|
||
|
for i := 0; i < b.N; i++ {
|
||
|
for j := 0; j < 30; j++ {
|
||
|
if err := l.Log(msg); err != nil {
|
||
|
b.Fatal(err)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|