mirror of
https://github.com/turbot/steampipe.git
synced 2025-12-19 18:12:43 -05:00
26 lines
730 B
Go
26 lines
730 B
Go
package db_common
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/turbot/pipe-fittings/v2/utils"
|
|
"github.com/turbot/steampipe/v2/pkg/query/queryresult"
|
|
)
|
|
|
|
// ExecuteQuery executes a single query. If shutdownAfterCompletion is true, shutdown the client after completion
|
|
func ExecuteQuery(ctx context.Context, client Client, queryString string, args ...any) (*queryresult.ResultStreamer, error) {
|
|
utils.LogTime("db.ExecuteQuery start")
|
|
defer utils.LogTime("db.ExecuteQuery end")
|
|
|
|
resultsStreamer := queryresult.NewResultStreamer()
|
|
result, err := client.Execute(ctx, queryString, args...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
go func() {
|
|
resultsStreamer.StreamResult(result)
|
|
resultsStreamer.Close()
|
|
}()
|
|
return resultsStreamer, nil
|
|
}
|