38 lines
660 B
Go
38 lines
660 B
Go
package widgets
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
ui "github.com/cjbassi/gotop/termui"
|
|
"github.com/cjbassi/gotop/utils"
|
|
ps "github.com/shirou/gopsutil/disk"
|
|
)
|
|
|
|
type Disk struct {
|
|
*ui.Gauge
|
|
fs string // which filesystem to get the disk usage of
|
|
interval time.Duration
|
|
}
|
|
|
|
func NewDisk() *Disk {
|
|
d := &Disk{ui.NewGauge(), "/", time.Second * 5}
|
|
d.Label = "Disk Usage"
|
|
|
|
go d.update()
|
|
ticker := time.NewTicker(d.interval)
|
|
go func() {
|
|
for range ticker.C {
|
|
d.update()
|
|
}
|
|
}()
|
|
|
|
return d
|
|
}
|
|
|
|
func (d *Disk) update() {
|
|
disk, _ := ps.Usage(d.fs)
|
|
d.Percent = int(disk.UsedPercent)
|
|
d.Description = fmt.Sprintf(" (%dGB free)", utils.BytesToGB(disk.Free))
|
|
}
|