sweeper

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

type ConfigExpiryQuerier interface {
	// GetActiveOrganizations returns all enabled organizations
	GetActiveOrganizations(ctx context.Context) ([]configdb.GetActiveOrganizationsRow, error)

	// GetOrganizationExpiry retrieves expiry configuration for an org/signal pair
	GetOrganizationExpiry(ctx context.Context, arg configdb.GetOrganizationExpiryParams) (configdb.OrganizationSignalExpiry, error)

	// GetExpiryLastRun retrieves when expiry was last run for an org/signal pair
	GetExpiryLastRun(ctx context.Context, arg configdb.GetExpiryLastRunParams) (configdb.ExpiryRunTracking, error)

	// UpsertExpiryRunTracking creates or updates expiry run tracking
	UpsertExpiryRunTracking(ctx context.Context, arg configdb.UpsertExpiryRunTrackingParams) error
}

type InstanceKey struct {
	OrganizationID uuid.UUID
	InstanceNum    int16
}

type LRDBExpiryQuerier interface {
	// CallFindOrgPartition finds the organization's partition for a given table
	CallFindOrgPartition(ctx context.Context, arg lrdb.CallFindOrgPartitionParams) (string, error)

	// CallExpirePublishedByIngestCutoff expires data older than the cutoff date
	CallExpirePublishedByIngestCutoff(ctx context.Context, arg lrdb.CallExpirePublishedByIngestCutoffParams) (int64, error)
}

type LogCleanupWorkItem struct {
	OrganizationID   uuid.UUID
	DateInt          int32
	NextRunTime      time.Time
	ConsecutiveEmpty int
	// contains filtered or unexported fields
}

type MetricCleanupWorkItem struct {
	OrganizationID   uuid.UUID
	DateInt          int32
	NextRunTime      time.Time
	ConsecutiveEmpty int
	// contains filtered or unexported fields
}

type PubSubHistoryQuerier interface {
	PubSubMessageHistoryCleanup(ctx context.Context, params lrdb.PubSubMessageHistoryCleanupParams) (pgconn.CommandTag, error)
}

type SyncQuerier interface {
	GetAllCOrganizations(ctx context.Context) ([]configdb.GetAllCOrganizationsRow, error)
	GetAllOrganizations(ctx context.Context) ([]configdb.GetAllOrganizationsRow, error)
	UpsertOrganizationSync(ctx context.Context, arg configdb.UpsertOrganizationSyncParams) error
	DeleteOrganizationsNotInList(ctx context.Context, idsToDelete []uuid.UUID) error
	GetAllCBucketData(ctx context.Context) ([]configdb.GetAllCBucketDataRow, error)
	GetAllBucketConfigurations(ctx context.Context) ([]configdb.GetAllBucketConfigurationsRow, error)
	GetAllOrganizationBucketMappings(ctx context.Context) ([]configdb.GetAllOrganizationBucketMappingsRow, error)
	UpsertBucketConfiguration(ctx context.Context, arg configdb.UpsertBucketConfigurationParams) (configdb.BucketConfiguration, error)
	UpsertOrganizationBucket(ctx context.Context, arg configdb.UpsertOrganizationBucketParams) error
	DeleteOrganizationBucketMappings(ctx context.Context, arg configdb.DeleteOrganizationBucketMappingsParams) error
	GetAllCOrganizationAPIKeys(ctx context.Context) ([]configdb.GetAllCOrganizationAPIKeysRow, error)
	GetAllOrganizationAPIKeyMappings(ctx context.Context) ([]configdb.GetAllOrganizationAPIKeyMappingsRow, error)
	UpsertOrganizationAPIKey(ctx context.Context, arg configdb.UpsertOrganizationAPIKeyParams) (configdb.OrganizationApiKey, error)
	UpsertOrganizationAPIKeyMapping(ctx context.Context, arg configdb.UpsertOrganizationAPIKeyMappingParams) error
	DeleteOrganizationAPIKeyMappingByHash(ctx context.Context, arg configdb.DeleteOrganizationAPIKeyMappingByHashParams) error
}

type TraceCleanupWorkItem struct {
	OrganizationID   uuid.UUID
	DateInt          int32
	NextRunTime      time.Time
	ConsecutiveEmpty int
	// contains filtered or unexported fields
}

type WorkItem interface {
	// Perform executes the work and returns the duration until it should run again.
	// Negative values indicate the work item should be dropped/removed.
	Perform(ctx context.Context) time.Duration

	// GetNextRunTime returns when this work should next be executed
	GetNextRunTime() time.Time

	// SetNextRunTime updates when this work should next be executed
	SetNextRunTime(t time.Time)

	// GetKey returns a unique key for this work item (for tracking)
	GetKey() string
}

type WorkItemHeap []WorkItem

type WorkQueueCleanupQuerier interface {
	workqueue.EnqueueDB
}

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