mirror of
https://github.com/turbot/steampipe.git
synced 2026-04-07 04:00:06 -04:00
67 lines
2.0 KiB
Go
67 lines
2.0 KiB
Go
package metaquery
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/spf13/viper"
|
|
"github.com/turbot/go-kit/types"
|
|
"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 {
|
|
// 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)
|
|
}()
|
|
command := strings.ToLower(input.args()[0])
|
|
switch command {
|
|
case constants.ArgOn, constants.ArgOff:
|
|
val := types.StringToBool(command)
|
|
viper.Set(constants.ArgClientCacheEnabled, val)
|
|
return db_common.SetCacheEnabled(ctx, val, 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())
|
|
}
|