-
Notifications
You must be signed in to change notification settings - Fork 0
/
fallover_writer.go
56 lines (48 loc) · 1.15 KB
/
fallover_writer.go
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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
package log
// NewFailoverWriter create a failover writer
func NewFailoverWriter(w Writer, bufSize int) *FailoverWriter {
fw := &FailoverWriter{writer: w}
fw.evtbuf = NewEventBuffer(bufSize)
return fw
}
// FailoverWriter wraps a log writer, cache log event if underlying log writer write failed.
// Resend the cached log events to the underlying log writer when the next log event come.
type FailoverWriter struct {
writer Writer
evtbuf *EventBuffer
}
// Write write event to underlying writer
func (fw *FailoverWriter) Write(le *Event) error {
err := fw.flush()
if err == nil {
err = fw.writer.Write(le)
if err != nil {
perror(err)
}
}
if err != nil {
fw.evtbuf.Push(le)
}
return nil
}
// flush flush buffered event
func (fw *FailoverWriter) flush() error {
for le, ok := fw.evtbuf.Peek(); ok; le, ok = fw.evtbuf.Peek() {
if err := fw.writer.Write(le); err != nil {
perror(err)
return err
}
fw.evtbuf.Poll()
}
return nil
}
// Flush implementing method. empty.
func (fw *FailoverWriter) Flush() {
fw.flush()
fw.writer.Flush()
}
// Close implementing method. empty.
func (fw *FailoverWriter) Close() {
fw.flush()
fw.writer.Close()
}