-
Notifications
You must be signed in to change notification settings - Fork 113
/
devops_diskio.go
68 lines (55 loc) · 1.88 KB
/
devops_diskio.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
package devops
import (
"fmt"
. "github.com/influxdata/influxdb-comparisons/bulk_data_gen/common"
"math/rand"
"time"
)
var (
DiskIOByteString = []byte("diskio") // heap optimization
SerialByteString = []byte("serial")
DiskIOFields = []LabeledDistributionMaker{
{[]byte("reads"), func() Distribution { return MWD(ND(50, 1), 0) }},
{[]byte("writes"), func() Distribution { return MWD(ND(50, 1), 0) }},
{[]byte("read_bytes"), func() Distribution { return MWD(ND(100, 1), 0) }},
{[]byte("write_bytes"), func() Distribution { return MWD(ND(100, 1), 0) }},
{[]byte("read_time"), func() Distribution { return MWD(ND(5, 1), 0) }},
{[]byte("write_time"), func() Distribution { return MWD(ND(5, 1), 0) }},
{[]byte("io_time"), func() Distribution { return MWD(ND(5, 1), 0) }},
}
)
type DiskIOMeasurement struct {
timestamp time.Time
serial []byte
distributions []Distribution
}
func NewDiskIOMeasurement(start time.Time) *DiskIOMeasurement {
distributions := make([]Distribution, len(DiskIOFields))
for i := range DiskIOFields {
distributions[i] = DiskIOFields[i].DistributionMaker()
}
serial := []byte(fmt.Sprintf("%03d-%03d-%03d", rand.Intn(1000), rand.Intn(1000), rand.Intn(1000)))
if Config != nil { // partial override from external config
serial = Config.GetTagBytesValue(DiskIOByteString, SerialByteString, true, serial)
}
return &DiskIOMeasurement{
serial: serial,
timestamp: start,
distributions: distributions,
}
}
func (m *DiskIOMeasurement) Tick(d time.Duration) {
m.timestamp = m.timestamp.Add(d)
for i := range m.distributions {
m.distributions[i].Advance()
}
}
func (m *DiskIOMeasurement) ToPoint(p *Point) bool {
p.SetMeasurementName(DiskIOByteString)
p.SetTimestamp(&m.timestamp)
p.AppendTag(SerialByteString, m.serial)
for i := range m.distributions {
p.AppendField(DiskIOFields[i].Label, int64(m.distributions[i].Get()))
}
return true
}