使用谷歌收藏夹对列表进行过滤和排序

sorting filter java guava

2022-09-01 15:38:57

假设我有一个列表（或集）：

List<String> testList = Lists.newArrayList("assocX","srcT","destA","srcX", "don't care Y", "garbage", "srcB");

我想找回一个不可变列表（Set），它以自然顺序对术语进行排序/分组，其中以“src”开头的术语排在第一位，“assoc”排在第二位，“dest”最后。如果某个术语不包含这些内容，则应将其从结果列表中删除。

因此，这里的结果是“srcB”，“srcT”，“assocX”，“destA”。

我认为我可以通过Iterables.filter或谓词的某种组合来做到这一点，但只是没有看到它。我认为必须有一种简洁的方法。

编辑：代替列表的集合也可以。

答案 1

只要这三个前缀是你唯一关心的事情，我就建议这样：

    Predicate<String> filter = new Predicate<String>() {
        @Override
        public boolean apply(String input) {
            return input.startsWith("src") || input.startsWith("assoc") || input.startsWith("dest");
        }
    };

    Function<String, Integer> assignWeights = new Function<String, Integer>() {
        @Override
        public Integer apply(String from) {
            if (from.startsWith("src")) {
                return 0;
            } else if (from.startsWith("assoc")) {
                return 1;
            } else if (from.startsWith("dest")) {
                return 2;
            } else {
                /* Shouldn't be possible but have to do something */
                throw new IllegalArgrumentException(from + " is not a valid argument");
            }
        }
    };

    ImmutableList<String> sortedFiltered = ImmutableList.copyOf(
            Ordering.natural().onResultOf(assignWeights).sortedCopy(
                    Iterables.filter(testList, filter)
            )
    );

如果您开始添加更多前缀来过滤或排序，此解决方案肯定不会很好地扩展，因为您必须不断更新过滤器和每个前缀的权重。

答案 2

看看这个谷歌收藏夹的例子。

Function<Fruit, String> getNameFunction = new Function<Fruit, String>() {
    public String apply(Fruit from) {
        return from.getName();
    }
};

Ordering<Fruit> nameOrdering = Ordering.natural().onResultOf(getNameFunction);

ImmutableSortedSet<Fruit> sortedFruits = ImmutableSortedSet.orderedBy(
    nameOrdering).addAll(fruits).build();

虽然，不可否认，这返回了一个集合。