mirror of
https://github.com/neilotoole/sq.git
synced 2024-12-24 08:36:43 +03:00
81f631e135
* Renamed source.Type to source.DriverType for clarity * More renaming wrt source.DriverType * Renamed output.Formatting to output.Printing
60 lines
1.3 KiB
Go
60 lines
1.3 KiB
Go
package jsonw
|
|
|
|
import (
|
|
"io"
|
|
"time"
|
|
|
|
"github.com/neilotoole/sq/cli/output"
|
|
"github.com/neilotoole/sq/cli/output/jsonw/internal"
|
|
jcolorenc "github.com/neilotoole/sq/cli/output/jsonw/internal/jcolorenc"
|
|
"github.com/neilotoole/sq/libsq/core/errz"
|
|
"github.com/neilotoole/sq/libsq/source"
|
|
)
|
|
|
|
var _ output.PingWriter = (*pingWriter)(nil)
|
|
|
|
// NewPingWriter returns JSON impl of output.PingWriter.
|
|
func NewPingWriter(out io.Writer, pr *output.Printing) output.PingWriter {
|
|
return &pingWriter{out: out, pr: pr}
|
|
}
|
|
|
|
type pingWriter struct {
|
|
out io.Writer
|
|
pr *output.Printing
|
|
}
|
|
|
|
// Open implements output.PingWriter.
|
|
func (p pingWriter) Open(_ []*source.Source) error {
|
|
return nil
|
|
}
|
|
|
|
// Result implements output.PingWriter.
|
|
func (p pingWriter) Result(src *source.Source, d time.Duration, err error) error {
|
|
r := struct {
|
|
*source.Source
|
|
Pong bool `json:"pong"`
|
|
Duration time.Duration `json:"duration"`
|
|
Error string `json:"error,omitempty"`
|
|
}{
|
|
Source: src,
|
|
Pong: err == nil,
|
|
Duration: d,
|
|
}
|
|
|
|
if err != nil {
|
|
r.Error = err.Error()
|
|
}
|
|
|
|
enc := jcolorenc.NewEncoder(p.out)
|
|
enc.SetColors(internal.NewColors(p.pr))
|
|
enc.SetEscapeHTML(false)
|
|
enc.SetIndent("", " ")
|
|
|
|
return errz.Err(enc.Encode(r))
|
|
}
|
|
|
|
// Close implements output.PingWriter.
|
|
func (p pingWriter) Close() error {
|
|
return nil
|
|
}
|