premium

Documentation

Index

• Constants
• func ContainsPaidTables(tables schema.Tables) bool
• func MakeAllTablesPaid(tables schema.Tables) schema.Tables
• func WithCancelOnQuotaExceeded(ctx context.Context, qm QuotaMonitor, ops ...QuotaCheckOption) (context.Context, error)
• func WithPluginKind(pluginKind string) cqapi.PluginKind
• func WithPluginName(pluginName string) cqapi.PluginName
• func WithPluginTeam(pluginTeam string) cqapi.PluginTeam
• type BatchUpdater
  • func NewUsageClient(pluginTeam cqapi.PluginTeam, pluginKind cqapi.PluginKind, ...) (*BatchUpdater, error)
  • func (u *BatchUpdater) Close() error
  • func (u *BatchUpdater) HasQuota(ctx context.Context) (bool, error)
  • func (u *BatchUpdater) Increase(rows uint32) error
  • func (u *BatchUpdater) TeamName() string
• type ErrNoQuota
  • func (e ErrNoQuota) Error() string
• type QuotaCheckOption
  • func WithQuotaCheckPeriod(duration time.Duration) QuotaCheckOption
  • func WithQuotaMaxConsecutiveFailures(n int) QuotaCheckOption
• type QuotaMonitor
• type TokenClient
• type UsageClient
• type UsageClientOptions
  • func WithAPIClient(apiClient *cqapi.ClientWithResponses) UsageClientOptions
  • func WithBatchLimit(batchLimit uint32) UsageClientOptions
  • func WithLogger(logger zerolog.Logger) UsageClientOptions
  • func WithMaxRetries(maxRetries int) UsageClientOptions
  • func WithMaxTimeBetweenFlushes(maxTimeBetweenFlushes time.Duration) UsageClientOptions
  • func WithMaxWaitTime(maxWaitTime time.Duration) UsageClientOptions
  • func WithMinTimeBetweenFlushes(minTimeBetweenFlushes time.Duration) UsageClientOptions
  • func WithURL(url string) UsageClientOptions

Constants

const DefaultMaxQuotaFailures = 10 // 5 minutes

const DefaultQuotaCheckInterval = 30 * time.Second

Functions

func ContainsPaidTables

func ContainsPaidTables(tables schema.Tables) bool

ContainsPaidTables returns true if any of the tables are paid

func MakeAllTablesPaid

func MakeAllTablesPaid(tables schema.Tables) schema.Tables

MakeAllTablesPaid sets all tables to paid (including relations)

func WithCancelOnQuotaExceeded

func WithCancelOnQuotaExceeded(ctx context.Context, qm QuotaMonitor, ops ...QuotaCheckOption) (context.Context, error)

WithCancelOnQuotaExceeded monitors the quota usage at intervals defined by duration and cancels the context if the quota is exceeded

func WithPluginKind

func WithPluginKind(pluginKind string) cqapi.PluginKind

func WithPluginName

func WithPluginName(pluginName string) cqapi.PluginName

func WithPluginTeam

func WithPluginTeam(pluginTeam string) cqapi.PluginTeam

Types

type BatchUpdater

type BatchUpdater struct {
	// contains filtered or unexported fields
}

func NewUsageClient

func NewUsageClient(pluginTeam cqapi.PluginTeam, pluginKind cqapi.PluginKind, pluginName cqapi.PluginName, ops ...UsageClientOptions) (*BatchUpdater, error)

func (*BatchUpdater) Close

func (u *BatchUpdater) Close() error

func (*BatchUpdater) HasQuota

func (u *BatchUpdater) HasQuota(ctx context.Context) (bool, error)

func (*BatchUpdater) Increase

func (u *BatchUpdater) Increase(rows uint32) error

func (*BatchUpdater) TeamName

func (u *BatchUpdater) TeamName() string

type ErrNoQuota

type ErrNoQuota struct {
	// contains filtered or unexported fields
}

func (ErrNoQuota) Error

func (e ErrNoQuota) Error() string

type QuotaCheckOption

type QuotaCheckOption func(*quotaChecker)

func WithQuotaCheckPeriod

func WithQuotaCheckPeriod(duration time.Duration) QuotaCheckOption

WithQuotaCheckPeriod controls the time interval between quota checks

func WithQuotaMaxConsecutiveFailures

func WithQuotaMaxConsecutiveFailures(n int) QuotaCheckOption

WithQuotaMaxConsecutiveFailures controls the number of consecutive failed quota checks before the context is cancelled

type QuotaMonitor

type QuotaMonitor interface {
	// TeamName returns the team name
	TeamName() string
	// HasQuota returns true if the quota has not been exceeded
	HasQuota(context.Context) (bool, error)
}

type TokenClient

type TokenClient interface {
	GetToken() (auth.Token, error)
	GetTokenType() auth.TokenType
}

type UsageClient

type UsageClient interface {
	QuotaMonitor
	// Increase updates the usage by the given number of rows
	Increase(uint32) error
	// Close flushes any remaining rows and closes the quota service
	Close() error
}

type UsageClientOptions

type UsageClientOptions func(updater *BatchUpdater)

func WithAPIClient

func WithAPIClient(apiClient *cqapi.ClientWithResponses) UsageClientOptions

WithAPIClient sets the API client to use - defaults to a client using a bearer token generated from the refresh token stored in the configuration

func WithBatchLimit

func WithBatchLimit(batchLimit uint32) UsageClientOptions

WithBatchLimit sets the maximum number of rows to update in a single request

func WithLogger

func WithLogger(logger zerolog.Logger) UsageClientOptions

WithLogger sets the logger to use - defaults to a no-op logger

func WithMaxRetries

func WithMaxRetries(maxRetries int) UsageClientOptions

WithMaxRetries sets the maximum number of retries to update the usage in case of an API error

func WithMaxTimeBetweenFlushes

func WithMaxTimeBetweenFlushes(maxTimeBetweenFlushes time.Duration) UsageClientOptions

WithMaxTimeBetweenFlushes sets the flush duration - the time at which an update will be triggered even if the batch limit is not reached

func WithMaxWaitTime

func WithMaxWaitTime(maxWaitTime time.Duration) UsageClientOptions

WithMaxWaitTime sets the maximum time to wait before retrying a failed update

func WithMinTimeBetweenFlushes

func WithMinTimeBetweenFlushes(minTimeBetweenFlushes time.Duration) UsageClientOptions

WithMinTimeBetweenFlushes sets the minimum time between updates

func WithURL

func WithURL(url string) UsageClientOptions

WithURL sets the API URL to use - defaults to https://api.cloudquery.io