kafkaconnect

type ApacheKafkaCluster struct {
	BootstrapServers string
	Vpc              Vpc
}

type AutoScaling struct {
	MinWorkerCount          int32
	MaxWorkerCount          int32
	McuCount                int32
	MaxAutoscalingTaskCount int32
	ScaleInCPUPercent       int32
	ScaleOutCPUPercent      int32
}

type Capacity struct {
	AutoScaling *AutoScaling
	Provisioned *ProvisionedCapacity
}

type CloudWatchLogsDelivery struct {
	LogGroup string
	Enabled  bool
}

type Connector struct {
	CreationTime                     time.Time
	PendingUntil                     time.Time
	Capacity                         Capacity
	WorkerConfiguration              *WorkerConfigRef
	WorkerLogDelivery                *WorkerLogDelivery
	ConnectorConfiguration           map[string]string
	Tags                             map[string]string
	State                            string
	Description                      string
	ARN                              string
	CurrentVersion                   string
	KafkaConnectVersion              string
	KafkaClusterClientAuthentication string
	KafkaClusterEncryptionInTransit  string
	ServiceExecutionRoleArn          string
	NetworkType                      string
	Name                             string
	ApacheKafkaCluster               ApacheKafkaCluster
	Plugins                          []PluginRef
}

type ConnectorOperation struct {
	CreationTime                 time.Time
	EndTime                      time.Time
	OriginConnectorConfiguration map[string]string
	TargetConnectorConfiguration map[string]string
	OriginCapacity               *Capacity
	TargetCapacity               *Capacity
	ARN                          string
	ConnectorArn                 string
	State                        string
	Type                         string
	Steps                        []ConnectorOperationStep
}

type ConnectorOperationStep struct {
	StepType  string
	StepState string
}

type ConnectorSpec struct {
	Capacity                         Capacity
	ConnectorConfiguration           map[string]string
	Tags                             map[string]string
	WorkerConfiguration              *WorkerConfigRef
	WorkerLogDelivery                *WorkerLogDelivery
	Description                      string
	Name                             string
	KafkaConnectVersion              string
	KafkaClusterClientAuthentication string
	KafkaClusterEncryptionInTransit  string
	ServiceExecutionRoleArn          string
	NetworkType                      string
	ApacheKafkaCluster               ApacheKafkaCluster
	Plugins                          []PluginRef
}

type ConnectorUpdate struct {
	Capacity               *Capacity
	ConnectorConfiguration map[string]string
}

type CustomPlugin struct {
	CreationTime   time.Time
	PendingUntil   time.Time
	Tags           map[string]string
	Name           string
	ARN            string
	Description    string
	State          string
	FailureMessage string
	ContentType    string
	BucketArn      string
	FileKey        string
	ObjectVersion  string
	FileMD5        string
	FileSizeBytes  int64
	Revision       int64
}

type FirehoseDelivery struct {
	DeliveryStream string
	Enabled        bool
}

type Handler struct {
	Backend StorageBackend

	AccountID     string
	DefaultRegion string
	// contains filtered or unexported fields
}

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

type PluginRef struct {
	CustomPluginArn string
	Revision        int64
}

type Provider struct{}

type ProvisionedCapacity struct {
	McuCount    int32
	WorkerCount int32
}

type S3LogDelivery struct {
	Bucket  string
	Prefix  string
	Enabled bool
}

type Snapshottable interface {
	Snapshot(ctx context.Context) []byte
	Restore(context.Context, []byte) error
}

type StorageBackend interface {
	CreateConnector(accountID, region string, spec ConnectorSpec) (*Connector, error)
	DescribeConnector(connectorArn string) (*Connector, error)
	ListConnectors(namePrefix, nextToken string, maxResults int) ([]*Connector, string, error)
	UpdateConnector(
		connectorArn, currentVersion string,
		update ConnectorUpdate,
	) (*Connector, *ConnectorOperation, error)
	DeleteConnector(connectorArn, currentVersion string) (*Connector, error)
	RestartConnector(connectorArn string, onlyFailedTasks bool) (*Connector, *ConnectorOperation, error)
	DescribeConnectorOperation(operationArn string) (*ConnectorOperation, error)
	ListConnectorOperations(connectorArn, nextToken string, maxResults int) ([]*ConnectorOperation, string, error)

	CreateCustomPlugin(
		accountID, region, name, description, contentType, bucketArn, fileKey, objectVersion string,
		tags map[string]string,
	) (*CustomPlugin, error)
	DescribeCustomPlugin(customPluginArn string) (*CustomPlugin, error)
	ListCustomPlugins(namePrefix, nextToken string, maxResults int) ([]*CustomPlugin, string, error)
	DeleteCustomPlugin(customPluginArn string) (*CustomPlugin, error)

	CreateWorkerConfiguration(
		accountID, region, name, description, propertiesFileContent string,
		tags map[string]string,
	) (*WorkerConfiguration, error)
	DescribeWorkerConfiguration(workerConfigurationArn string) (*WorkerConfiguration, error)
	ListWorkerConfigurations(namePrefix, nextToken string, maxResults int) ([]*WorkerConfiguration, string, error)
	DeleteWorkerConfiguration(workerConfigurationArn string) (*WorkerConfiguration, error)

	TagResource(resourceArn string, tags map[string]string) error
	UntagResource(resourceArn string, tagKeys []string) error
	ListTagsForResource(resourceArn string) (map[string]string, error)

	Reset()
}

type Vpc struct {
	SecurityGroups []string
	Subnets        []string
}

type WorkerConfigRef struct {
	Arn      string
	Revision int64
}

type WorkerConfigRevision struct {
	CreationTime          time.Time
	Description           string
	PropertiesFileContent string
	Revision              int64
}

type WorkerConfiguration struct {
	CreationTime   time.Time
	PendingUntil   time.Time
	Tags           map[string]string
	Name           string
	ARN            string
	Description    string
	State          string
	LatestRevision WorkerConfigRevision
}

type WorkerLogDelivery struct {
	CloudWatchLogs *CloudWatchLogsDelivery
	Firehose       *FirehoseDelivery
	S3             *S3LogDelivery
}