110 lines
3.0 KiB
Go
110 lines
3.0 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"sort"
|
|
"strings"
|
|
|
|
"github.com/fatih/color"
|
|
"gopkg.in/urfave/cli.v2"
|
|
|
|
"github.com/filecoin-project/sector-storage/storiface"
|
|
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
lcli "github.com/filecoin-project/lotus/cli"
|
|
)
|
|
|
|
var workersCmd = &cli.Command{
|
|
Name: "workers",
|
|
Usage: "interact with workers",
|
|
Subcommands: []*cli.Command{
|
|
workersListCmd,
|
|
},
|
|
}
|
|
|
|
var workersListCmd = &cli.Command{
|
|
Name: "list",
|
|
Usage: "list workers",
|
|
Flags: []cli.Flag{
|
|
&cli.BoolFlag{Name: "color"},
|
|
},
|
|
Action: func(cctx *cli.Context) error {
|
|
color.NoColor = !cctx.Bool("color")
|
|
|
|
nodeApi, closer, err := lcli.GetStorageMinerAPI(cctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer closer()
|
|
|
|
ctx := lcli.ReqContext(cctx)
|
|
|
|
stats, err := nodeApi.WorkerStats(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
type sortableStat struct {
|
|
id uint64
|
|
storiface.WorkerStats
|
|
}
|
|
|
|
st := make([]sortableStat, 0, len(stats))
|
|
for id, stat := range stats {
|
|
st = append(st, sortableStat{id, stat})
|
|
}
|
|
|
|
sort.Slice(st, func(i, j int) bool {
|
|
return st[i].id < st[j].id
|
|
})
|
|
|
|
for _, stat := range st {
|
|
gpuUse := "not "
|
|
gpuCol := color.FgBlue
|
|
if stat.GpuUsed {
|
|
gpuCol = color.FgGreen
|
|
gpuUse = ""
|
|
}
|
|
|
|
|
|
fmt.Printf("Worker %d, host %s\n", stat.id, color.MagentaString(stat.Info.Hostname))
|
|
|
|
var barCols = uint64(64)
|
|
cpuBars := int(stat.CpuUse * barCols / stat.Info.Resources.CPUs)
|
|
cpuBar := strings.Repeat("|", cpuBars) + strings.Repeat(" ", int(barCols) - cpuBars)
|
|
|
|
fmt.Printf("\tCPU: [%s] %d core(s) in use\n", color.GreenString(cpuBar), stat.CpuUse)
|
|
|
|
ramBarsRes := int(stat.Info.Resources.MemReserved*barCols/stat.Info.Resources.MemPhysical)
|
|
ramBarsUsed := int(stat.MemUsedMin*barCols/stat.Info.Resources.MemPhysical)
|
|
ramBar := color.YellowString(strings.Repeat("|", ramBarsRes)) +
|
|
color.GreenString(strings.Repeat("|", ramBarsUsed)) +
|
|
strings.Repeat(" ", int(barCols) - ramBarsUsed - ramBarsRes)
|
|
|
|
vmem := stat.Info.Resources.MemPhysical+stat.Info.Resources.MemSwap
|
|
|
|
vmemBarsRes := int(stat.Info.Resources.MemReserved*barCols/vmem)
|
|
vmemBarsUsed := int(stat.MemUsedMax*barCols/vmem)
|
|
vmemBar := color.YellowString(strings.Repeat("|", vmemBarsRes)) +
|
|
color.GreenString(strings.Repeat("|", vmemBarsUsed)) +
|
|
strings.Repeat(" ", int(barCols) - vmemBarsUsed - vmemBarsRes)
|
|
|
|
fmt.Printf("\tRAM: [%s] %d%% %s/%s\n", ramBar,
|
|
(stat.Info.Resources.MemReserved + stat.MemUsedMin)*100/stat.Info.Resources.MemPhysical,
|
|
types.SizeStr(types.NewInt(stat.Info.Resources.MemReserved + stat.MemUsedMin)),
|
|
types.SizeStr(types.NewInt(stat.Info.Resources.MemPhysical)))
|
|
|
|
fmt.Printf("\tVMEM: [%s] %d%% %s/%s\n", vmemBar,
|
|
(stat.Info.Resources.MemReserved + stat.MemUsedMax)*100/vmem,
|
|
types.SizeStr(types.NewInt(stat.Info.Resources.MemReserved + stat.MemUsedMax)),
|
|
types.SizeStr(types.NewInt(vmem)))
|
|
|
|
for _, gpu := range stat.Info.Resources.GPUs {
|
|
fmt.Printf("\tGPU: %s\n", color.New(gpuCol).Sprintf("%s, %sused", gpu, gpuUse))
|
|
}
|
|
}
|
|
|
|
return nil
|
|
},
|
|
}
|