番石榴的排序方式根据另一个列表?

时间:2012-06-07 10:58:35

标签: java collections guava

我有List<Integer>组成我的用户ID。在数据库查询之后,我正在检索List<User>。我想根据第一个Id列表订购此列表。 List<User>可能不包含部分ID。什么是Guava排序这个列表的方法?

5 个答案:

答案 0 :(得分:13)

使用Guava的完全“功能”方式将Ordering#explicit()Ordering#onResultOf()

组合在一起
public class UserService {

    @Inject private UserDao userDao;

    public List<User> getUsersWithIds(List<Integer> userIds) {
        List<User> users = userDao.loadUsersWithIds(userIds);
        Ordering<User> orderById = Ordering.explicit(userIds).onResultOf(UserFunctions.getId());
        return orderById.immutableSortedCopy(users);
    }

}

你可以声明一个内联的匿名函数,但是我喜欢在一个单独的类中将我的函数声明为静态工厂方法,以获得更清晰的代码(Java的函数声明的详细程度隐藏在实用程序类中):

/**
 * Static factory methods to create {@link Function}s for {@link User}s.
 */
public final class UserFunctions {
    private UserFunctions() { /* prevents instantiation */ }

    /**
     * @return a {@link Function} that returns an {@link User}'s id.
     */
    public static Function<User, Integer> getId() {
        return GetIdFunction.INSTANCE;
    }

    // enum singleton pattern
    private enum GetIdFunction implements Function<User, Integer> {
        INSTANCE;

        public Integer apply(User user) {
            return user.getId();
        }
    }

}

正如@Arne在评论中所提到的,这可以在Java 8中简化,使用方法引用而不是UserFunctions类:

public class UserService {

    @Inject private UserDao userDao;

    public List<User> getUsersWithIds(List<Integer> userIds) {
        List<User> users = userDao.loadUsersWithIds(userIds);
        Ordering<User> orderById = Ordering.explicit(userIds).onResultOf(User::getId);
        return orderById.immutableSortedCopy(immutableSortedCopy(users));
    }

}

答案 1 :(得分:9)

我不认为番石榴有任何具体的做法。但这只是写这个比较器的问题:

Collections.sort(userList, new Comparator<User>() {
    @Override
    public int compare(User u1, User u2) {
         int i1 = idList.indexOf(u1.getId());
         int i2 = idList.indexOf(u2.getId());
         return Ints.compare(i1, i2);
    }
}

现在我考虑一下,它也可以这样实现:

final Ordering<Integer> idOrdering = Ordering.explicit(idList);
Collections.sort(userList, new Comparator<User>() {
    @Override
    public int compare(User u1, User u2) {
         return idOrdering.compare(u1.getId(), u2.getId());
    }
}

这可能更有效率。

答案 2 :(得分:2)

其他人已经使用番石榴回答了您的问题。这是一个Functional Java答案。

请注意,您必须使用库中的不可变数据结构才能获得所有优点。

F<User, Integer> indexInIdList = new F<User, Integer>() {
  public Integer f(User u) {
    return idList.elementIndex(Equal.intEqual, u.getId()).toNull();
  }
};
userList.sort(Ord.intOrd.comap(indexInIdList));   

答案 3 :(得分:0)

使用Google Guava更简单的答案

class Form {
  public Integer index; // for simplicity, no setter/getter included
}

List<Form> forms = ... // list instances, each of each with values for index

// ordering of forms by the ui sort index.
private static final Ordering<Form> sorter = Ordering.natural().onResultOf(new Function<Form, Integer>() {

    @Override
    public Integer apply(Form form) {
       return form.index;
    }
});

private List<Form> sortForms(List<Form> forms) {
    return sorter.sortedCopy(forms);
}

答案 4 :(得分:0)

以下是如何使用Java 8 lambdas执行此操作。

List<Integer> ids = ...; List<User> users = ...;
//map ids to their list indices, to avoid repeated indexOf calls
Map<Integer, Integer> rankMap = IntStream.range(0, ids.size()).boxed()
    .collect(Collectors.toMap(ids::get, Function.identity()));
//sort on the id's position in the list
users.sort(Comparator.comparing(u -> rankMap.get(u.id())));