jprdonnelly/steampipe
1
0
Fork 0
mirror of https://github.com/turbot/steampipe.git synced 2026-03-31 18:00:19 -04:00
Code Issues Packages Projects Releases Wiki Activity

Add support for sharing snapshots. Add snapshot output and export format for query and check. Add ability to run named dashboards. #2473

Browse Source
This commit is contained in:
kaidaguerre
2022-10-05 15:30:57 +01:00
committed by GitHub
parent 404dd35e21
commit ef845286c1
78 changed files with 1435 additions and 928 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
pkg/query/queryexecute/execute.go
View File

@@ -53,23 +53,28 @@ func RunBatchSession(ctx context.Context, initData *query.InitData) int {
failures := 0
if len(initData.Queries) > 0 {
// if we have resolved any queries, run them
failures = executeQueries(ctx, initData.Queries, initData.Client)
failures = executeQueries(ctx, initData)
}
// set global exit code
return failures
}
func executeQueries(ctx context.Context, queries []string, client db_common.Client) int {
func executeQueries(ctx context.Context, initData *query.InitData) int {
utils.LogTime("queryexecute.executeQueries start")
defer utils.LogTime("queryexecute.executeQueries end")
// run all queries
failures := 0
t := time.Now()
for i, q := range queries {
if err := executeQuery(ctx, q, client); err != nil {
// build ordered list of queries
// (ordered for testing repeatability)
var queryNames []string = utils.SortedMapKeys(initData.Queries)
for i, name := range queryNames {
q := initData.Queries[name]
if err := executeQuery(ctx, q, initData.Client); err != nil {
failures++
error_helpers.ShowWarning(fmt.Sprintf("executeQueries: query %d of %d failed: %v", i+1, len(queries), err))
error_helpers.ShowWarning(fmt.Sprintf("executeQueries: query %d of %d failed: %v", i+1, len(queryNames), err))
// if timing flag is enabled, show the time taken for the query to fail
if cmdconfig.Viper().GetBool(constants.ArgTiming) {
display.DisplayErrorTiming(t)
@@ -77,7 +82,7 @@ func executeQueries(ctx context.Context, queries []string, client db_common.Clie
}
// TODO move into display layer
// Only show the blank line between queries, not after the last one
if (i < len(queries)-1) && showBlankLineBetweenResults() {
if (i < len(queryNames)-1) && showBlankLineBetweenResults() {
fmt.Println()
}
}