tracing/shivizlog.go at main · DistributedClocks/tracing · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
package tracing

import (
	"bytes"
	"io"
	"strconv"
	"strings"
)

var header = "(?<host>\\S*) (?<clock>{.*})\\n(?<event>.*)"

type shivizLogger struct {
	w io.Writer
}

func newShivizLogger(w io.Writer) (*shivizLogger, error) {
	if _, err := w.Write([]byte(header + "\n\n")); err != nil {
		return nil, err
	}
	return &shivizLogger{w: w}, nil
}

func (s *shivizLogger) log(tRecord TraceRecord) error {
	var buffer bytes.Buffer
	line1 := []string{tRecord.TracerIdentity, tRecord.VectorClock.ReturnVCString()}
	if _, err := buffer.WriteString(strings.Join(line1, " ") + "\n"); err != nil {
		return err
	}

	line2 := []string{strconv.FormatUint(tRecord.TraceID, 10), tRecord.Tag, string(tRecord.Body)}
	if _, err := buffer.WriteString(strings.Join(line2, " ") + "\n"); err != nil {
		return err
	}

	if _, err := s.w.Write(buffer.Bytes()); err != nil {
		return err
	}
	return nil
}