Dragonestia/RoomPicker
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed count users inside nodes

Browse Source
This commit is contained in:
Andrey Terentev 2024-03-15 00:46:10 +07:00 committed by Andrey Terentev
parent 4b3520aa67
commit d362cc31c6
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
server/src/main/java/ru/dragonestia/picker/repository/impl/UserRepositoryImpl.java
View File

@ -166,16 +166,26 @@ public class UserRepositoryImpl implements UserRepository {
@Override
public Map<String, Integer> countUsersForNodes() {
var map = new HashMap<String, Integer>();
var map = new HashMap<String, Set<User>>();
lock.readLock().lock();
try {
roomUsers.forEach((path, users) -> map.put(path.node, users.size()));
roomUsers.forEach((path, users) -> {
if (map.containsKey(path.node)) {
map.get(path.node).addAll(users);
return;
}
map.put(path.node, new HashSet<>(users));
});
} finally {
lock.readLock().unlock();
}
return map;
var result = new HashMap<String, Integer>();
map.forEach((node, users) -> result.put(node, users.size()));
return result;
}
private record NodeRoomPath(String node, String room) {