Как вернуть подсписок объектов Generic типа?

Question

[Павло Пономаренко]

Добрый день. Вопрос пока более теоретический. Допустим у нас есть иерархия

class Vehicle {}
class Light  : Vehicle {}
class Medium : Vehicle {}
class Heavy  : Vehicle {}

Так же есть коллекция:

class VehicleCollection {
  
  private List<Vehicle> vehicles = new List<Vehicle>();

  public void AddVehicle (Vehicle vehicle) {
    vehicles.Add( vehicle );
  }

  public void List<T> GetAllByType() where T : Vehicle {
    List<T> list = new List<T>();
    // todo
    return list;
  }
}

Подскажите, пожалуйста, как корректно написать метод, чтобы он возвращал массив экземпляров подтипа, при этом чтобы можно было передать только один из подклассов (скажем, RocketMan должен валиться на этапе компиляции). Как такое написать и на C# и на Java? Насколько такой код хорошо оптимизируется компилятором?

List<Light> lightVehicles = vehicleCollection.GetAllByType<Light>();

Самый простой пример подобного использования, который я знаю - GetComponent<>() в юнити:

Rectangle rectangle = GetComponent<Rectangle>();

Answer 1

[dordzhiev]

Подскажите, пожалуйста, как корректно написать метод, чтобы он возвращал массив экземпляров подтипа,

public void List<T> GetAllByType() where T : Vehicle 
  {
    return vehicles.OfType<T>().ToList();
  }

при этом чтобы можно было передать только один из подклассов (скажем, RocketMan должен валиться на этапе компиляции)

За счет where T : Vehicle он и так не даст скомпилировать.

Comments

[Павло Пономаренко]

Спасибо, очень клево. Но как разобрать OfType? Что внутри? В том смысле, вызвать библиотечный метод очень удобно для разработки, но неудобно для понимания внутренностей.

[Рифат]

TheShock:

public static IEnumerable<TResult> OfType<TResult>(this IEnumerable source) {
    if (source == null) throw Error.ArgumentNull("source");
    return OfTypeIterator<TResult>(source);
}
 
static IEnumerable<TResult> OfTypeIterator<TResult>(IEnumerable source) {
    foreach (object obj in source) {
        if (obj is TResult) yield return (TResult)obj;
    }
}

Ссылка

[dordzhiev]

TheShock: Почти весь код .NET Framework открыт. Большая часть уже на гитхабе ( https://github.com/dotnet https://github.com/aspnet и еще несколько). Все остальное можно найти на referencesource.microsoft.com .
Реализация OfType() лежит здесь https://github.com/dotnet/corefx/blob/master/src/S... .

Answer 2

[SaertiS]

по части Java: <T extends Venichle> в сигнатуру метода добавить и при формировании выдачи проверять на инстанс.

ps. <? extends Venichle> конечно же )

Comments

[Павло Пономаренко]

public <T extends Vehicle> void List<T> GetAllByType() {
    List<T> list = new List<T>();
    for (Vehicle item : vehicles) {
      if ( item instanceof T ) {
        list.Add( (T) item );
      }
    }
    return list;
}

Такое? Есть мнение, что instanceof не работает с Generic типами.

[SaertiS]

TheShock: примерно все же актуальнее vehicles.stream().filter().map.collect(), джава 9 все же на носу. по поводу инстанса ,он прекрасно работает , в сигнатуре следует писать ? extends Vehicle

[Павло Пономаренко]

Не совсем понимаю, как использовать ? в таком примере.

[Павло Пономаренко]

т.е. <? extends Venichle>