-
Notifications
You must be signed in to change notification settings - Fork 0
/
logger-fifo-linux.go
75 lines (64 loc) · 1.44 KB
/
logger-fifo-linux.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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
//go:build linux
package commonlog
import (
"bufio"
"os"
"path/filepath"
"syscall"
"github.com/segmentio/ksuid"
)
//
// LoggerFIFO
//
// A Linux FIFO file that forwards all lines written to it to a [Logger].
type LoggerFIFO struct {
Path string
Log Logger
Level Level
}
func NewLoggerFIFO(prefix string, log Logger, level Level) *LoggerFIFO {
path := filepath.Join(os.TempDir(), prefix+ksuid.New().String())
return &LoggerFIFO{
Path: path,
Log: NewKeyValueLogger(log, "fifo", path),
Level: level,
}
}
func (self *LoggerFIFO) Start() error {
if err := self.create(); err == nil {
go self.start()
return nil
} else {
return err
}
}
func (self *LoggerFIFO) create() error {
if err := os.Remove(self.Path); err != nil {
if !os.IsNotExist(err) {
return err
}
}
self.Log.Debug("creating logger FIFO")
return syscall.Mkfifo(self.Path, 0600)
}
func (self *LoggerFIFO) start() {
// Note: os.Open will block until the FIFO will be opened for write
if file, err := os.Open(self.Path); err == nil {
scanner := bufio.NewScanner(file)
for scanner.Scan() {
self.Log.Log(self.Level, 0, scanner.Text())
}
if err := scanner.Err(); err != nil {
self.Log.Error(err.Error())
}
self.Log.Debug("closing logger FIFO")
if err := file.Close(); err != nil {
self.Log.Error(err.Error())
}
if err := os.Remove(self.Path); err != nil {
self.Log.Error(err.Error())
}
} else {
self.Log.Critical(err.Error())
}
}