store

type InternalAggregationGroup struct {
	GroupValues []*commonpb.Payload
	Count       int64
}

type InternalCountExecutionsResponse struct {
	Count  int64
	Groups []InternalAggregationGroup
}

type InternalExecutionInfo struct {
	WorkflowID           string
	RunID                string
	TypeName             string
	StartTime            time.Time
	ExecutionTime        time.Time
	CloseTime            time.Time
	ExecutionDuration    time.Duration
	Status               enumspb.WorkflowExecutionStatus
	HistoryLength        int64
	HistorySizeBytes     int64
	StateTransitionCount int64
	Memo                 *commonpb.DataBlob
	TaskQueue            string
	SearchAttributes     *commonpb.SearchAttributes
	ParentWorkflowID     string
	ParentRunID          string
	RootWorkflowID       string
	RootRunID            string
}

type InternalGetWorkflowExecutionResponse struct {
	Execution *InternalExecutionInfo
}

type InternalListExecutionsResponse struct {
	Executions []*InternalExecutionInfo
	// 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 InternalRecordWorkflowExecutionClosedRequest struct {
	*InternalVisibilityRequestBase
	CloseTime            time.Time
	HistoryLength        int64
	HistorySizeBytes     int64
	ExecutionDuration    time.Duration
	StateTransitionCount int64
}

type InternalRecordWorkflowExecutionStartedRequest struct {
	*InternalVisibilityRequestBase
}

type InternalUpsertWorkflowExecutionRequest struct {
	*InternalVisibilityRequestBase
}

type InternalVisibilityRequestBase struct {
	NamespaceID      string
	WorkflowID       string
	RunID            string
	WorkflowTypeName string
	StartTime        time.Time
	Status           enumspb.WorkflowExecutionStatus
	ExecutionTime    time.Time
	TaskID           int64
	ShardID          int32
	Memo             *commonpb.DataBlob
	TaskQueue        string
	SearchAttributes *commonpb.SearchAttributes
	ParentWorkflowID *string
	ParentRunID      *string
	RootWorkflowID   string
	RootRunID        string
}

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

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

type VisibilityStore interface {
	persistence.Closeable
	GetName() string
	GetIndexName() string

	// Validate search attributes based on the store constraints. It returns a new map containing
	// only search attributes with valid values. If there are invalid values, it returns error of type
	// serviceerror.InvalidArgument.
	ValidateCustomSearchAttributes(searchAttributes map[string]any) (map[string]any, error)

	// Write APIs.
	RecordWorkflowExecutionStarted(ctx context.Context, request *InternalRecordWorkflowExecutionStartedRequest) error
	RecordWorkflowExecutionClosed(ctx context.Context, request *InternalRecordWorkflowExecutionClosedRequest) error
	UpsertWorkflowExecution(ctx context.Context, request *InternalUpsertWorkflowExecutionRequest) error
	DeleteWorkflowExecution(ctx context.Context, request *manager.VisibilityDeleteWorkflowExecutionRequest) error

	// Read APIs.
	ListWorkflowExecutions(ctx context.Context, request *manager.ListWorkflowExecutionsRequestV2) (*InternalListExecutionsResponse, error)
	ListChasmExecutions(ctx context.Context, request *manager.ListChasmExecutionsRequest) (*InternalListExecutionsResponse, error)
	CountWorkflowExecutions(ctx context.Context, request *manager.CountWorkflowExecutionsRequest) (*InternalCountExecutionsResponse, error)
	CountChasmExecutions(ctx context.Context, request *manager.CountChasmExecutionsRequest) (*InternalCountExecutionsResponse, error)
	GetWorkflowExecution(ctx context.Context, request *manager.GetWorkflowExecutionRequest) (*InternalGetWorkflowExecutionResponse, error)

	// AddSearchAttributes makes schema changes to add the search attributes. This function must be
	// idempotent, ie., if a search attribute already exists, this function must be no-op, and must
	// not return any error.
	AddSearchAttributes(ctx context.Context, request *manager.AddSearchAttributesRequest) error
}