jprdonnelly/steampipe
1
0
Fork 0
mirror of https://github.com/turbot/steampipe.git synced 2026-04-14 07:00:08 -04:00
Code Issues Packages Projects Releases Wiki Activity

Fix updating prepared statements during file watch

Browse Source
This commit is contained in:
kai
2021-09-03 11:37:53 +01:00
parent 7ccf5b4b73
commit 49bd110ef0
1 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
db/db_common/prepared_statements.go
View File

@@ -28,7 +28,8 @@ func CreatePreparedStatements(ctx context.Context, queryMap map[string]*modconfi
// execute the query, passing 'true' to disable the spinner
_, err := client.ExecuteSync(ctx, sql, true)
if err != nil {
return fmt.Errorf("failed to create prepared statements table %s: %v", name, err)
log.Printf("[TRACE] failed to create prepared statement for %s: %v", name, err)
return fmt.Errorf("failed to create prepared statement for %s: %v", name, err)
}
}
@@ -48,7 +49,8 @@ func CreatePreparedStatements(ctx context.Context, queryMap map[string]*modconfi
// execute the query, passing 'true' to disable the spinner
_, err := client.ExecuteSync(ctx, sql, true)
if err != nil {
return fmt.Errorf("failed to create prepared statements table %s: %v", name, err)
log.Printf("[TRACE] failed to create prepared statement for %s: %v", name, err)
return fmt.Errorf("failed to create prepared statements for %s: %v", name, err)
}
}
@@ -69,7 +71,7 @@ func UpdatePreparedStatements(ctx context.Context, queryMap map[string]*modconfi
if !strings.HasPrefix(name, "query.") {
continue
}
sql = append(sql, fmt.Sprintf("DEALLOCATE %s ", query.PreparedStatementName()))
sql = append(sql, fmt.Sprintf("DEALLOCATE %s;", query.PreparedStatementName()))
}
for name, control := range controlMap {
@@ -81,11 +83,16 @@ func UpdatePreparedStatements(ctx context.Context, queryMap map[string]*modconfi
if control.Query != nil {
continue
}
sql = append(sql, fmt.Sprintf("DEALLOCATE %s ", control.PreparedStatementName()))
sql = append(sql, fmt.Sprintf("DEALLOCATE %s;", control.PreparedStatementName()))
}
// execute the query, passing 'true' to disable the spinner
// ignore errors
client.ExecuteSync(ctx, strings.Join(sql, "\n"), true)
s := strings.Join(sql, "\n")
_, err := client.ExecuteSync(ctx, s, true)
if err != nil {
log.Printf("[TRACE] failed to update prepared statements - deallocate returned error %v", err)
return err
}
// now recreate them
return CreatePreparedStatements(ctx, queryMap, controlMap, client)