68 lines
1.5 KiB
Go
68 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/influxdata/flux"
|
|
"github.com/influxdata/flux/repl"
|
|
_ "github.com/influxdata/flux/stdlib"
|
|
_ "github.com/influxdata/influxdb/query/stdlib"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var queryFlags struct {
|
|
org organization
|
|
}
|
|
|
|
func cmdQuery() *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "query [query literal or @/path/to/query.flux]",
|
|
Short: "Execute a Flux query",
|
|
Long: `Execute a literal Flux query provided as a string,
|
|
or execute a literal Flux query contained in a file by specifying the file prefixed with an @ sign.`,
|
|
Args: cobra.ExactArgs(1),
|
|
RunE: wrapCheckSetup(fluxQueryF),
|
|
}
|
|
queryFlags.org.register(cmd, true)
|
|
|
|
return cmd
|
|
}
|
|
|
|
func fluxQueryF(cmd *cobra.Command, args []string) error {
|
|
if flags.local {
|
|
return fmt.Errorf("local flag not supported for query command")
|
|
}
|
|
|
|
if err := queryFlags.org.validOrgFlags(); err != nil {
|
|
return err
|
|
}
|
|
|
|
q, err := repl.LoadQuery(args[0])
|
|
if err != nil {
|
|
return fmt.Errorf("failed to load query: %v", err)
|
|
}
|
|
|
|
orgSvc, err := newOrganizationService()
|
|
if err != nil {
|
|
return fmt.Errorf("failed to initialized organization service client: %v", err)
|
|
}
|
|
|
|
orgID, err := queryFlags.org.getID(orgSvc)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
flux.FinalizeBuiltIns()
|
|
|
|
r, err := getFluxREPL(flags.host, flags.token, flags.skipVerify, orgID)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to get the flux REPL: %v", err)
|
|
}
|
|
|
|
if err := r.Input(q); err != nil {
|
|
return fmt.Errorf("failed to execute query: %v", err)
|
|
}
|
|
|
|
return nil
|
|
}
|