Dropped SkipInodeUsage option as "drop" achieves the same results.

Fixed a bug in restricting Disk reporting to specific mountpoints
Added tests for the Disk.Mountpoints option
Fixed minor bug in usage of assert for the cpu tests where expected and actual values were swapped.
This commit is contained in:
subhachandrachandra 2015-10-08 14:17:04 -07:00
parent 4f6526e1a5
commit cf1dcfe37c
2 changed files with 69 additions and 34 deletions

View File

@ -10,7 +10,6 @@ type DiskStats struct {
ps PS ps PS
Mountpoints []string Mountpoints []string
SkipInodeUsage bool
} }
func (_ *DiskStats) Description() string { func (_ *DiskStats) Description() string {
@ -18,12 +17,10 @@ func (_ *DiskStats) Description() string {
} }
var diskSampleConfig = ` var diskSampleConfig = `
# By default, telegraf gather stats for all mountpoints and for inodes. # By default, telegraf gather stats for all mountpoints.
# Setting mountpoints will restrict the stats to the specified ones. # Setting mountpoints will restrict the stats to the specified ones.
# mountpoints. # mountpoints.
# Mountpoints=["/"] # Mountpoints=["/"]
# Setting SkipInodeUsage will skip the reporting of inode stats.
# SkipInodeUsage=true
` `
func (_ *DiskStats) SampleConfig() string { func (_ *DiskStats) SampleConfig() string {
@ -36,14 +33,18 @@ func (s *DiskStats) Gather(acc plugins.Accumulator) error {
return fmt.Errorf("error getting disk usage info: %s", err) return fmt.Errorf("error getting disk usage info: %s", err)
} }
var restrictMpoints bool
mPoints := make(map[string]bool) mPoints := make(map[string]bool)
for _, mp := range s.Mountpoints { if len(s.Mountpoints) != 0 {
mPoints[mp] = true restrictMpoints = true
for _, mp := range s.Mountpoints {
mPoints[mp] = true
}
} }
for _, du := range disks { for _, du := range disks {
_, member := mPoints[ du.Path ] _, member := mPoints[du.Path]
if !member { if restrictMpoints && !member {
continue continue
} }
tags := map[string]string{ tags := map[string]string{
@ -53,11 +54,9 @@ func (s *DiskStats) Gather(acc plugins.Accumulator) error {
acc.Add("total", du.Total, tags) acc.Add("total", du.Total, tags)
acc.Add("free", du.Free, tags) acc.Add("free", du.Free, tags)
acc.Add("used", du.Total-du.Free, tags) acc.Add("used", du.Total-du.Free, tags)
if !s.SkipInodeUsage { acc.Add("inodes_total", du.InodesTotal, tags)
acc.Add("inodes_total", du.InodesTotal, tags) acc.Add("inodes_free", du.InodesFree, tags)
acc.Add("inodes_free", du.InodesFree, tags) acc.Add("inodes_used", du.InodesTotal-du.InodesFree, tags)
acc.Add("inodes_used", du.InodesTotal-du.InodesFree, tags)
}
} }
return nil return nil

View File

@ -51,16 +51,26 @@ func TestSystemStats_GenerateStats(t *testing.T) {
mps.On("CPUTimes").Return([]cpu.CPUTimesStat{cts}, nil) mps.On("CPUTimes").Return([]cpu.CPUTimesStat{cts}, nil)
du := &disk.DiskUsageStat{ du := []*disk.DiskUsageStat{
Path: "/", {
Fstype: "ext4", Path: "/",
Total: 128, Fstype: "ext4",
Free: 23, Total: 128,
InodesTotal: 1234, Free: 23,
InodesFree: 234, InodesTotal: 1234,
InodesFree: 234,
},
{
Path: "/home",
Fstype: "ext4",
Total: 256,
Free: 46,
InodesTotal: 2468,
InodesFree: 468,
},
} }
mps.On("DiskUsage").Return([]*disk.DiskUsageStat{du}, nil) mps.On("DiskUsage").Return(du, nil)
diskio := disk.DiskIOCountersStat{ diskio := disk.DiskIOCountersStat{
ReadCount: 888, ReadCount: 888,
@ -128,7 +138,7 @@ func TestSystemStats_GenerateStats(t *testing.T) {
numCPUPoints := len(acc.Points) - preCPUPoints numCPUPoints := len(acc.Points) - preCPUPoints
expectedCPUPoints := 10 expectedCPUPoints := 10
assert.Equal(t, numCPUPoints, expectedCPUPoints) assert.Equal(t, expectedCPUPoints, numCPUPoints)
// Computed values are checked with delta > 0 becasue of floating point arithmatic // Computed values are checked with delta > 0 becasue of floating point arithmatic
// imprecision // imprecision
@ -153,7 +163,7 @@ func TestSystemStats_GenerateStats(t *testing.T) {
numCPUPoints = len(acc.Points) - (preCPUPoints + numCPUPoints) numCPUPoints = len(acc.Points) - (preCPUPoints + numCPUPoints)
expectedCPUPoints = 20 expectedCPUPoints = 20
assert.Equal(t, numCPUPoints, expectedCPUPoints) assert.Equal(t, expectedCPUPoints, numCPUPoints)
assertContainsTaggedFloat(t, acc, "time_user", 11.4, 0, cputags) assertContainsTaggedFloat(t, acc, "time_user", 11.4, 0, cputags)
assertContainsTaggedFloat(t, acc, "time_system", 10.9, 0, cputags) assertContainsTaggedFloat(t, acc, "time_system", 10.9, 0, cputags)
@ -177,20 +187,46 @@ func TestSystemStats_GenerateStats(t *testing.T) {
assertContainsTaggedFloat(t, acc, "usage_guest", 4.8, 0.0005, cputags) assertContainsTaggedFloat(t, acc, "usage_guest", 4.8, 0.0005, cputags)
assertContainsTaggedFloat(t, acc, "usage_guest_nice", 2.2, 0.0005, cputags) assertContainsTaggedFloat(t, acc, "usage_guest_nice", 2.2, 0.0005, cputags)
err = (&DiskStats{&mps}).Gather(&acc) preDiskPoints := len(acc.Points)
err = (&DiskStats{ps: &mps}).Gather(&acc)
require.NoError(t, err) require.NoError(t, err)
tags := map[string]string{ numDiskPoints := len(acc.Points) - preDiskPoints
expectedAllDiskPoints := 12
assert.Equal(t, expectedAllDiskPoints, numDiskPoints)
tags1 := map[string]string{
"path": "/", "path": "/",
"fstype": "ext4", "fstype": "ext4",
} }
tags2 := map[string]string{
"path": "/home",
"fstype": "ext4",
}
assert.True(t, acc.CheckTaggedValue("total", uint64(128), tags)) assert.True(t, acc.CheckTaggedValue("total", uint64(128), tags1))
assert.True(t, acc.CheckTaggedValue("used", uint64(105), tags)) assert.True(t, acc.CheckTaggedValue("used", uint64(105), tags1))
assert.True(t, acc.CheckTaggedValue("free", uint64(23), tags)) assert.True(t, acc.CheckTaggedValue("free", uint64(23), tags1))
assert.True(t, acc.CheckTaggedValue("inodes_total", uint64(1234), tags)) assert.True(t, acc.CheckTaggedValue("inodes_total", uint64(1234), tags1))
assert.True(t, acc.CheckTaggedValue("inodes_free", uint64(234), tags)) assert.True(t, acc.CheckTaggedValue("inodes_free", uint64(234), tags1))
assert.True(t, acc.CheckTaggedValue("inodes_used", uint64(1000), tags)) assert.True(t, acc.CheckTaggedValue("inodes_used", uint64(1000), tags1))
assert.True(t, acc.CheckTaggedValue("total", uint64(256), tags2))
assert.True(t, acc.CheckTaggedValue("used", uint64(210), tags2))
assert.True(t, acc.CheckTaggedValue("free", uint64(46), tags2))
assert.True(t, acc.CheckTaggedValue("inodes_total", uint64(2468), tags2))
assert.True(t, acc.CheckTaggedValue("inodes_free", uint64(468), tags2))
assert.True(t, acc.CheckTaggedValue("inodes_used", uint64(2000), tags2))
// We expect 6 more DiskPoints to show up with an explicit match on "/"
// and /home not matching the /dev in Mountpoints
err = (&DiskStats{ps: &mps, Mountpoints: []string{"/", "/dev"}}).Gather(&acc)
assert.Equal(t, preDiskPoints+expectedAllDiskPoints+6, len(acc.Points))
// We should see all the diskpoints as Mountpoints includes both
// / and /home
err = (&DiskStats{ps: &mps, Mountpoints: []string{"/", "/home"}}).Gather(&acc)
assert.Equal(t, preDiskPoints+2*expectedAllDiskPoints+6, len(acc.Points))
err = (&NetIOStats{ps: &mps, skipChecks: true}).Gather(&acc) err = (&NetIOStats{ps: &mps, skipChecks: true}).Gather(&acc)
require.NoError(t, err) require.NoError(t, err)