forked from arne-cl/tile38
/
output.go
42 lines (38 loc) 路 1.03 KB
/
output.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
package controller
import (
"strings"
"time"
"github.com/tidwall/resp"
"github.com/quesurifn/tile38/pkg/server"
)
func (c *Controller) cmdOutput(msg *server.Message) (res resp.Value, err error) {
start := time.Now()
vs := msg.Values[1:]
var arg string
var ok bool
if len(vs) != 0 {
if _, arg, ok = tokenval(vs); !ok || arg == "" {
return server.NOMessage, errInvalidNumberOfArguments
}
// Setting the original message output type will be picked up by the
// server prior to the next command being executed.
switch strings.ToLower(arg) {
default:
return server.NOMessage, errInvalidArgument(arg)
case "json":
msg.OutputType = server.JSON
case "resp":
msg.OutputType = server.RESP
}
return server.OKMessage(msg, start), nil
}
// return the output
switch msg.OutputType {
default:
return server.NOMessage, nil
case server.JSON:
return resp.StringValue(`{"ok":true,"output":"json","elapsed":` + time.Now().Sub(start).String() + `}`), nil
case server.RESP:
return resp.StringValue("resp"), nil
}
}