jprdonnelly/steampipe
1
0
Fork 0
mirror of https://github.com/turbot/steampipe.git synced 2025-12-25 03:00:48 -05:00
Code Issues Packages Projects Releases Wiki Activity

Update database code to use pgx interface so we can leverage the connection pool hook functions to pre-warm connections. Closes #2422 (#2438)

Browse Source 
* Provide feedback for failed prepared statements
* Move error functions to error_helpers
* Make maintenance client retriable
This commit is contained in:
kaidaguerre
2022-10-05 12:38:57 +01:00
committed by GitHub
parent e2d12b3fc2
commit 404dd35e21
297 changed files with 1170 additions and 1128 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -11,6 +11,7 @@ import (
"github.com/turbot/steampipe/pkg/contexthelpers"
"github.com/turbot/steampipe/pkg/db/db_common"
"github.com/turbot/steampipe/pkg/display"
"github.com/turbot/steampipe/pkg/error_helpers"
"github.com/turbot/steampipe/pkg/interactive"
"github.com/turbot/steampipe/pkg/query"
"github.com/turbot/steampipe/pkg/utils"
@@ -22,7 +23,7 @@ func RunInteractiveSession(ctx context.Context, initData *query.InitData) {
// the db executor sends result data over resultsStreamer
resultsStreamer, err := interactive.RunInteractivePrompt(ctx, initData)
utils.FailOnError(err)
error_helpers.FailOnError(err)
// print the data as it comes
for r := range resultsStreamer.Results {
@@ -43,7 +44,7 @@ func RunBatchSession(ctx context.Context, initData *query.InitData) int {
// wait for init
<-initData.Loaded
if err := initData.Result.Error; err != nil {
utils.FailOnError(err)
error_helpers.FailOnError(err)
}
// display any initialisation messages/warnings
@@ -68,7 +69,7 @@ func executeQueries(ctx context.Context, queries []string, client db_common.Clie
for i, q := range queries {
if err := executeQuery(ctx, q, client); err != nil {
failures++
utils.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(queries), err))
// if timing flag is enabled, show the time taken for the query to fail
if cmdconfig.Viper().GetBool(constants.ArgTiming) {
display.DisplayErrorTiming(t)