mirror of
https://github.com/turbot/steampipe.git
synced 2026-02-15 04:00:14 -05:00
40804a3201
- Execute RefreshConnections asyncronously - Add connection_state table to indicate the loading state of connections - Optimise RefreshConnections by cloning connection schemas - Add locking to ensure only a single instance of RefreshConnections runs - Start executing queries without waiting for connections to load, add smart error handling to wait for required connection - Optimise autocomplete for high connection count - Autocomplete and inspect data available before all conections are refreshed - Update file watcher to respond to CHMOD, so thaat it pickes up deletion of file contents Closes #3394 Closes #3267
67 lines
2.1 KiB
Go
67 lines
2.1 KiB
Go
package metaquery
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/spf13/viper"
|
|
"github.com/turbot/steampipe/pkg/constants"
|
|
"github.com/turbot/steampipe/pkg/db/db_common"
|
|
"github.com/turbot/steampipe/sperr"
|
|
)
|
|
|
|
// controls the cache in the connected FDW
|
|
func cacheControl(ctx context.Context, input *HandlerInput) error {
|
|
command := input.args()[0]
|
|
// just get the active session from the connection pool
|
|
// and set the cache parameters on it.
|
|
// NOTE: this works because the interactive client
|
|
// always has only one active connection due to the way it works
|
|
sessionResult := input.Client.AcquireSession(ctx)
|
|
if sessionResult.Error != nil {
|
|
return sessionResult.Error
|
|
}
|
|
conn := sessionResult.Session.Connection.Conn()
|
|
defer func() {
|
|
// we need to do this in a closure, otherwise the ctx will be evaluated immediately
|
|
// and not in call-time
|
|
sessionResult.Session.Close(false)
|
|
}()
|
|
switch command {
|
|
case constants.ArgOn:
|
|
viper.Set(constants.ArgClientCacheEnabled, true)
|
|
return db_common.SetCacheEnabled(ctx, true, sessionResult.Session.Connection.Conn())
|
|
case constants.ArgOff:
|
|
viper.Set(constants.ArgClientCacheEnabled, false)
|
|
return db_common.SetCacheEnabled(ctx, false, sessionResult.Session.Connection.Conn())
|
|
case constants.ArgClear:
|
|
return db_common.CacheClear(ctx, conn)
|
|
}
|
|
|
|
return fmt.Errorf("invalid command")
|
|
}
|
|
|
|
// sets the cache TTL
|
|
func cacheTTL(ctx context.Context, input *HandlerInput) error {
|
|
seconds, err := strconv.Atoi(input.args()[0])
|
|
if err != nil {
|
|
return sperr.WrapWithMessage(err, "valid value is the number of seconds")
|
|
}
|
|
if seconds < 0 {
|
|
return sperr.New("ttl must be greater than 0")
|
|
}
|
|
sessionResult := input.Client.AcquireSession(ctx)
|
|
if sessionResult.Error != nil {
|
|
return sessionResult.Error
|
|
}
|
|
defer func() {
|
|
// we need to do this in a closure, otherwise the ctx will be evaluated immediately
|
|
// and not in call-time
|
|
sessionResult.Session.Close(false)
|
|
viper.Set(constants.ArgCacheTtl, seconds)
|
|
}()
|
|
return db_common.SetCacheTtl(ctx, time.Duration(seconds)*time.Second, sessionResult.Session.Connection.Conn())
|
|
}
|