repository

type CompensationRecord struct {
	ID               string             `json:"id" bson:"_id"`
	WorkflowID       string             `json:"workflowId" bson:"workflowId"`
	RunID            string             `json:"runId,omitempty" bson:"runId,omitempty"`
	ActivityName     string             `json:"activityName" bson:"activityName"`
	Status           CompensationStatus `json:"status" bson:"status"`
	Error            string             `json:"error,omitempty" bson:"error,omitempty"`
	StartedAt        time.Time          `json:"startedAt,omitempty" bson:"startedAt,omitempty"`
	CompletedAt      time.Time          `json:"completedAt,omitempty" bson:"completedAt,omitempty"`
	Duration         time.Duration      `json:"duration,omitempty" bson:"duration,omitempty"`
	Retries          int                `json:"retries" bson:"retries"`
	CompensationData json.RawMessage    `json:"compensationData,omitempty" bson:"compensationData,omitempty"`
	Metadata         map[string]string  `json:"metadata,omitempty" bson:"metadata,omitempty"`
	CreatedAt        time.Time          `json:"createdAt" bson:"createdAt"`
	UpdatedAt        time.Time          `json:"updatedAt" bson:"updatedAt"`
}

type CompensationRepository interface {
	// SaveCompensationRecord creates or updates a compensation record.
	SaveCompensationRecord(ctx context.Context, record *CompensationRecord) error

	// GetCompensationRecord retrieves a compensation record by ID.
	GetCompensationRecord(ctx context.Context, recordID string) (*CompensationRecord, error)

	// GetCompensationHistory retrieves all compensation records for a workflow.
	GetCompensationHistory(ctx context.Context, workflowID string) ([]CompensationRecord, error)

	// ListCompensations retrieves compensation records with filtering.
	ListCompensations(ctx context.Context, filter ListCompensationsFilter) ([]CompensationRecord, error)

	// MarkCompensationStarted updates record status to running.
	MarkCompensationStarted(ctx context.Context, recordID string) error

	// MarkCompensationCompleted updates record status to completed.
	MarkCompensationCompleted(ctx context.Context, recordID string) error

	// MarkCompensationFailed updates record status to failed with error.
	MarkCompensationFailed(ctx context.Context, recordID string, err error) error

	// GetCompensationSummary retrieves aggregated compensation statistics.
	GetCompensationSummary(ctx context.Context, workflowID string) (*CompensationSummary, error)

	// DeleteCompensationHistory removes all compensation records for a workflow.
	DeleteCompensationHistory(ctx context.Context, workflowID string) error

	// GetPendingCompensations retrieves all pending compensations for retry.
	GetPendingCompensations(ctx context.Context, limit int) ([]CompensationRecord, error)

	// IncrementRetryCount increments the retry counter for a compensation record.
	IncrementRetryCount(ctx context.Context, recordID string) error
}

type CompensationStatus string

type CompensationSummary struct {
	WorkflowID         string `json:"workflowId"`
	TotalCompensations int    `json:"totalCompensations"`
	Completed          int    `json:"completed"`
	Failed             int    `json:"failed"`
	Skipped            int    `json:"skipped"`
	Pending            int    `json:"pending"`
}

type ListCompensationsFilter struct {
	WorkflowID   string
	ActivityName string
	Status       CompensationStatus
	StartTime    time.Time
	EndTime      time.Time
	Limit        int
	Offset       int
}

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