Kotlin sortBy在TreeMap上

时间:2016-01-27 14:06:59

标签: group-by treemap kotlin

实现这一目标的kotlin方式是什么:

public static TreeMap<Integer, ArrayList<HomeItem>> getHomeItemListsSeparatedByPageId(List<HomeItem> homeItemsList) {

        if (homeItemsList.size() == 0) {
            return new TreeMap<>();
        }

        TreeMap<Integer, ArrayList<HomeItem>> map = new TreeMap<>();

        for (HomeItem hl : homeItemsList) {
            if (map.get(hl.getVisiblePageId()) == null) {
                map.put(hl.getVisiblePageId(), new ArrayList<>());
            }
            map.get(hl.getVisiblePageId()).add(hl);
        }

        return map;
    }

我试图在我的args列表中应用groupBy,但它返回一个HashMap。

编辑:基本上,查询原始列表,并按visiblePageId进行分组,并将其添加到树形图中,其键为id,值为具有该ID的项

1 个答案:

答案 0 :(得分:4)

您可以使用groupByTo method并传递TreeMap实例作为第一个参数。

fun getHomeItemListsSeparatedByPageId(homeItems: List<HomeItem>) =
    homeItems.groupByTo(TreeMap()) { it.visiblePageId }