persistence

type ActivityInfo struct {
	ScheduleID               int64
	ScheduledEvent           []byte
	StartedID                int64
	StartedEvent             []byte
	ActivityID               string
	RequestID                string
	Details                  []byte
	ScheduleToStartTimeout   int32
	ScheduleToCloseTimeout   int32
	StartToCloseTimeout      int32
	HeartbeatTimeout         int32
	CancelRequested          bool
	CancelRequestID          int64
	LastHeartBeatUpdatedTime time.Time
}

type ActivityTask struct {
	TaskID     int64
	DomainID   string
	TaskList   string
	ScheduleID int64
}

type ActivityTimeoutTask struct {
	VisibilityTimestamp time.Time
	TaskID              int64
	TimeoutType         int
	EventID             int64
}

type AppendHistoryEventsRequest struct {
	DomainID      string
	Execution     workflow.WorkflowExecution
	FirstEventID  int64
	RangeID       int64
	TransactionID int64
	Events        *SerializedHistoryEventBatch
	Overwrite     bool
}

type CancelExecutionTask struct {
	TaskID           int64
	TargetDomainID   string
	TargetWorkflowID string
	TargetRunID      string
	ScheduleID       int64
}

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

type ChildExecutionInfo struct {
	InitiatedID     int64
	InitiatedEvent  []byte
	StartedID       int64
	StartedEvent    []byte
	CreateRequestID string
}

type Closeable interface {
	Close()
}

type CompleteTaskRequest struct {
	TaskList *TaskListInfo
	TaskID   int64
}

type CompleteTimerTaskRequest struct {
	VisibilityTimestamp time.Time
	TaskID              int64
}

type CompleteTransferTaskRequest struct {
	TaskID int64
}

type ConditionFailedError struct {
	Msg string
}

type CreateDomainRequest struct {
	Name        string
	Status      int
	Description string
	OwnerEmail  string
	Retention   int32
	EmitMetric  bool
}

type CreateDomainResponse struct {
	ID string
}

type CreateShardRequest struct {
	ShardInfo *ShardInfo
}

type CreateTaskInfo struct {
	Execution workflow.WorkflowExecution
	Data      *TaskInfo
	TaskID    int64
}

type CreateTasksRequest struct {
	DomainID     string
	TaskList     string
	TaskListType int
	RangeID      int64
	Tasks        []*CreateTaskInfo
}

type CreateTasksResponse struct {
}

type CreateWorkflowExecutionRequest struct {
	RequestID                   string
	DomainID                    string
	Execution                   workflow.WorkflowExecution
	ParentDomainID              string
	ParentExecution             *workflow.WorkflowExecution
	InitiatedID                 int64
	TaskList                    string
	WorkflowTypeName            string
	DecisionTimeoutValue        int32
	ExecutionContext            []byte
	NextEventID                 int64
	LastProcessedEvent          int64
	TransferTasks               []Task
	TimerTasks                  []Task
	RangeID                     int64
	DecisionScheduleID          int64
	DecisionStartedID           int64
	DecisionStartToCloseTimeout int32
	ContinueAsNew               bool
}

type CreateWorkflowExecutionResponse struct {
	TaskID string
}

type DecisionTask struct {
	TaskID     int64
	DomainID   string
	TaskList   string
	ScheduleID int64
}

type DecisionTimeoutTask struct {
	VisibilityTimestamp time.Time
	TaskID              int64
	EventID             int64
}

type DeleteDomainByNameRequest struct {
	Name string
}

type DeleteDomainRequest struct {
	ID string
}

type DeleteExecutionTask struct {
	TaskID int64
}

type DeleteHistoryEventTask struct {
	VisibilityTimestamp time.Time
	TaskID              int64
}

type DeleteWorkflowExecutionHistoryRequest struct {
	DomainID  string
	Execution workflow.WorkflowExecution
}

type DeleteWorkflowExecutionRequest struct {
	ExecutionInfo *WorkflowExecutionInfo
}

type DomainConfig struct {
	Retention  int32
	EmitMetric bool
}

type DomainInfo struct {
	ID          string
	Name        string
	Status      int
	Description string
	OwnerEmail  string
}

type ExecutionManager interface {
	Closeable
	CreateWorkflowExecution(request *CreateWorkflowExecutionRequest) (*CreateWorkflowExecutionResponse, error)
	GetWorkflowExecution(request *GetWorkflowExecutionRequest) (*GetWorkflowExecutionResponse, error)
	UpdateWorkflowExecution(request *UpdateWorkflowExecutionRequest) error
	DeleteWorkflowExecution(request *DeleteWorkflowExecutionRequest) error
	GetCurrentExecution(request *GetCurrentExecutionRequest) (*GetCurrentExecutionResponse, error)
	GetTransferTasks(request *GetTransferTasksRequest) (*GetTransferTasksResponse, error)
	CompleteTransferTask(request *CompleteTransferTaskRequest) error

	// Timer related methods.
	GetTimerIndexTasks(request *GetTimerIndexTasksRequest) (*GetTimerIndexTasksResponse, error)
	CompleteTimerTask(request *CompleteTimerTaskRequest) error
}

type ExecutionManagerFactory interface {
	CreateExecutionManager(shardID int) (ExecutionManager, error)
}

type GetClosedWorkflowExecutionRequest struct {
	DomainUUID string
	Execution  s.WorkflowExecution
}

type GetClosedWorkflowExecutionResponse struct {
	Execution *s.WorkflowExecutionInfo
}

type GetCurrentExecutionRequest struct {
	DomainID   string
	WorkflowID string
}

type GetCurrentExecutionResponse struct {
	RunID string
}

type GetDomainRequest struct {
	ID   string
	Name string
}

type GetDomainResponse struct {
	Info   *DomainInfo
	Config *DomainConfig
}

type GetShardRequest struct {
	ShardID int
}

type GetShardResponse struct {
	ShardInfo *ShardInfo
}

type GetTasksRequest struct {
	DomainID     string
	TaskList     string
	TaskType     int
	ReadLevel    int64
	MaxReadLevel int64 // inclusive
	BatchSize    int
	RangeID      int64
}

type GetTasksResponse struct {
	Tasks []*TaskInfo
}

type GetTimerIndexTasksRequest struct {
	MinTimestamp time.Time
	MaxTimestamp time.Time
	BatchSize    int
}

type GetTimerIndexTasksResponse struct {
	Timers []*TimerTaskInfo
}

type GetTransferTasksRequest struct {
	ReadLevel    int64
	MaxReadLevel int64
	BatchSize    int
}

type GetTransferTasksResponse struct {
	Tasks []*TransferTaskInfo
}

type GetWorkflowExecutionHistoryRequest struct {
	DomainID  string
	Execution workflow.WorkflowExecution
	// Get the history events upto NextEventID.  Not Inclusive.
	NextEventID int64
	// Maximum number of history append transactions per page
	PageSize int
	// Token to continue reading next page of history append transactions.  Pass in empty slice for first page
	NextPageToken []byte
}

type GetWorkflowExecutionHistoryResponse struct {
	// Slice of history append transaction batches
	Events []SerializedHistoryEventBatch
	// Token to read next page if there are more events beyond page size.
	// Use this to set NextPageToken on GetworkflowExecutionHistoryRequest to read the next page.
	NextPageToken []byte
}

type GetWorkflowExecutionRequest struct {
	DomainID  string
	Execution workflow.WorkflowExecution
}

type GetWorkflowExecutionResponse struct {
	State *WorkflowMutableState
}

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

type HistoryEventBatch struct {
	Version int
	Events  []*workflow.HistoryEvent
}

type HistoryManager interface {
	Closeable
	AppendHistoryEvents(request *AppendHistoryEventsRequest) error
	// GetWorkflowExecutionHistory retrieves the paginated list of history events for given execution
	GetWorkflowExecutionHistory(request *GetWorkflowExecutionHistoryRequest) (*GetWorkflowExecutionHistoryResponse,
		error)
	DeleteWorkflowExecutionHistory(request *DeleteWorkflowExecutionHistoryRequest) error
}

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

type HistorySerializer interface {
	Serialize(batch *HistoryEventBatch) (*SerializedHistoryEventBatch, error)
	Deserialize(batch *SerializedHistoryEventBatch) (*HistoryEventBatch, error)
}

type HistorySerializerFactory interface {
	// Get returns a history serializer corresponding
	// to a given encoding type
	Get(encodingType common.EncodingType) (HistorySerializer, error)
}

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

type LeaseTaskListRequest struct {
	DomainID string
	TaskList string
	TaskType int
}

type LeaseTaskListResponse struct {
	TaskListInfo *TaskListInfo
}

type ListClosedWorkflowExecutionsByStatusRequest struct {
	ListWorkflowExecutionsRequest
	Status s.WorkflowExecutionCloseStatus
}

type ListWorkflowExecutionsByTypeRequest struct {
	ListWorkflowExecutionsRequest
	WorkflowTypeName string
}

type ListWorkflowExecutionsByWorkflowIDRequest struct {
	ListWorkflowExecutionsRequest
	WorkflowID string
}

type ListWorkflowExecutionsRequest struct {
	DomainUUID        string
	EarliestStartTime int64
	LatestStartTime   int64
	// Maximum number of workflow executions per page
	PageSize int
	// Token to continue reading next page of workflow executions.
	// Pass in empty slice for first page.
	NextPageToken []byte
}

type ListWorkflowExecutionsResponse struct {
	Executions []*s.WorkflowExecutionInfo
	// Token to read next page if there are more workflow executions beyond page size.
	// Use this to set NextPageToken on ListWorkflowExecutionsRequest to read the next page.
	NextPageToken []byte
}

type MetadataManager interface {
	Closeable
	CreateDomain(request *CreateDomainRequest) (*CreateDomainResponse, error)
	GetDomain(request *GetDomainRequest) (*GetDomainResponse, error)
	UpdateDomain(request *UpdateDomainRequest) error
	DeleteDomain(request *DeleteDomainRequest) error
	DeleteDomainByName(request *DeleteDomainByNameRequest) error
}

type RecordWorkflowExecutionClosedRequest struct {
	DomainUUID       string
	Execution        s.WorkflowExecution
	WorkflowTypeName string
	StartTimestamp   int64
	CloseTimestamp   int64
	Status           s.WorkflowExecutionCloseStatus
	HistoryLength    int64
	RetentionSeconds int64
}

type RecordWorkflowExecutionStartedRequest struct {
	DomainUUID       string
	Execution        s.WorkflowExecution
	WorkflowTypeName string
	StartTimestamp   int64
}

type RequestCancelInfo struct {
	InitiatedID     int64
	CancelRequestID string
}

type SerializedHistoryEventBatch struct {
	EncodingType common.EncodingType
	Version      int
	Data         []byte
}

type ShardAlreadyExistError struct {
	Msg string
}

type ShardInfo struct {
	ShardID          int
	Owner            string
	RangeID          int64
	StolenSinceRenew int
	UpdatedAt        time.Time
	TransferAckLevel int64
	TimerAckLevel    time.Time
}

type ShardManager interface {
	Closeable
	CreateShard(request *CreateShardRequest) error
	GetShard(request *GetShardRequest) (*GetShardResponse, error)
	UpdateShard(request *UpdateShardRequest) error
}

type ShardOwnershipLostError struct {
	ShardID int
	Msg     string
}

type StartChildExecutionTask struct {
	TaskID           int64
	TargetDomainID   string
	TargetWorkflowID string
	InitiatedID      int64
}

type Task interface {
	GetType() int
	GetTaskID() int64
	SetTaskID(id int64)
}

type TaskInfo struct {
	DomainID               string
	WorkflowID             string
	RunID                  string
	TaskID                 int64
	ScheduleID             int64
	ScheduleToStartTimeout int32
}

type TaskListInfo struct {
	DomainID string
	Name     string
	TaskType int
	RangeID  int64
	AckLevel int64
}

type TaskManager interface {
	Closeable
	LeaseTaskList(request *LeaseTaskListRequest) (*LeaseTaskListResponse, error)
	UpdateTaskList(request *UpdateTaskListRequest) (*UpdateTaskListResponse, error)
	CreateTasks(request *CreateTasksRequest) (*CreateTasksResponse, error)
	GetTasks(request *GetTasksRequest) (*GetTasksResponse, error)
	CompleteTask(request *CompleteTaskRequest) error
}

type TestBase struct {
	ShardMgr            ShardManager
	ExecutionMgrFactory ExecutionManagerFactory
	WorkflowMgr         ExecutionManager
	TaskMgr             TaskManager
	HistoryMgr          HistoryManager
	MetadataManager     MetadataManager
	VisibilityMgr       VisibilityManager
	ShardInfo           *ShardInfo
	ShardContext        *TestShardContext

	CassandraTestCluster
	// contains filtered or unexported fields
}

type TestBaseOptions struct {
	ClusterHost  string
	KeySpace     string
	Datacenter   string
	DropKeySpace bool
	SchemaDir    string
}

type TestShardContext struct {
	sync.RWMutex
	// contains filtered or unexported fields
}

type TimeoutError struct {
	Msg string
}

type TimerInfo struct {
	TimerID    string
	StartedID  int64
	ExpiryTime time.Time
	TaskID     int64
}

type TimerTaskInfo struct {
	DomainID            string
	WorkflowID          string
	RunID               string
	VisibilityTimestamp time.Time
	TaskID              int64
	TaskType            int
	TimeoutType         int
	EventID             int64
}

type TransferTaskInfo struct {
	DomainID         string
	WorkflowID       string
	RunID            string
	TaskID           int64
	TargetDomainID   string
	TargetWorkflowID string
	TargetRunID      string
	TaskList         string
	TaskType         int
	ScheduleID       int64
}

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

type UpdateDomainRequest struct {
	Info   *DomainInfo
	Config *DomainConfig
}

type UpdateShardRequest struct {
	ShardInfo       *ShardInfo
	PreviousRangeID int64
}

type UpdateTaskListRequest struct {
	TaskListInfo *TaskListInfo
}

type UpdateTaskListResponse struct {
}

type UpdateWorkflowExecutionRequest struct {
	ExecutionInfo   *WorkflowExecutionInfo
	TransferTasks   []Task
	TimerTasks      []Task
	DeleteTimerTask Task
	Condition       int64
	RangeID         int64
	ContinueAsNew   *CreateWorkflowExecutionRequest
	CloseExecution  bool

	// Mutable state
	UpsertActivityInfos       []*ActivityInfo
	DeleteActivityInfo        *int64
	UpserTimerInfos           []*TimerInfo
	DeleteTimerInfos          []string
	UpsertChildExecutionInfos []*ChildExecutionInfo
	DeleteChildExecutionInfo  *int64
	UpsertRequestCancelInfos  []*RequestCancelInfo
	DeleteRequestCancelInfo   *int64
}

type UserTimerTask struct {
	VisibilityTimestamp time.Time
	TaskID              int64
	EventID             int64
}

type VisibilityManager interface {
	Closeable
	RecordWorkflowExecutionStarted(request *RecordWorkflowExecutionStartedRequest) error
	RecordWorkflowExecutionClosed(request *RecordWorkflowExecutionClosedRequest) error
	ListOpenWorkflowExecutions(request *ListWorkflowExecutionsRequest) (*ListWorkflowExecutionsResponse, error)
	ListClosedWorkflowExecutions(request *ListWorkflowExecutionsRequest) (*ListWorkflowExecutionsResponse, error)
	ListOpenWorkflowExecutionsByType(request *ListWorkflowExecutionsByTypeRequest) (*ListWorkflowExecutionsResponse, error)
	ListClosedWorkflowExecutionsByType(request *ListWorkflowExecutionsByTypeRequest) (*ListWorkflowExecutionsResponse, error)
	ListOpenWorkflowExecutionsByWorkflowID(request *ListWorkflowExecutionsByWorkflowIDRequest) (*ListWorkflowExecutionsResponse, error)
	ListClosedWorkflowExecutionsByWorkflowID(request *ListWorkflowExecutionsByWorkflowIDRequest) (*ListWorkflowExecutionsResponse, error)
	ListClosedWorkflowExecutionsByStatus(request *ListClosedWorkflowExecutionsByStatusRequest) (*ListWorkflowExecutionsResponse, error)
	GetClosedWorkflowExecution(request *GetClosedWorkflowExecutionRequest) (*GetClosedWorkflowExecutionResponse, error)
}

type WorkflowExecutionInfo struct {
	DomainID             string
	WorkflowID           string
	RunID                string
	ParentDomainID       string
	ParentWorkflowID     string
	ParentRunID          string
	InitiatedID          int64
	CompletionEvent      []byte
	TaskList             string
	WorkflowTypeName     string
	DecisionTimeoutValue int32
	ExecutionContext     []byte
	State                int
	CloseStatus          int
	NextEventID          int64
	LastProcessedEvent   int64
	StartTimestamp       time.Time
	LastUpdatedTimestamp time.Time
	CreateRequestID      string
	DecisionScheduleID   int64
	DecisionStartedID    int64
	DecisionRequestID    string
	DecisionTimeout      int32
	CancelRequested      bool
	CancelRequestID      string
}

type WorkflowMutableState struct {
	ActivitInfos        map[int64]*ActivityInfo
	TimerInfos          map[string]*TimerInfo
	ChildExecutionInfos map[int64]*ChildExecutionInfo
	RequestCancelInfos  map[int64]*RequestCancelInfo
	ExecutionInfo       *WorkflowExecutionInfo
}

type WorkflowTimeoutTask struct {
	VisibilityTimestamp time.Time
	TaskID              int64
}