Java 类祖先流

Question

这个问题已经有答案了:

Java8 Streaming a class hierarchy [duplicate] (1 个回答)

已关闭 5 年前。

如何创建Stream类祖先？

我能做的是:

public static Stream<Class<?>> streamOfAncestors(Class<?> klass) {
    ArrayList<Class<?>> ancestorsList = new ArrayList<>();
    while (true) {
        klass = klass.getSuperclass();
        if (klass == null) {
            return ancestorsList.stream();
        } else {
            ancestorsList.add(klass);
        }
    }
}

但它并不优雅，而且我的流也不懒惰。如何变懒Stream ？

Answer 1

一种选择是使用Stream.iterate()。您必须将其与 takeWhile(在 Java 9 中引入)结合起来，以使 Stream 有限。

public static Stream<Class<?>> streamOfAncestors(Class<?> klass) {
    return Stream.iterate (klass.getSuperclass(), k -> k != null ? k.getSuperclass() : null)
                 .takeWhile(Objects::nonNull);
}

或者，如果您不想使用 Java 9 功能，您可以编写类似于 Stream.iterate() 的方法，但使其生成有限的 Stream >:

public static Stream<Class<?>> streamOfAncestors(Class<?> klass) {
    final Iterator<Class<?>> iterator = new Iterator<Class<?>>() {
        @SuppressWarnings("unchecked")
        Class<?> k = klass.getSuperclass ();

        @Override
        public boolean hasNext() {
            return k != null;
        }

        @Override
        public Class<?> next() {
            Class<?> curr = k;
            k = k.getSuperclass ();
            return curr;
        }
    };
    return StreamSupport.stream(Spliterators.spliteratorUnknownSize(
            iterator,
            Spliterator.ORDERED | Spliterator.IMMUTABLE), false);
}

第二种方法的执行示例:

streamOfAncestors (StringBuilder.class).forEach (System.out::println);

输出:

class java.lang.AbstractStringBuilder
class java.lang.Object

编辑:

正如 Holger 评论的那样，如果您使用的是 Java 9，那么所有这些都是不必要的，Java 9 有 Stream.iterate产生有限Stream的变体。

您需要写的是:

public static Stream<Class<?>> streamOfAncestors(Class<?> klass) {
    return Stream.iterate (klass.getSuperclass(), Objects::nonNull, Class::getSuperclass);
}