57 lines
1.0 KiB
Go
57 lines
1.0 KiB
Go
|
package widgets
|
||
|
|
|
||
|
|
import (
|
||
|
|
"time"
|
||
|
|
|
||
|
|
ui "github.com/cjbassi/gotop/termui"
|
||
|
|
ps "github.com/shirou/gopsutil/net"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Net struct {
|
||
|
|
*ui.Sparklines
|
||
|
|
interval time.Duration
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewNet() *Net {
|
||
|
|
recv := ui.NewSparkline()
|
||
|
|
recv.Title = "Receiving"
|
||
|
|
recv.Data = []int{0}
|
||
|
|
recv.Total = 0
|
||
|
|
|
||
|
|
sent := ui.NewSparkline()
|
||
|
|
sent.Title = "Transfering"
|
||
|
|
sent.Data = []int{0}
|
||
|
|
sent.Total = 0
|
||
|
|
|
||
|
|
spark := ui.NewSparklines(recv, sent)
|
||
|
|
n := &Net{spark, time.Second}
|
||
|
|
n.Label = "Network Usage"
|
||
|
|
|
||
|
|
go n.update()
|
||
|
|
ticker := time.NewTicker(n.interval)
|
||
|
|
go func() {
|
||
|
|
for range ticker.C {
|
||
|
|
n.update()
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
|
||
|
|
return n
|
||
|
|
}
|
||
|
|
|
||
|
|
func (n *Net) update() {
|
||
|
|
interfaces, _ := ps.IOCounters(false)
|
||
|
|
recv := int(interfaces[0].BytesRecv)
|
||
|
|
sent := int(interfaces[0].BytesSent)
|
||
|
|
|
||
|
|
if n.Lines[0].Total != 0 { // if this isn't the first update
|
||
|
|
curRecv := recv - n.Lines[0].Total
|
||
|
|
curSent := sent - n.Lines[1].Total
|
||
|
|
|
||
|
|
n.Lines[0].Data = append(n.Lines[0].Data, curRecv)
|
||
|
|
n.Lines[1].Data = append(n.Lines[1].Data, curSent)
|
||
|
|
}
|
||
|
|
|
||
|
|
n.Lines[0].Total = recv
|
||
|
|
n.Lines[1].Total = sent
|
||
|
|
}
|