mirror of
https://github.com/turbot/steampipe.git
synced 2026-01-24 08:01:31 -05:00
299697ae2f
STEAMPIPE_CACHE environment variable resolves to service cache enabled as well as client cache enabled service cache enabled is used by the plugin manager to enable/disable caching on the plugins during startup (runtime toggle is not allowed) - with a max_ttl client cache enabled is used to enable/disable the cache on the database connection (fdw) A TTL can also be set on the client side capped to max_ttl on the server
116 lines
3.1 KiB
Go
116 lines
3.1 KiB
Go
package query
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/spf13/viper"
|
|
"github.com/turbot/steampipe/pkg/constants"
|
|
"github.com/turbot/steampipe/pkg/export"
|
|
"github.com/turbot/steampipe/pkg/initialisation"
|
|
"github.com/turbot/steampipe/pkg/statushooks"
|
|
"github.com/turbot/steampipe/pkg/steampipeconfig/modconfig"
|
|
"github.com/turbot/steampipe/pkg/workspace"
|
|
)
|
|
|
|
type InitData struct {
|
|
initialisation.InitData
|
|
|
|
cancelInitialisation context.CancelFunc
|
|
Loaded chan struct{}
|
|
// map of query name to resolved query (key is the query text for command line queries)
|
|
Queries map[string]*modconfig.ResolvedQuery
|
|
}
|
|
|
|
// NewInitData returns a new InitData object
|
|
// It also starts an asynchronous population of the object
|
|
// InitData.Done closes after asynchronous initialization completes
|
|
func NewInitData(ctx context.Context, args []string) *InitData {
|
|
i := &InitData{
|
|
InitData: *initialisation.NewInitData(),
|
|
Loaded: make(chan struct{}),
|
|
}
|
|
go i.init(ctx, args)
|
|
|
|
return i
|
|
}
|
|
|
|
func queryExporters() []export.Exporter {
|
|
return []export.Exporter{&export.SnapshotExporter{}}
|
|
}
|
|
|
|
func (i *InitData) Cancel() {
|
|
// cancel any ongoing operation
|
|
if i.cancelInitialisation != nil {
|
|
i.cancelInitialisation()
|
|
}
|
|
i.cancelInitialisation = nil
|
|
}
|
|
|
|
// Cleanup overrides the initialisation.InitData.Cleanup to provide syncronisation with the loaded channel
|
|
func (i *InitData) Cleanup(ctx context.Context) {
|
|
// cancel any ongoing operation
|
|
i.Cancel()
|
|
|
|
// ensure that the initialisation was completed
|
|
// and that we are not in a race condition where
|
|
// the client is set after the cancel hits
|
|
<-i.Loaded
|
|
|
|
// if a client was initialised, close it
|
|
if i.Client != nil {
|
|
i.Client.Close(ctx)
|
|
}
|
|
if i.ShutdownTelemetry != nil {
|
|
i.ShutdownTelemetry()
|
|
}
|
|
}
|
|
|
|
func (i *InitData) init(ctx context.Context, args []string) {
|
|
defer func() {
|
|
close(i.Loaded)
|
|
// clear the cancelInitialisation function
|
|
i.cancelInitialisation = nil
|
|
}()
|
|
|
|
// validate export args
|
|
if len(viper.GetStringSlice(constants.ArgExport)) > 0 {
|
|
i.RegisterExporters(queryExporters()...)
|
|
|
|
// validate required export formats
|
|
if err := i.ExportManager.ValidateExportFormat(viper.GetStringSlice(constants.ArgExport)); err != nil {
|
|
i.Result.Error = err
|
|
return
|
|
}
|
|
}
|
|
|
|
statushooks.SetStatus(ctx, "Loading workspace")
|
|
w, errAndWarnings := workspace.LoadWorkspacePromptingForVariables(ctx)
|
|
if errAndWarnings.GetError() != nil {
|
|
i.Result.Error = fmt.Errorf("failed to load workspace: %s", errAndWarnings.GetError().Error())
|
|
return
|
|
}
|
|
i.Result.AddWarnings(errAndWarnings.Warnings...)
|
|
i.Workspace = w
|
|
|
|
// set max DB connections to 1
|
|
viper.Set(constants.ArgMaxParallel, 1)
|
|
|
|
statushooks.SetStatus(ctx, "Resolving arguments")
|
|
|
|
// convert the query or sql file arg into an array of executable queries - check names queries in the current workspace
|
|
resolvedQueries, err := w.GetQueriesFromArgs(args)
|
|
if err != nil {
|
|
i.Result.Error = err
|
|
return
|
|
}
|
|
// create a cancellable context so that we can cancel the initialisation
|
|
ctx, cancel := context.WithCancel(ctx)
|
|
// and store it
|
|
i.cancelInitialisation = cancel
|
|
i.Queries = resolvedQueries
|
|
|
|
// and call base init
|
|
i.InitData.Init(ctx, constants.InvokerQuery)
|
|
}
|