42 lines
719 B
Go
42 lines
719 B
Go
package export
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
|
|
"github.com/influxdata/influx-cli/v2/api"
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
type OutEncoding int
|
|
|
|
const (
|
|
YamlEncoding OutEncoding = iota
|
|
JsonEncoding
|
|
)
|
|
|
|
type OutParams struct {
|
|
Out io.Writer
|
|
Encoding OutEncoding
|
|
}
|
|
|
|
func (o OutParams) writeTemplate(template []api.TemplateEntry) error {
|
|
switch o.Encoding {
|
|
case JsonEncoding:
|
|
enc := json.NewEncoder(o.Out)
|
|
enc.SetIndent("", "\t")
|
|
return enc.Encode(template)
|
|
case YamlEncoding:
|
|
enc := yaml.NewEncoder(o.Out)
|
|
for _, entry := range template {
|
|
if err := enc.Encode(entry); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
default:
|
|
return fmt.Errorf("encoding %q is not recognized", o.Encoding)
|
|
}
|
|
return nil
|
|
}
|