forked from heroku/lumbermill
-
Notifications
You must be signed in to change notification settings - Fork 0
/
lpxgen.go
101 lines (86 loc) · 1.89 KB
/
lpxgen.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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
package lpxgen
import (
"bytes"
"fmt"
"math/rand"
"net/http"
)
const (
TimeFormat = "2006-01-02T15:04:05.000000+00:00"
)
type Log interface {
PrivalVersion() string
Time() string
Hostname() string
Name() string
Procid() string
Msgid() string
Msg() string
String() string
}
type LPXGenerator struct {
Mincount int
Maxcount int
Log Log
}
func NewGenerator(minBatch, maxBatch int, loggen Log) *LPXGenerator {
return &LPXGenerator{
Mincount: minBatch,
Maxcount: maxBatch,
Log: loggen,
}
}
func (g *LPXGenerator) Generate(url string) *http.Request {
var body bytes.Buffer
batchSize := g.Mincount + rand.Intn(g.Maxcount-g.Mincount)
for i := 0; i < batchSize; i++ {
body.WriteString(g.Log.String())
}
request, _ := http.NewRequest("POST", url, bytes.NewReader(body.Bytes()))
request.Header.Add("Content-Length", string(body.Len()))
request.Header.Add("Content-Type", "application/logplex-1")
request.Header.Add("Logplex-Msg-Count", string(batchSize))
request.Header.Add("Logplex-Frame-Id", RandomFrameId())
request.Header.Add("Logplex-Drain-Token", RandomDrainToken())
return request
}
func FormatSyslog(l Log) string {
return fmt.Sprintf("%s %s %s %s %s %s %s\n",
l.PrivalVersion(),
l.Time(),
l.Hostname(),
l.Name(),
l.Procid(),
l.Msgid(),
l.Msg())
}
var hexchars = []byte("0123456789abcdef")
func randomHexString(i int) string {
x := make([]byte, i)
for i > 0 {
i--
x[i] = hexchars[rand.Intn(16)]
}
return string(x)
}
func UUID4() string {
return fmt.Sprintf("%s-%s-%s-%s-%s",
randomHexString(8),
randomHexString(4),
randomHexString(4),
randomHexString(4),
randomHexString(12))
}
func RandomFrameId() string {
return randomHexString(32)
}
func RandomDrainToken() string {
return fmt.Sprintf("t.%s", UUID4())
}
func RandomIPv4() string {
return fmt.Sprintf("%d.%d.%d.%d",
rand.Intn(255),
rand.Intn(255),
rand.Intn(255),
rand.Intn(255))
}