executorclient

type Config struct {
	Namespaces []NamespaceConfig `yaml:"namespaces"`
}

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

	GetShardProcess(ctx context.Context, shardID string) (SP, error)
	// Used during the migration during local-passthrough and local-passthrough-shadow
	AssignShardsFromLocalLogic(ctx context.Context, shardAssignment map[string]*types.ShardAssignment)
}

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 NamespaceConfig struct {
	Namespace         string        `yaml:"namespace"`
	HeartBeatInterval time.Duration `yaml:"heartbeat_interval"`
	MigrationMode     string        `yaml:"migration_mode"`
}

type Params[SP ShardProcessor] struct {
	fx.In

	YarpcClient           sharddistributorv1.ShardDistributorExecutorAPIYARPCClient
	MetricsScope          tally.Scope
	Logger                log.Logger
	ShardProcessorFactory ShardProcessorFactory[SP]
	Config                Config
	TimeSource            clock.TimeSource
}

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

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

type ShardReport struct {
	ShardLoad float64
	Status    types.ShardStatus
}