types

type Cluster struct {
	GRPCEndpoint string `yaml:"grpcEndpoint"`

	AdminClient    admin.Client    `yaml:"-"`
	FrontendClient frontend.Client `yaml:"-"`
}

type Operation struct {
	Type          ReplicationSimulationOperation `yaml:"op"`
	At            time.Duration                  `yaml:"at"`
	Cluster       string                         `yaml:"cluster"`
	SourceCluster string                         `yaml:"sourceCluster"`
	TargetCluster string                         `yaml:"targetCluster"`

	WorkflowType                         string        `yaml:"workflowType"`
	WorkflowID                           string        `yaml:"workflowID"`
	WorkflowExecutionStartToCloseTimeout time.Duration `yaml:"workflowExecutionStartToCloseTimeout"`
	WorkflowDuration                     time.Duration `yaml:"workflowDuration"`
	ActivityCount                        int           `yaml:"activityCount"`
	DelayStartSeconds                    int32         `yaml:"delayStartSeconds"`
	CronSchedule                         string        `yaml:"cronSchedule"`

	Query            string `yaml:"query"`
	ConsistencyLevel string `yaml:"consistencyLevel"`

	SignalName  string `yaml:"signalName"`
	SignalInput any    `yaml:"signalInput"`

	EventID int64 `yaml:"eventID"`

	Domain                    string            `yaml:"domain"`
	NewActiveCluster          string            `yaml:"newActiveCluster"`
	NewActiveClustersByRegion map[string]string `yaml:"newActiveClustersByRegion"`
	FailoverTimeout           *int32            `yaml:"failoverTimeoutSec"`

	Want Validation `yaml:"want"`
}

type ReplicationDomainConfig struct {
	ActiveClusterName string `yaml:"activeClusterName"`

	ActiveClustersByRegion map[string]string `yaml:"activeClustersByRegion"`
}

type ReplicationSimulationConfig struct {
	// Clusters is the map of all clusters
	Clusters map[string]*Cluster `yaml:"clusters"`

	// PrimaryCluster is used for domain registration
	PrimaryCluster string `yaml:"primaryCluster"`

	Domains map[string]ReplicationDomainConfig `yaml:"domains"`

	Operations []*Operation `yaml:"operations"`
}

type ReplicationSimulationOperation string

type Validation struct {
	Status                      string `yaml:"status"`
	StartedByWorkersInCluster   string `yaml:"startedByWorkersInCluster"`
	CompletedByWorkersInCluster string `yaml:"completedByWorkersInCluster"`
	Error                       string `yaml:"error"`
	QueryResult                 any    `yaml:"queryResult"`
}

type WorkflowInput struct {
	Duration      time.Duration
	ActivityCount int
}

type WorkflowOutput struct {
	Count int
}