types

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

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

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

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"`
	ActiveClusterSelectionPolicy         *types.ActiveClusterSelectionPolicy `yaml:"activeClusterSelectionPolicy"`

	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"`
	// TODO(active-active): Remove this once we have completely migrated to AttributeScopes
	NewActiveClustersByRegion map[string]string `yaml:"newActiveClustersByRegion"`
	// NewClusterAttributes specifies the AttributeScopes to change for the domain
	// This can be a sub-set of the total AttributeScopes for the domain
	NewClusterAttributes ClusterAttributesMap `yaml:"newClusterAttributes"`
	FailoverTimeout      *int32               `yaml:"failoverTimeoutSec"`

	// RunIDKey specifies a key to store/retrieve RunID for this operation
	RunIDKey string `yaml:"runIDKey"`

	Want Validation `yaml:"want"`
}

type OperationFunction func(t *testing.T, op *Operation, simCfg *ReplicationSimulationConfig) error

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

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

	// ClusterAttributes specifies the Attributes for a domain and is passed to ActiveClusters in the RegisterDomainRequest
	// It is a simplified expression for the AttributeScopes type
	// The format should be expressed as follows:
	// clusterAttributes: // This will be mapped to AttributeScopes
	//   region: // This is the key for a ClusterAttributeScope. All children are mapped to ClusterAttributes
	//	 	us-west1: cluster0 // us-west1 is the key within a ClusterAttributeScope
	//      us-east1: cluster1 // cluster1 is the ActiveClusterName for the corresponding ActiveClusterInfo
	//   cityID:
	//      ...
	ClusterAttributes ClusterAttributesMap `yaml:"clusterAttributes"`
}

type ReplicationSimulation struct {
	RunIDRegistry map[string]string
}

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
}