timestreamquery

type AccountSettings struct {
	MaxQueryTCU       *int32
	QueryCompute      *QueryCompute
	QueryPricingModel string
}

type AccountSettingsNotificationConfiguration struct {
	SnsConfiguration *SnsConfiguration `json:"SnsConfiguration,omitempty"`
	RoleArn          string            `json:"RoleArn,omitempty"`
}

type Catalog interface {
	DatabaseExists(region, database string) bool
	TableExists(region, database, table string) bool
}

type ColumnInfo struct {
	Name string     `json:"Name,omitempty"`
	Type ColumnType `json:"Type"`
}

type ColumnType struct {
	ArrayColumnInfo                  *ColumnInfo  `json:"ArrayColumnInfo,omitempty"`
	TimeSeriesMeasureValueColumnInfo *ColumnInfo  `json:"TimeSeriesMeasureValueColumnInfo,omitempty"`
	ScalarType                       string       `json:"ScalarType,omitempty"`
	RowColumnInfo                    []ColumnInfo `json:"RowColumnInfo,omitempty"`
}

type Datum struct {
	ScalarValue     *string               `json:"ScalarValue,omitempty"`
	NullValue       *bool                 `json:"NullValue,omitempty"`
	ArrayValue      []Datum               `json:"ArrayValue,omitempty"`
	RowValue        *Row                  `json:"RowValue,omitempty"`
	TimeSeriesValue []TimeSeriesDataPoint `json:"TimeSeriesValue,omitempty"`
}

type DimensionMapping struct {
	Name               string `json:"Name"`
	DimensionValueType string `json:"DimensionValueType"`
}

type ErrorReportLocation struct {
	S3ReportLocation *S3ReportLocation `json:"S3ReportLocation,omitempty"`
}

type ExecutionStats struct {
	ExecutionTimeInMillis  int64 `json:"ExecutionTimeInMillis,omitempty"`
	DataWrites             int64 `json:"DataWrites,omitempty"`
	BytesMetered           int64 `json:"BytesMetered,omitempty"`
	CumulativeBytesScanned int64 `json:"CumulativeBytesScanned,omitempty"`
	QueryResultRows        int64 `json:"QueryResultRows,omitempty"`
	RecordsIngested        int64 `json:"RecordsIngested,omitempty"`
}

type Handler struct {
	Backend StorageBackend
	// contains filtered or unexported fields
}

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

type LastRunSummary struct {
	ErrorReportLocation *ErrorReportLocation `json:"ErrorReportLocation,omitempty"`
	ExecutionStats      *ExecutionStats      `json:"ExecutionStats,omitempty"`
	FailureReason       string               `json:"FailureReason,omitempty"`
	RunStatus           string               `json:"RunStatus,omitempty"`
	TriggerTime         float64              `json:"TriggerTime,omitempty"`
	InvocationTime      float64              `json:"InvocationTime,omitempty"`
}

type LastUpdate struct {
	TargetQueryTCU *int32 `json:"TargetQueryTCU,omitempty"`
	Status         string `json:"Status,omitempty"`
}

type ListScheduledQueriesResult struct {
	NextToken string
	Items     []ScheduledQueryListEntry
}

type MixedMeasureMapping struct {
	MeasureName                   string                         `json:"MeasureName,omitempty"`
	SourceColumn                  string                         `json:"SourceColumn,omitempty"`
	TargetMeasureName             string                         `json:"TargetMeasureName,omitempty"`
	MeasureValueType              string                         `json:"MeasureValueType"`
	MultiMeasureAttributeMappings []MultiMeasureAttributeMapping `json:"MultiMeasureAttributeMappings,omitempty"`
}

type MultiMeasureAttributeMapping struct {
	SourceColumn                    string `json:"SourceColumn"`
	TargetMultiMeasureAttributeName string `json:"TargetMultiMeasureAttributeName,omitempty"`
	MeasureValueType                string `json:"MeasureValueType"`
}

type MultiMeasureMappings struct {
	TargetMultiMeasureName        string                         `json:"TargetMultiMeasureName,omitempty"`
	MultiMeasureAttributeMappings []MultiMeasureAttributeMapping `json:"MultiMeasureAttributeMappings"`
}

type PrepareQueryResult struct {
	QueryString string
	Columns     []ColumnInfo
	Selected    []SelectColumn
	Parameters  []ColumnInfo
}

type Provider struct{}

type ProvisionedCapacity struct {
	ActiveQueryTCU            *int32                                    `json:"ActiveQueryTCU,omitempty"`
	LastUpdate                *LastUpdate                               `json:"LastUpdate,omitempty"`
	NotificationConfiguration *AccountSettingsNotificationConfiguration `json:"NotificationConfiguration,omitempty"`
}

type QueryCompute struct {
	ProvisionedCapacity *ProvisionedCapacity `json:"ProvisionedCapacity,omitempty"`
	ComputeMode         string               `json:"ComputeMode,omitempty"` // ON_DEMAND | PROVISIONED
}

type QueryComputeUpdate struct {
	TargetQueryTCU            *int32
	NotificationConfiguration *AccountSettingsNotificationConfiguration
	ComputeMode               string
}

type QueryInsightsResponse struct {
	OutputRows           int64   `json:"OutputRows"`
	OutputBytes          int64   `json:"OutputBytes"`
	UnloadPartitionCount int64   `json:"UnloadPartitionCount,omitempty"`
	UnloadWrittenRows    int64   `json:"UnloadWrittenRows,omitempty"`
	UnloadWrittenBytes   int64   `json:"UnloadWrittenBytes,omitempty"`
	QuerySpatialCoverage float64 `json:"QuerySpatialCoverage,omitempty"`
}

type QueryOptions struct {
	QueryString  string
	ClientToken  string
	NextToken    string
	InsightsMode string
	MaxRows      int32
}

type QueryPage struct {
	Insights    *QueryInsightsResponse
	QueryID     string
	NextToken   string
	Rows        []Row
	Columns     []ColumnInfo
	QueryStatus QueryStatusDetail
}

type QueryResult struct {
	QueryID     string
	Rows        []Row
	Columns     []ColumnInfo
	Insights    QueryInsightsResponse
	QueryStatus QueryStatusDetail
	// Cancelled records whether CancelQuery has already been issued for this
	// query. CancelQueryOutput.CancellationMessage is documented as returned
	// "when a CancelQuery request for the query ... has already been issued",
	// i.e. cancellation is idempotent -- a repeat CancelQuery call for the
	// same QueryId must still succeed, not 404.
	Cancelled bool
}

type QueryStatusDetail struct {
	ProgressPercentage     float64 `json:"ProgressPercentage"`
	CumulativeBytesScanned int64   `json:"CumulativeBytesScanned"`
	CumulativeBytesMetered int64   `json:"CumulativeBytesMetered"`
}

type Row struct {
	Data []Datum `json:"Data"`
}

type S3ReportLocation struct {
	ObjectKey  string `json:"ObjectKey,omitempty"`
	BucketName string `json:"BucketName,omitempty"`
}

type ScheduledQuery struct {
	LastRunTime             time.Time                   `json:"last_run_time"`
	CreationTime            time.Time                   `json:"creation_time"`
	TargetDetail            *ScheduledQueryTargetDetail `json:"target_detail,omitempty"`
	Tags                    map[string]string           `json:"tags"`
	NotificationTopicArn    string                      `json:"notification_topic_arn"`
	ScheduleExpression      string                      `json:"schedule_expression"`
	ExecutionRoleArn        string                      `json:"execution_role_arn"`
	QueryString             string                      `json:"query_string"`
	ErrorReportS3BucketName string                      `json:"error_report_s3_bucket_name"`
	TargetDatabase          string                      `json:"target_database"`
	TargetTable             string                      `json:"target_table"`
	State                   string                      `json:"state"`
	Name                    string                      `json:"name"`
	Arn                     string                      `json:"arn"`
	KmsKeyID                string                      `json:"kms_key_id,omitempty"`
	// ErrorReportEncryptionOption/ErrorReportObjectKeyPrefix mirror
	// types.S3Configuration.EncryptionOption/ObjectKeyPrefix (both optional,
	// api_op_CreateScheduledQuery.go) -- accepted and echoed back, but this
	// emulator performs no actual at-rest encryption of error reports (same
	// honest scoping as KmsKeyID, see PARITY.md gaps).
	ErrorReportEncryptionOption string `json:"error_report_encryption_option,omitempty"`
	ErrorReportObjectKeyPrefix  string `json:"error_report_object_key_prefix,omitempty"`
}

type ScheduledQueryListEntry struct {
	TargetDestination      *TargetDestinationForList `json:"TargetDestination,omitempty"`
	Arn                    string                    `json:"Arn"`
	Name                   string                    `json:"Name"`
	State                  string                    `json:"State"`
	LastRunStatus          string                    `json:"LastRunStatus,omitempty"`
	CreationTime           float64                   `json:"CreationTime,omitempty"`
	NextInvocationTime     float64                   `json:"NextInvocationTime,omitempty"`
	PreviousInvocationTime float64                   `json:"PreviousInvocationTime,omitempty"`
}

type ScheduledQuerySummary struct {
	Arn   string `json:"Arn"`
	Name  string `json:"Name"`
	State string `json:"State"`
}

type ScheduledQueryTargetDetail struct {
	MultiMeasureMappings *MultiMeasureMappings
	TimeColumn           string
	MeasureNameColumn    string
	DimensionMappings    []DimensionMapping
	MixedMeasureMappings []MixedMeasureMapping
}

type SelectColumn struct {
	DatabaseName string
	TableName    string
	ColumnInfo
	Aliased bool
}

type SnsConfiguration struct {
	TopicArn string `json:"TopicArn,omitempty"`
}

type StorageBackend interface {
	AccountID() string
	Region() string
	CreateScheduledQuery(
		ctx context.Context,
		name, queryString, scheduleExpression, executionRoleArn,
		notificationTopicArn, errorReportS3BucketName, targetDatabase, targetTable, clientToken, kmsKeyID string,
		errorReportEncryptionOption, errorReportObjectKeyPrefix string,
		tags map[string]string,
		targetDetail *ScheduledQueryTargetDetail,
	) (*ScheduledQuery, error)
	DescribeScheduledQuery(ctx context.Context, arnStr string) (*ScheduledQuery, error)
	DeleteScheduledQuery(ctx context.Context, arnStr string) error
	ListScheduledQueries(ctx context.Context) []ScheduledQuerySummary
	ListScheduledQueriesFull(ctx context.Context) []*ScheduledQuery
	ListScheduledQueriesEnriched(ctx context.Context, nextToken string, maxResults int32) ListScheduledQueriesResult
	UpdateScheduledQuery(ctx context.Context, arnStr, state string) error
	ExecuteScheduledQuery(ctx context.Context, arnStr string, invocationTime time.Time) error
	Query(ctx context.Context, queryString string) *QueryResult
	QueryWithOptions(ctx context.Context, opts QueryOptions) (*QueryPage, error)
	CancelQuery(ctx context.Context, queryID string) error
	TagResource(ctx context.Context, arnStr string, tags map[string]string) error
	UntagResource(ctx context.Context, arnStr string, tagKeys []string) error
	ListTagsForResource(ctx context.Context, arnStr string) ([]map[string]string, error)
	DescribeAccountSettings(ctx context.Context) AccountSettings
	PrepareQuery(ctx context.Context, queryString string, validateOnly bool) (*PrepareQueryResult, error)
	UpdateAccountSettings(
		ctx context.Context, queryPricingModel string, maxQueryTCU *int32, queryCompute *QueryComputeUpdate,
	) (AccountSettings, error)
}

type TagWriteBackend interface {
	TagResource(arn string, tags map[string]string) error
	UntagResource(arn string, tagKeys []string) error
}

type TargetDestinationForList struct {
	TimestreamDestination *TimestreamDestinationForList `json:"TimestreamDestination,omitempty"`
}

type TimeSeriesDataPoint struct {
	Time  string `json:"Time"`
	Value Datum  `json:"Value"`
}

type TimestreamDestinationForList struct {
	DatabaseName string `json:"DatabaseName,omitempty"`
	TableName    string `json:"TableName,omitempty"`
}