Рекурсивный подсчет

Контейнер Мар в PetCount3.PetCounter был заполнен всеми классами Pet. Вместо предварительного заполнения карты мы также можем воспользоваться методом Class.isAssignableFrom() и создать обобщенный инструмент подсчета, не ограни­ченный подсчетом Pet:

//: net/mindview/util/TypeCounter.java

// Подсчет экземпляров в семействе типов

package net.mindview.util;

import java.util.*;

public class TypeCounter extends HashMap<Class<?>,Integer>{

private Class<?> baseType;

public TypeCounter(Class<?> baseType) {

this.baseType = baseType;

}

public void count(Object obj) {

Class<?> type = obj.getClass();

if(!baseType.isAssignableFrom(type))

throw new RuntimeException(obj + " incorrect type: "

+ type + ", should be type or subtype of "

+ baseType);

countClass(type);

}

private void countClass(Class<?> type) {

Integer quantity = get(type);

put(type, quantity == null ? 1 : quantity + 1);

Class<?> superClass = type.getSuperclass();

if(superClass != null &&

baseType.isAssignableFrom(superClass))

countClass(superClass);

}

public String toString() {

StringBuilder result = new StringBuilder("{");

for(Map.Entry<Class<?>,Integer> pair : entrySet()) {

result.append(pair.getKey().getSimpleName());

result.append("=");

result.append(pair.getValue());

result.append(", ");

}

result.delete(result.length()-2, result.length());

result.append("}");

return result.toString();

}

}

Метод count() получает Class для своего аргумента, а затем использует .isAssignableFrom() для проверки принадлежности объекта к интересующей вас иерархии. Метод countClass() сначала производит подсчет для точного типа класса, а затем, если baseType допускает присваивание из суперкласса, рекурсивно вызывает countClass() для суперкласса.

//: typeinfo/PetCount4.java

import typeinfo.pets.*;

import net.mindview.util.*;

import static net.mindview.util.Print.*;

public class PetCount4 {

public static void main(String[] args) {

TypeCounter counter = new TypeCounter(Pet.class);

for(Pet pet : Pets.createArray(20)) {

printnb(pet.getClass().getSimpleName() + " ");

counter.count(pet);

}

print();

print(counter);

}

}

<spoiler text="Output:"> (Пример)

Rat Manx Cymric Mutt Pug Cymric Pug Manx Cymric Rat EgyptianMau Hamster EgyptianMau Mutt Mutt

Cymric Mouse Pug Mouse Cymric

{Mouse=2, Dog=6, Manx=7, EgyptianMau=2, Rodent=5, Pug=3, Mutt=3. Cymric=5, Cat=9. Hamster=l,

Pet=20, Rat=2}

</spoiler> Как видно из результатов, подсчитываются как базовые, так и конкретные типы.