java - 循环迭代器直到它为空

Question

我想从列表中获取所有对象并将它们放入 Map 中按创建日期分组，这意味着 map 是这样的:Map<String, List<MyObject>> . MyObject对象有一个存储其创建日期的字段。

我想过做一个嵌套的 while 循环，看起来像这样:

public Map<String, List<Expense>> getExpensesSorted(SortType type){
    Map<String, List<Expense>> map = new HashMap<String, List<Expense>>();

    List<Expense> expenses = getAllExpenses(budgetId).getExpenses()
            .getList();

    if (type.equals(SortType.DAY)) {
        Iterator<Expense> expIter = expenses.iterator();

        while (expIter.hasNext()) {
            List<Expense> list = new ArrayList<Expense>();
            Expense exp = (Expense) expIter.next();
            list.add(exp);

            String day = exp.getDate().format("YYYY-MM-DD");
            expIter.remove();

            while (expIter.hasNext()) {
                Expense exp2 = (Expense) expIter.next();
                if (exp2.getDate().format("YYYY-MM-DD").equals(day)) {
                    list.add(exp2);
                    expIter.remove();
                }
            }
            map.put(day, expenses);
        }

    } else if (type.equals(SortType.WEEK)) {
        ...
    } else if (type.equals(SortType.TYPE)) {
        ...
    } else if (type.equals(SortType.CATEGORY)) {
        ...
    }
    return map;
}

但这是错误的，它只获取所有与第一个元素具有同一天的元素，所以我的 map 最终只有一个元素。

我真的不知道如何解决这个...

在此先感谢您的帮助。

Answer 1

Map<Data, List<MyObject>> result = new HashMap<Data, List<MyObject>>;
for (List<MyObject> list : myMap.values()) {
    for (MyObject myObject : list) {
        Date date = myObject.getDate();
        List<MyObject> newList = result.get(date);
        if (newList == null) {
            newList = new Arraylist<MyObject>;
            result.put(date. newList);
        }
        newList.add(myObject);
    }
}