【Groovy】集合遍历 ( 使用集合的 collect 循环遍历集合并根据指定闭包规则生成新集合 | 代码示例 )

韩曙亮

发布于 2023-03-30 10:09:37

2.5K0

发布于 2023-03-30 10:09:37

文章被收录于专栏：韩曙亮的移动开发专栏

文章目录

一、使用集合的 collect 循环遍历集合并根据指定闭包规则生成新集合

调用集合的 collect 方法进行遍历 , 与调用 each 方法进行遍历 , 实现的功能是不同的 ;

collect 方法主要是根据一定的转换规则 , 将现有的集合 , 转换为一个新的集合 ;

新集合是重新创建的集合 , 与原集合无关 ;

分析集合的 collect 方法 , 其传入的的参数是一个闭包 transform , 这是新生成集合的规则 ;

在该函数中调用了 collect 重载函数 collect(self, new ArrayList<T>(self.size()), transform) , 传入了新的 ArrayList 集合作为参数 , 该新的 ArrayList 集合是新创建的集合 , 其大小等于被遍历的集合 ;

    /**
     * 使用<code>transform</code>闭包遍历此集合，将每个条目转换为新值
     * 返回已转换值的列表。
     * <pre class="groovyTestCase">assert [2,4,6] == [1,2,3].collect { it * 2 }</pre>
     *
     * @param self      一个集合
     * @param transform 用于转换集合中每个项的闭包
     * @return 转换值的列表
     * @since 1.0
     */
    public static <S,T> List<T> collect(Collection<S> self, @ClosureParams(FirstParam.FirstGenericType.class) Closure<T> transform) {
        return (List<T>) collect(self, new ArrayList<T>(self.size()), transform);
    }

在重载的 collect 方法中 , 为新创建的集合赋值 , 根据 transform 闭包逻辑和原集合的值 , 计算新集合中对应位置元素的值 ;

    /**
     * 方法遍历此集合，将每个值转换为新值 <code>transform</code> 闭包
     * 并将其添加到所提供的 <code>collector</code> 中.
     * <pre class="groovyTestCase">assert [1,2,3] as HashSet == [2,4,5,6].collect(new HashSet()) { (int)(it / 2) }</pre>
     *
     * @param self      一个集合
     * @param collector 将转换值添加到其中的集合
     * @param transform 用于转换集合中的每一项的闭包
     * @return 将所有转换后的值添加到其上的收集器
     * @since 1.0
     */
    public static <T,E> Collection<T> collect(Collection<E> self, Collection<T> collector, @ClosureParams(FirstParam.FirstGenericType.class) Closure<? extends T> transform) {
        for (Object item : self) {
            collector.add(transform.call(item));
            if (transform.getDirective() == Closure.DONE) {
                break;
            }
        }
        return collector;
    }

二、代码示例

代码示例 :

class Test {
    static void main(args) {

        // 为 ArrayList 设置初始值
        def list = ["1", "2", "3"]


        // I. 使用 collate 遍历集合 , 返回一个新集合 , 集合的元素可以在闭包中计算得来

        
        def list3 = list.collect{
            // 字符串乘法就是将元素进行叠加
            it * 2
        }
        // 打印 [1, 2, 3]
        println list
        // 打印 [11, 22, 33]
        println list3
    }
}

执行结果 :