hcq/influx-cli
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: port stacks command from influxdb (#168)

Browse Source
This commit is contained in:
Daniel Moran
2021-06-30 17:21:50 -04:00
committed by GitHub
parent 8126bd8397
commit 40fc1845e9
8 changed files with 770 additions and 74 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
clients/export/export.go
View File

@ -6,6 +6,7 @@ import (
"github.com/influxdata/influx-cli/v2/api"
"github.com/influxdata/influx-cli/v2/clients"
"github.com/influxdata/influx-cli/v2/pkg/template"
)
type Client struct {
@ -15,7 +16,7 @@ type Client struct {
}
type Params struct {
OutParams
template.OutParams
StackId string
IdsPerType map[string][]string
@ -52,14 +53,14 @@ func (c Client) Export(ctx context.Context, params *Params) error {
if err != nil {
return fmt.Errorf("failed to export template: %w", err)
}
if err := params.OutParams.writeTemplate(tmpl); err != nil {
if err := params.OutParams.WriteTemplate(tmpl); err != nil {
return fmt.Errorf("failed to write exported template: %w", err)
}
return nil
}
type AllParams struct {
OutParams
template.OutParams
OrgId string
OrgName string
@ -105,14 +106,14 @@ func (c Client) ExportAll(ctx context.Context, params *AllParams) error {
if err != nil {
return fmt.Errorf("failed to export template: %w", err)
}
if err := params.OutParams.writeTemplate(tmpl); err != nil {
if err := params.OutParams.WriteTemplate(tmpl); err != nil {
return fmt.Errorf("failed to write exported template: %w", err)
}
return nil
}
type StackParams struct {
OutParams
template.OutParams
StackId string
}
@ -127,7 +128,7 @@ func (c Client) ExportStack(ctx context.Context, params *StackParams) error {
if err != nil {
return fmt.Errorf("failed to export stack %q: %w", params.StackId, err)
}
if err := params.OutParams.writeTemplate(tmpl); err != nil {
if err := params.OutParams.WriteTemplate(tmpl); err != nil {
return fmt.Errorf("failed to write exported template: %w", err)
}

41
clients/export/out.go
View File

@ -1,41 +0,0 @@
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
}