Add disk.IOCountersForNames function

Operates like disk.IOCounters, but accepts an array of names to limit the results.
shirou · Apr 7, 2017 · 3f35f00 · 3f35f00
1 parent e49a95f
commit 3f35f00
Show file tree

Hide file tree

Showing 7 changed files with 31 additions and 5 deletions.
diff --git a/disk/disk.go b/disk/disk.go
@@ -62,3 +62,7 @@ func (d IOCountersStat) String() string {
 	s, _ := json.Marshal(d)
 	return string(s)
 }
+
+func IOCounters() (map[string]IOCountersStat, error) {
+	return IOCountersForNames([]string{})
+}
diff --git a/disk/disk_darwin.go b/disk/disk_darwin.go
@@ -87,6 +87,10 @@ func Partitions(all bool) ([]PartitionStat, error) {
 	return ret, nil
 }
 
+func IOCountersForNames(names []string) (map[string]IOCountersStat, error) {
+	return nil, common.ErrNotImplementedError
+}
+
 func Getfsstat(buf []Statfs_t, flags int) (n int, err error) {
 	var _p0 unsafe.Pointer
 	var bufsize uintptr

diff --git a/disk/disk_fallback.go b/disk/disk_fallback.go
@@ -4,7 +4,7 @@ package disk
 
 import "github.com/shirou/gopsutil/internal/common"
 
-func IOCounters() (map[string]IOCountersStat, error) {
+func IOCountersForNames(names []string) (map[string]IOCountersStat, error) {
 	return nil, common.ErrNotImplementedError
 }
 

diff --git a/disk/disk_freebsd.go b/disk/disk_freebsd.go
@@ -94,7 +94,7 @@ func Partitions(all bool) ([]PartitionStat, error) {
 	return ret, nil
 }
 
-func IOCounters() (map[string]IOCountersStat, error) {
+func IOCountersForNames(names []string) (map[string]IOCountersStat, error) {
 	// statinfo->devinfo->devstat
 	// /usr/include/devinfo.h
 	ret := make(map[string]IOCountersStat)
@@ -119,6 +119,10 @@ func IOCounters() (map[string]IOCountersStat, error) {
 		un := strconv.Itoa(int(d.Unit_number))
 		name := common.IntToString(d.Device_name[:]) + un
 
+		if len(names) > 0 && !common.StringsHas(names, name) {
+			continue
+		}
+
 		ds := IOCountersStat{
 			ReadCount:  d.Operations[DEVSTAT_READ],
 			WriteCount: d.Operations[DEVSTAT_WRITE],

diff --git a/disk/disk_linux.go b/disk/disk_linux.go
@@ -272,7 +272,7 @@ func getFileSystems() ([]string, error) {
 	return ret, nil
 }
 
-func IOCounters() (map[string]IOCountersStat, error) {
+func IOCountersForNames(names []string) (map[string]IOCountersStat, error) {
 	filename := common.HostProc("diskstats")
 	lines, err := common.ReadLines(filename)
 	if err != nil {
@@ -288,6 +288,11 @@ func IOCounters() (map[string]IOCountersStat, error) {
 			continue
 		}
 		name := fields[2]
+
+		if len(names) > 0 && !common.StringsHas(names, name) {
+			continue
+		}
+
 		reads, err := strconv.ParseUint((fields[3]), 10, 64)
 		if err != nil {
 			return ret, err

diff --git a/disk/disk_openbsd.go b/disk/disk_openbsd.go
@@ -63,7 +63,7 @@ func Partitions(all bool) ([]PartitionStat, error) {
 	return ret, nil
 }
 
-func IOCounters() (map[string]IOCountersStat, error) {
+func IOCountersForNames(names []string) (map[string]IOCountersStat, error) {
 	ret := make(map[string]IOCountersStat)
 
 	r, err := syscall.Sysctl("hw.diskstats")
@@ -84,6 +84,10 @@ func IOCounters() (map[string]IOCountersStat, error) {
 		}
 		name := common.IntToString(d.Name[:])
 
+		if len(names) > 0 && !common.StringsHas(names, name) {
+			continue
+		}
+
 		ds := IOCountersStat{
 			ReadCount:  d.Rxfer,
 			WriteCount: d.Wxfer,

diff --git a/disk/disk_windows.go b/disk/disk_windows.go
@@ -129,7 +129,7 @@ func Partitions(all bool) ([]PartitionStat, error) {
 	return ret, nil
 }
 
-func IOCounters() (map[string]IOCountersStat, error) {
+func IOCountersForNames(names []string) (map[string]IOCountersStat, error) {
 	ret := make(map[string]IOCountersStat, 0)
 	var dst []Win32_PerfFormattedData
 
@@ -141,6 +141,11 @@ func IOCounters() (map[string]IOCountersStat, error) {
 		if len(d.Name) > 3 { // not get _Total or Harddrive
 			continue
 		}
+
+		if len(names) > 0 && !common.StringsHas(names, name) {
+			continue
+		}
+
 		ret[d.Name] = IOCountersStat{
 			Name:       d.Name,
 			ReadCount:  uint64(d.AvgDiskReadQueueLength),