42 lines
716 B
Go
42 lines
716 B
Go
package widgets
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
ui "github.com/cjbassi/gotop/termui"
|
|
"github.com/cjbassi/gotop/utils"
|
|
psDisk "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{
|
|
Gauge: ui.NewGauge(),
|
|
fs: "/",
|
|
interval: 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() {
|
|
usage, _ := psDisk.Usage(d.fs)
|
|
d.Percent = int(usage.UsedPercent)
|
|
d.Description = fmt.Sprintf(" (%dGB free)", int(utils.BytesToGB(usage.Free)))
|
|
}
|