java

记一次使用Lamada表达式的代码优化

2019-04-12  本文已影响0人  singlezero

最近一直在敲代码,没有挑战性的日子都要吐了,昨天给前端写好的接口,今天让我修改一下数据格式,随口就答应了,可写起来就很尴尬.
这个接口需求是按照map中的ItemIndex的value进行分类,组装成一个map返回给前端.当时脑子有点蒙,就写了下面的代码

        List<Map<String, Object>> itemindex1 =  new ArrayList<>();
        List<Map<String, Object>> itemindex2 =  new ArrayList<>();
        List<Map<String, Object>> itemindex3 =  new ArrayList<>();
        List<Map<String, Object>> itemindex4 =  new ArrayList<>();
        List<Map<String, Object>> itemindex5 =  new ArrayList<>();

        for (Map<String, Object> map : mapList) {
            String itemIndex = map.get("ItemIndex").toString();
            switch (itemIndex){
                case "1":
                    itemindex1.add(map);
                    break;
                case "2":
                    itemindex2.add(map);
                    break;
                case "3":
                    itemindex3.add(map);
                    break;
                case "4":
                    itemindex4.add(map);
                    break;
                case "5":
                    itemindex5.add(map);
                    break;
                default:
                    break;
            }
        }

        objectMap.put("1",itemindex1);
        objectMap.put("2",itemindex2);
        objectMap.put("3",itemindex3);
        objectMap.put("4",itemindex4);
        objectMap.put("5",itemindex5);

这样的代码有很多问题,重复定义实体类,没有可扩展性,一旦ItemIndex新增,这个代码就又要改,我知道这样写是不对的,但是前端催的急,我就先临时写好,决定下午再优化他,

下午有时间后,重新看了一遍需求觉得应该可以用Lamada写(可能是中午一觉睡醒了...),因为我本人也不是很熟,所以Google一下就差不多有了答案

        Map<Object, List<Map<String, Object>>> mapWall = mapList.stream().collect(Collectors.toMap(key -> key.get("ItemIndex"),
                Lists::newArrayList,
                (List<Map<String, Object>> newValueList, List<Map<String, Object>> oldValueList) -> {
                    oldValueList.addAll(newValueList);
                    return oldValueList; }));

这样写很多好处就不说了...自己也感觉高大上了不少,看来接下来要好好学一下Lamada表达式了

2019-05-05 更新,Lamada表达式高阶函数用法

之前的方法确实很实用,但接下来学习中才真的见识到,高阶函数的厉害,总之上面的Lamada表达式方法,可以总结为一个高阶函数方法:

Map<Object, List<Map<String, Object>>> type = lists.stream().collect(groupingBy(e -> e.get("type")));

collect收集器,中的groupingBy进行数据分组.

上一篇 下一篇

猜你喜欢

热点阅读