Stream groupingBy分组案例

小鱼儿 2022-10-16 07:51 194阅读 0赞

package com.gblfy.gxts;
    
    import com.alibaba.fastjson.JSON;
    import com.google.common.collect.Lists;
    import lombok.AllArgsConstructor;
    import lombok.Data;
    import lombok.NoArgsConstructor;
    import org.junit.Test;
    
    import java.util.ArrayList;
    import java.util.List;
    import java.util.Map;
    import java.util.Optional;
    import java.util.stream.Collectors;
    import java.util.stream.Stream;
    
    /**
     * 案例4：
     * 设计一个对外提供服务的接口，支持调用方传入多个账户编号查询订单。
     */
    public class CaseFour { 
    
        /**
         * 订单模型
         */
        @Data
        @AllArgsConstructor
        @NoArgsConstructor
        class Order { 
    
            /**
             * 订单编号
             */
            private Integer orderId;
    
            /**
             * 账户编号
             */
            private String accuntId;
        }
    
        /**
         * 模拟数据库查询，创建初始化订单数据
         */
        public List<Order> selectFromDB(List<String> accountIds) { 
            List<Order> orderList = new ArrayList<>();
            for (int i = 0; i < 10; i++) { 
                orderList.add(
                        new Order(i,
                                accountIds.get(i % accountIds.size())));
            }
            return orderList;
        }
    
    
        /**
         * @param accountIds
         * @return
         */
        public Map<String, List<Order>> queryOrderByAccountIds(List<String> accountIds) { 
            return Optional.ofNullable(selectFromDB(accountIds))
                    .map(List::stream)
                    .orElseGet(Stream::empty)
                    //TODO 分组
                    .collect(Collectors.groupingBy(order -> order.getAccuntId()));
    
    
        }
    
        /**
         * 按照【账户】显示订单列表
         */
        @Test
        public void findNoScoreStudentList() { 
    
            Map<String, List<Order>> orders = queryOrderByAccountIds(Lists.newArrayList("张三", "李四", "王五"));
            System.out.println(JSON.toJSONString(orders, true));
    
        }
    }