如何匹配地图返回类型的键和值？

Question

我正在研究一种方法，其中getOrdersForCustomer方法的返回类型为Map<Integer, List<Order>>并且每个customerId（地图的Integer）在遍历订单列表时都需要使用其各自的订单。 订单已与客户ID相关联。 我得出以下结论：

public Map<Integer, List<Order>> getOrdersForCustomer(Company company) {

    List<Order> orders = orderDao.findAll(company);
    Map<Integer, List<Order>> map = new HashMap<>();

    for(Order order: orders){

        int customerId = order.getCustomerId();
        if( map.get(customerId).size() ==0){

            map.put( customerId, new ArrayList<Order>());
        }

        else{

            map.get(customerId).add(order);
        }
    }

    return map; 
}

如何更正代码才能正常工作？

Answer 1

似乎第一个Order对象不会被添加到Map中的List中。 我相信您应该始终在for的每次迭代中进行添加：

for（订单订单：订单）{int customerId = order.getCustomerId（）;

    if( map.get(customerId).size() ==0){

        map.put( customerId, new ArrayList<Order>());
    }

    map.get(customerId).add(order);
}

Answer 2

我假设您要在这里获得NPE？

if( map.get(customerId).size() ==0){

做：

if (!map.contains(customerId)) {

Answer 3

您需要修复此检查。 如果customerId不在地图中，则将获得空指针异常。

 if( map.get(customerId).size() ==0) {
   map.put( customerId, new ArrayList<Order>());
  }

逻辑也不正确。 您尚未将订单添加到要添加的列表中。 您可以将订单添加到列表构造函数中，也可以删除else语句。

Answer 4

是的，逻辑上有一个错误，我按如下纠正了@djechlin所建议的错误，

@Override
public Map<Integer, List<Order>> getOrdersForCustomer(Company company) {

    List<Order> orders = orderDao.findAll(company);
    Map<Integer, List<Order>> map = new HashMap<>();

    for(Order order: orders){

        int customerId = order.getCustomerId();
        if(!map.containsKey(customerId)){

            map.put(customerId, new ArrayList<Order>());
        }

        map.get(customerId).add(order);
    }

    return map; 
}