Make ps follow new standard

This commit is contained in:
Ian Adam Naval 2014-09-16 21:43:43 -04:00
parent 1d57896123
commit fb8165fa80
3 changed files with 37 additions and 9 deletions

View File

@ -4,9 +4,10 @@ package jsh
import ( import (
"encoding/json" "encoding/json"
"fmt"
) )
type JshOutput struct { type JshFrame struct {
StdOut interface{} StdOut interface{}
StdErr interface{} StdErr interface{}
} }
@ -22,8 +23,8 @@ const (
TB Unit = 2 ^ 40 TB Unit = 2 ^ 40
) )
// Converts a Jshoutput into a JSON string // Converts a JshFrame into a JSON string
func (j JshOutput) ToJson() *string { func (j JshFrame) ToJson() *string {
jsonOut, err := json.Marshal(j) jsonOut, err := json.Marshal(j)
if err != nil { if err != nil {
panic(err) panic(err)
@ -31,3 +32,25 @@ func (j JshOutput) ToJson() *string {
jsonString := string(jsonOut) jsonString := string(jsonOut)
return &jsonString return &jsonString
} }
// goroutine for outputing frames. Pass it a channel of pointers to JshFrames,
// and it will send "true" to the done channel once you close the queue channel.
func OutputFrames(queue chan *JshFrame, done chan bool) {
fmt.Printf("[")
isFirst := true
for {
frame, more := <-queue
if more {
if !isFirst {
fmt.Printf(",")
} else {
isFirst = false
}
fmt.Printf(*frame.ToJson())
} else {
fmt.Printf("]\n")
done <- true
return
}
}
}

View File

@ -53,10 +53,10 @@ func parseLine(line string) (string, jsh.MemStat, error) {
return key, jsh.MemStat{val, unit}, nil return key, jsh.MemStat{val, unit}, nil
} }
func parseMemInfo() jsh.JshOutput { func parseMemInfo() jsh.JshFrame {
file, err := os.Open("/proc/meminfo") file, err := os.Open("/proc/meminfo")
if err != nil { if err != nil {
return jsh.JshOutput{[]string{}, []error{err}} return jsh.JshFrame{[]string{}, []error{err}}
} }
defer file.Close() defer file.Close()
@ -72,7 +72,7 @@ func parseMemInfo() jsh.JshOutput {
memInfo[key] = val memInfo[key] = val
} }
finalOut := jsh.JshOutput{memInfo, errors} finalOut := jsh.JshFrame{memInfo, errors}
return finalOut return finalOut
} }

View File

@ -26,10 +26,15 @@ func PsOutputToProcesses(out string) *[]jsh.Process {
func runJsonMode() { func runJsonMode() {
// Run procps-ng "ps" with full output // Run procps-ng "ps" with full output
psOut := string(*jsh.FallbackWithArgs("/usr/bin/ps", []string{"auxww"})) psOut := string(*jsh.FallbackWithArgs("/usr/bin/ps", []string{"auxww"}))
processesPtr := PsOutputToProcesses(psOut) processesPtr := PsOutputToProcesses(psOut)
finalOut := jsh.JshOutput{*processesPtr, []string{}} frame := jsh.JshFrame{*processesPtr, []string{}}
fmt.Printf("%s", *finalOut.ToJson())
queue := make(chan *jsh.JshFrame)
done := make(chan bool)
go jsh.OutputFrames(queue, done)
queue <- &frame
close(queue)
<-done
} }
func main() { func main() {