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
|
package device
import "strings"
// Device struct
type Device struct {
Name string
MountPoint string
Fstype string
Size int64
Free int64
}
// GetUsage returns used size of device
func (d Device) GetUsage() int64 {
return d.Size - d.Free
}
// DevicesInfoGetter is type for GetDevicesInfo function
type DevicesInfoGetter interface {
GetMounts() (Devices, error)
GetDevicesInfo() (Devices, error)
}
// Devices if slice of Device items
type Devices []*Device
// ByUsedSize sorts devices by used size
type ByUsedSize Devices
func (f ByUsedSize) Len() int { return len(f) }
func (f ByUsedSize) Swap(i, j int) { f[i], f[j] = f[j], f[i] }
func (f ByUsedSize) Less(i, j int) bool {
return f[i].GetUsage() < f[j].GetUsage()
}
// ByName sorts devices by device name
type ByName Devices
func (f ByName) Len() int { return len(f) }
func (f ByName) Swap(i, j int) { f[i], f[j] = f[j], f[i] }
func (f ByName) Less(i, j int) bool {
return f[i].Name < f[j].Name
}
// GetNestedMountpointsPaths returns paths of nested mount points
func GetNestedMountpointsPaths(path string, mounts Devices) []string {
paths := make([]string, 0, len(mounts))
for _, mount := range mounts {
if strings.HasPrefix(mount.MountPoint, path) && mount.MountPoint != path {
paths = append(paths, mount.MountPoint)
}
}
return paths
}
|
