java - 按对象集的值过滤对象列表

Question

我需要检索具有特定角色的用户列表。用户对象包含一组具有 id 和角色名称作为字符串的角色。带有用户的示例 JSON:

[  
 {
    "id": "test-id-user2",
    "email": "user2@email.com",
    "password": "$2a$04$JodczCrLPGYA1cCvtAWZoe47qhiBIMNjSexRFwCLevzRJQQSqZvaO",
    "firstName": "Test2",
    "lastName": "User2",
    "enabled": true,
    "confirmationToken": "0123token0123",
    "location": {
      "id": 1,
      "name": "West Vinewood"
    },
    "roles": [
      {
        "id": 1,
        "role": "ROLE_USER"
      }
    ]
  },
  {
    "id": "test-id-admin",
    "email": "admin@email.com",
    "password": "$2a$04$Ll0WUAuU1p.sjqH.g.f03eHcfS8ox1Pen9tYk8/JdWlgfBr71nNO.",
    "firstName": "AdminTest",
    "lastName": null,
    "enabled": true,
    "confirmationToken": null,
    "location": null,
    "roles": [
      {
        "id": 2,
        "role": "ROLE_ADMIN"
      }
    ]
  }
]

我可以通过在 for 循环中使用 if 循环来实现这一点。一个用户只有一个角色(出于某些原因我必须为此使用 set):

List<User> member = new ArrayList<>();
for (User user : allUsers) {
    Set<UserRole> roles = user.getRoles();
    if (roles.iterator().next().getRole().equals("ROLE_USER")) {
        member.add(user);
    }
}

有没有其他更简单的方法，比如 lambda 来过滤列表中的集合？我试过这个解决方案 Java 8 lambda list inside list filter它会过滤，但也会复制对象。

Answer 1

可以使用流 API 按如下方式完成:

allUsers.stream()           
        .filter(u -> u.getRoles().iterator().next().getRole().equals("ROLE_USER"))
        .collect(Collectors.toCollection(ArrayList::new));