executorclient

type Executor[SP ShardProcessor] interface {
	Start(ctx context.Context)
	Stop()

	GetShardProcess(ctx context.Context, shardID string) (SP, error)

	// Get the namespace this executor is responsible for
	GetNamespace() string

	// Set metadata for the executor
	SetMetadata(metadata map[string]string)
	// Get the current metadata of the executor
	GetMetadata() map[string]string

	// AssignShardsFromLocalLogic is used for the migration during local-passthrough, local-passthrough-shadow, distributed-passthrough
	AssignShardsFromLocalLogic(ctx context.Context, shardAssignment map[string]*types.ShardAssignment) error
	// RemoveShardsFromLocalLogic is used for the migration during local-passthrough, local-passthrough-shadow, distributed-passthrough
	RemoveShardsFromLocalLogic(shardIDs []string) error
}

type ExecutorMetadata map[string]string

type MockExecutor[SP ShardProcessor] struct {
	// contains filtered or unexported fields
}

type MockExecutorMockRecorder[SP ShardProcessor] struct {
	// contains filtered or unexported fields
}

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

type MockShardProcessorFactory[SP ShardProcessor] struct {
	// contains filtered or unexported fields
}

type MockShardProcessorFactoryMockRecorder[SP ShardProcessor] struct {
	// contains filtered or unexported fields
}

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

type Params[SP ShardProcessor] struct {
	fx.In

	YarpcClient           sharddistributorv1.ShardDistributorExecutorAPIYARPCClient
	MetricsScope          tally.Scope
	Logger                log.Logger
	ShardProcessorFactory ShardProcessorFactory[SP]
	Config                clientcommon.Config
	TimeSource            clock.TimeSource
	Metadata              ExecutorMetadata `optional:"true"`
}

type ShardProcessor interface {
	Start(ctx context.Context) error
	Stop()
	GetShardReport() ShardReport
}

type ShardProcessorFactory[SP ShardProcessor] interface {
	NewShardProcessor(shardID string) (SP, error)
}

type ShardReport struct {
	ShardLoad float64
	Status    types.ShardStatus
}