mirror of
https://github.com/turbot/steampipe.git
synced 2026-02-16 07:00:18 -05:00
96 lines
2.7 KiB
Go
96 lines
2.7 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
|
|
"github.com/turbot/go-kit/helpers"
|
|
"github.com/turbot/steampipe-plugin-sdk/logging"
|
|
"github.com/turbot/steampipe/cmdconfig"
|
|
"github.com/turbot/steampipe/display"
|
|
|
|
"github.com/turbot/steampipe/constants"
|
|
"github.com/turbot/steampipe/db"
|
|
"github.com/turbot/steampipe/utils"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// QueryCmd :: represents the query command
|
|
func QueryCmd() *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "query",
|
|
TraverseChildren: true,
|
|
Args: cobra.ArbitraryArgs,
|
|
Run: runQueryCmd,
|
|
Short: "Execute SQL queries interactively or by argument",
|
|
Long: `Execute SQL queries interactively, or by a query argument.
|
|
|
|
Open a interactive SQL query console to Steampipe to explore your data and run
|
|
multiple queries. If QUERY is passed on the command line then it will be run
|
|
immediately and the command will exit.
|
|
|
|
Examples:
|
|
|
|
# Open an interactive query console
|
|
steampipe query
|
|
|
|
# Run a specific query directly
|
|
steampipe query "select * from cloud"`,
|
|
}
|
|
|
|
// Notes:
|
|
// * In the future we may add --csv and --json flags as shortcuts for --output
|
|
cmdconfig.
|
|
OnCmd(cmd).
|
|
AddBoolFlag(constants.ArgHeader, "", true, "Include column headers csv and table output").
|
|
AddStringFlag(constants.ArgSeparator, "", ",", "Separator string for csv output").
|
|
AddStringFlag(constants.ArgOutput, "", "table", "Output format: line, csv, json or table").
|
|
AddBoolFlag(constants.ArgTimer, "", false, "Turn on the timer which reports query time.")
|
|
|
|
return cmd
|
|
}
|
|
|
|
func runQueryCmd(cmd *cobra.Command, args []string) {
|
|
logging.LogTime("runQueryCmd start")
|
|
|
|
defer func() {
|
|
logging.LogTime("runQueryCmd end")
|
|
if r := recover(); r != nil {
|
|
utils.ShowError(helpers.ToError(r))
|
|
}
|
|
}()
|
|
|
|
queryString := getQuery(cmd, args)
|
|
|
|
// set the flag to not show spinner
|
|
showSpinner := (queryString == "")
|
|
cmdconfig.Viper().Set(constants.ShowInteractiveOutputConfigKey, showSpinner)
|
|
|
|
// the db executor sends result data over resultsStreamer
|
|
resultsStreamer, err := db.ExecuteQuery(queryString)
|
|
utils.FailOnError(err)
|
|
|
|
// print the data as it comes
|
|
for r := range resultsStreamer.Results {
|
|
display.ShowOutput(r)
|
|
//signal to the resultStreamer that we are done with this chunk of the stream
|
|
resultsStreamer.Done()
|
|
}
|
|
}
|
|
|
|
// retrieve query from args or determine whether to run the interactive shell
|
|
func getQuery(cmd *cobra.Command, args []string) (query string) {
|
|
log.Println("[TRACE] getQuery")
|
|
if len(args) == 0 {
|
|
if cmdconfig.Viper().GetBool(constants.ArgListAllTableNames) {
|
|
query = ".tables"
|
|
} else if table := cmdconfig.Viper().GetString(constants.ArgSelectAll); table != "" {
|
|
query = fmt.Sprintf("select * from %s", table)
|
|
}
|
|
} else {
|
|
query = args[0]
|
|
}
|
|
return
|
|
}
|