使用List的stream将List集合转成Map
具体的情况是查询出来店铺的订单列表信息,然后每条信息需要加上用户的信息----->订单表中有用户id,最开始偷懒想快点写完(怕数据量大导致查的慢就也没写关联sql),查出来当前店铺的订单列表后,直接使用了for循环查询数据库,后来自己测一遍发现这种方式太离谱,就改造了一下。
具体实现方式如下:
// 定义返回的对象 List<OrderDTO> resultOrders = new ArrayList<>(); String storeId = "店铺id"; // 查询出店铺订单列表 List<StoreOrderListDTO> storeOrderList = orderMapper.findStoreOrderList(storeId); // 将userId抽离出来组成一个userId集合 List<String> userIds = orderList .stream() .map(StoreOrderListDTO::getUserId) .collect(Collectors.toList()); // 此时查询来这些用户信息 List<UserInfo> userList = userMapper.listByIds(userIds); // 将用户信息集合转成Map<Long, UserInfo> Map<String, UserInfo> userMap = userList .stream() .collect(Collectors .toMap(CyConsumerUser::getUserId, cyConsumerUser -> cyConsumerUser)); Map<String, List<UserInfo>> finalUserInfoMap = userMap; // 然后将用户信息设置到返回的DTO中 storeOrderList.forEach(storeOrderItem->{ OrderDTO orderDTO = new OrderDTO; // 拷贝对象属性,第一个参数数源,第二个参数是目标 BeanUtils.copyProperties(storeOrderItem, orderDTO ); // 设置用户信息 UserInfo userInfo = finalUserInfoMap.get(storeOrderItem.getUserId()); BeanUtils.copyProperties(storeOrderItem, userInfo); resultOrders.add(orderDTO); }) // 最后返回结果 return resultOrders;
记录一下碰到的一个问题,方便以后用到stream的这些用法然后直接复制拿来用,大神看到了别喷,刚工作比较菜,只能想到这种方式来进行改造。
(判空逻辑和其他的逻辑就没写了,只是自己针对此问题的一个实现方式)