Отримати розмір Iterable в Java

Мені потрібно з’ясувати кількість елементів Iterableу Java. Я знаю, що можу це зробити:

Iterable values = ...
it = values.iterator();
while (it.hasNext()) {
  it.next();
  sum++;
}

Я також міг би зробити щось подібне, тому що мені більше не потрібні об'єкти в Iterable:

it = values.iterator();
while (it.hasNext()) {
  it.remove();
  sum++;
}

Невеликий орієнтир не показав великої різниці у продуктивності, коментарів чи інших ідей щодо цієї проблеми?

TL; DR: Використовуйте метод корисності Iterables.size(Iterable)великої бібліотеки Гуави .

З ваших двох фрагментів коду ви повинні використовувати перший, оскільки другий видалить усі елементи values, тому він порожній. Зміна структури даних для такого простого запиту, як його розмір, є дуже несподіваною.

Для продуктивності це залежить від вашої структури даних. Якщо це, наприклад, насправді an ArrayList, видалення елементів з самого початку (що робить ваш другий метод) відбувається дуже повільно (обчислення розміру стає O (n * n) замість O (n), як це повинно бути).

Загалом, якщо є ймовірність, що valuesнасправді є, Collectionа не лише Iterable, перевірте це та зателефонуйте size()на випадок:

if (values instanceof Collection<?>) {
  return ((Collection<?>)values).size();
}
// use Iterator here...

Виклик size(), як правило, набагато швидше , ніж підрахунок кількості елементів, і цей трюк саме Iterables.size(Iterable)з гуави робить для вас.

Якщо ви працюєте з java 8, ви можете використовувати:

Iterable values = ...
long size = values.spliterator().getExactSizeIfKnown();

це спрацює лише в тому випадку, якщо ітерабельне джерело має визначений розмір. Більшість Сплітераторів для колекцій будуть, але у вас можуть виникнути проблеми, якщо вони походять від HashSetабо, ResultSetнаприклад.

Ви можете перевірити javadoc тут.

Якщо Java 8 не є опцією або якщо ви не знаєте, звідки береться ітерація, ви можете використовувати той самий підхід, що і гуава:

  if (iterable instanceof Collection) {
        return ((Collection<?>) iterable).size();
    } else {
        int count = 0;
        Iterator iterator = iterable.iterator();
        while(iterator.hasNext()) {
            iterator.next();
            count++;
        }
        return count;
    }

Це, можливо, трохи пізно, але може комусь допомогти. Я стикаюся з подібною проблемою Iterableу своїй кодовій базі, і рішення було використовувати for eachбез явного виклику values.iterator();.

int size = 0;
for(T value : values) {
   size++;
}

Ви можете додати свій ітерабель до списку, а потім використовувати .size () на ньому.

Lists.newArrayList(iterable).size();

Для наочності для наведеного методу буде потрібно наступне імпортування:

import com.google.common.collect.Lists;

Власне кажучи, Iterable не має розміру. Думайте про структуру даних як про цикл.

І подумайте про наступний екземпляр Iterable, No size:

    new Iterable(){

        @Override public Iterator iterator() {
            return new Iterator(){

                @Override
                public boolean hasNext() {
                    return isExternalSystemAvailble();
                }

                @Override
                public Object next() {
                    return fetchDataFromExternalSystem();
                }};
        }};

Я б пішов з it.next()тієї простої причини, яка next()гарантовано буде реалізована, хоча remove()це необов’язкова операція.

E next()

Повертає наступний елемент в ітерації.

void remove()

Видаляє з базової колекції останній елемент, повернутий ітератором (необов’язкова операція) .

java 8 і вище

StreamSupport.stream(data.spliterator(), false).count();

Як на мене, це просто різні методи. Перший залишає об’єкт, для якого ви ітерацію, незмінним, а секунди залишає його порожнім. Питання в тому, що ти хочеш робити. Складність видалення заснована на реалізації вашого ітерабельного об'єкта. Якщо ви використовуєте колекції - просто отримайте розмір, запропонований Kazekage Gaara, - як правило, це найкращий підхід.

Чому б вам просто не використовувати size()метод на своєму, Collectionщоб отримати кількість елементів?

Iterator просто призначений для ітерації, нічого іншого.

Замість того, щоб використовувати цикли та підраховувати кожен елемент, або за допомогою сторонніх бібліотек, ми можемо просто ввести в ідентифікатор ітерабель в ArrayList і отримати його розмір.

((ArrayList) iterable).size();