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 @Override
public Map<String, Integer> countUsersForNodes() { public Map<String, Integer> countUsersForNodes() {
var map = new HashMap<String, Integer>(); var map = new HashMap<String, Set<User>>();
lock.readLock().lock(); lock.readLock().lock();
try { 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 { } finally {
lock.readLock().unlock(); 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) { private record NodeRoomPath(String node, String room) {