executorclient

type Client interface {
	Heartbeat(context.Context, *types.ExecutorHeartbeatRequest, ...yarpc.CallOption) (*types.ExecutorHeartbeatResponse, error)
}

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

	// IsOnboardedToSD is returning true if the executor relies on SD for distribution
	IsOnboardedToSD() bool
}

type ExecutorMetadata map[string]string

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

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

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

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

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

type MockShardDistributorExecutorAPIYARPCClientMockRecorder 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

	ExecutorClient        Client
	MetricsScope          tally.Scope
	Logger                log.Logger
	ShardProcessorFactory ShardProcessorFactory[SP]
	Config                clientcommon.Config
	TimeSource            clock.TimeSource
	Metadata              ExecutorMetadata                 `optional:"true"`
	DrainObserver         clientcommon.DrainSignalObserver `optional:"true"`
}

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

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

type ShardReport struct {
	ShardLoad float64
	Status    types.ShardStatus
}