Java函数如何返回两个ArrayList？在线/离线用户列表实现疑问

嘿，这个问题我之前也碰到过！Java里确实没法直接返回两个独立的集合，但结合你的Spring Controller场景，有几种很实用的解决办法，给你唠唠：

方案1：自定义DTO类（最规范，推荐生产环境用） #

最稳妥的方式是创建一个专门的数据传输对象（DTO），把在线、离线用户列表作为它的属性。这样API的返回结构清晰，后续要加字段也方便扩展。

先写一个DTO类：

// 可以放在你的model或dto包下
public class UserStatusResponse {
    private List<User> onlineUsers;
    private List<User> offlineUsers;

    // 构造方法、getter和setter（如果用Lombok的@Data注解，能省掉这些代码）
    public UserStatusResponse(List<User> onlineUsers, List<User> offlineUsers) {
        this.onlineUsers = onlineUsers;
        this.offlineUsers = offlineUsers;
    }

    // getter和setter方法...
}

然后修改你的Controller方法，返回这个DTO：

@RequestMapping(value = "/loggedUsers", method = RequestMethod.GET)
public @ResponseBody UserStatusResponse getLoggedUsers() {
    // 先把所有用户转成Map，方便快速查找
    Map<String, User> allUsersMap = User.getUsers().stream()
        .collect(Collectors.toMap(User::getUsername, Function.identity()));
    
    // 填充在线用户列表
    List<User> onlineUsers = new ArrayList<>();
    for(String username : activeUsers.getUsers()) {
        onlineUsers.add(allUsersMap.get(username));
    }
    
    // 计算离线用户：从所有用户中过滤掉在线的
    Set<String> onlineUsernames = new HashSet<>(activeUsers.getUsers()); // 用Set提升contains效率
    List<User> offlineUsers = User.getUsers().stream()
        .filter(user -> !onlineUsernames.contains(user.getUsername()))
        .collect(Collectors.toList());
    
    // 返回封装好的DTO
    return new UserStatusResponse(onlineUsers, offlineUsers);
}

前端拿到的JSON会是这样的结构，非常直观：

{
  "onlineUsers": [{"username": "张三", ...}, ...],
  "offlineUsers": [{"username": "李四", ...}, ...]
}

方案2：直接返回Map（快速实现，适合临时需求） #

如果不想额外写DTO类，也可以直接返回一个Map<String, List<User>>，用键名区分两个列表：

@RequestMapping(value = "/loggedUsers", method = RequestMethod.GET)
public @ResponseBody Map<String, List<User>> getLoggedUsers() {
    // 前面的在线、离线用户计算逻辑和方案1一致
    Map<String, List<User>> resultMap = new HashMap<>();
    resultMap.put("onlineUsers", onlineUsers);
    resultMap.put("offlineUsers", offlineUsers);
    
    return resultMap;
}

这种方式不用额外创建类，适合快速验证功能，但后续如果要扩展返回字段，不如DTO灵活。

方案3：用ResponseEntity包装（需要自定义响应头/状态码时用） #

如果需要设置HTTP状态码、响应头这类信息，可以用Spring的ResponseEntity来包装上面的DTO或Map：

@RequestMapping(value = "/loggedUsers", method = RequestMethod.GET)
public ResponseEntity<UserStatusResponse> getLoggedUsers() {
    // 生成onlineUsers和offlineUsers的逻辑同方案1
    UserStatusResponse responseBody = new UserStatusResponse(onlineUsers, offlineUsers);
    // 返回200 OK响应，同时携带响应体
    return ResponseEntity.ok(responseBody);
    // 如果要自定义状态码，比如：return ResponseEntity.status(HttpStatus.OK).body(responseBody);
}

小优化提示 #

计算离线用户时，把在线用户名转成Set（比如new HashSet<>(activeUsers.getUsers())），比直接用List的contains方法效率更高，尤其是用户量多的时候。

内容的提问来源于stack exchange，提问作者user9658240