mirror of
https://github.com/hamburghammer/gstat.git
synced 2024-06-17 08:46:01 +02:00
Augusto Dwenger J.
f004c93773
This implementation still isn't perfect. One of the encoutered problems is that it does not omit Memory sturcts that are technicly empty. Resulting in JSON with empty objects. This should be a problem for other parsers but it doesn't look good.
52 lines
1 KiB
Go
52 lines
1 KiB
Go
package commands
|
|
|
|
import (
|
|
"encoding/json"
|
|
|
|
"github.com/hamburghammer/gstat/args"
|
|
"github.com/shirou/gopsutil/mem"
|
|
)
|
|
|
|
// Mem holds the memory usage for the json transformation
|
|
type Mem struct {
|
|
ReadVirtualMemoryStat func() (*mem.VirtualMemoryStat, error)
|
|
}
|
|
|
|
// NewMem is a constructor for the Mem struct
|
|
func NewMem() Mem {
|
|
return Mem{mem.VirtualMemory}
|
|
}
|
|
|
|
// Exec gets the mem value and maps it to the executiondata struct
|
|
func (m Mem) Exec(args args.Arguments) ([]byte, error) {
|
|
usage, err := m.PureExec(args)
|
|
if err != nil {
|
|
return []byte{}, err
|
|
}
|
|
|
|
data := struct {
|
|
Mem Memory `json:"mem"`
|
|
}{usage}
|
|
return json.Marshal(data)
|
|
}
|
|
|
|
func (m Mem) PureExec(args args.Arguments) (Memory, error) {
|
|
if !args.Mem {
|
|
return Memory{}, nil
|
|
}
|
|
|
|
mem, err := m.ReadVirtualMemoryStat()
|
|
if err != nil {
|
|
return Memory{}, err
|
|
}
|
|
|
|
memory := Memory{Used: bytesToMegaByte(mem.Used), Total: bytesToMegaByte(mem.Total)}
|
|
return memory, nil
|
|
}
|
|
|
|
func bytesToMegaByte(bytes uint64) uint64 {
|
|
kb := bytes / 1024
|
|
mb := kb / 1024
|
|
return mb
|
|
}
|