-
Notifications
You must be signed in to change notification settings - Fork 1
/
fluent.go
79 lines (66 loc) · 1.42 KB
/
fluent.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
76
77
78
79
package log
import (
"fmt"
"strconv"
"strings"
"time"
"github.com/94peter/sterna/util"
"github.com/fluent/fluent-logger-golang/fluent"
)
const LogTargetFluent = "fluent"
type fluentLog struct {
Host string `yaml:"host"`
Port string `yaml:"port"`
Timezone string `yaml:"timezone"`
service string
}
func (rl *fluentLog) getClient() *fluent.Fluent {
host := rl.Host
port := rl.Port
p, err := strconv.Atoi(port)
if err != nil {
panic(err)
}
logger, err := fluent.New(fluent.Config{
FluentHost: host,
FluentPort: p,
})
if err != nil {
return nil
}
return logger
}
var fluentTag = ""
func (rl *fluentLog) getTag() string {
if fluentTag == "" {
fluentTag = rl.service + "-log"
}
return fluentTag
}
func (rl *fluentLog) Write(p []byte) (n int, err error) {
splitMsg := strings.SplitN(string(p), " ", 5)
msg := map[string]interface{}{
"service": rl.service,
"severity": splitMsg[0],
"timeUnix": getTimeUnix(splitMsg[1], splitMsg[2], rl.Timezone),
"filePath": splitMsg[3],
"message": strings.TrimSuffix(splitMsg[4], "\n"),
}
logger := rl.getClient()
if logger == nil {
return 0, nil
}
defer logger.Close()
err = logger.Post(rl.getTag(), msg)
if err != nil {
fmt.Println(err)
}
return 0, nil
}
func getTimeUnix(d, t, z string) int64 {
tt, err := time.Parse("2006/01/02 15:04:05.000000 -0700", util.StrAppend(d, " ", t, " ", z))
if err != nil {
return 0
}
return tt.Unix()
}