Исправление ошибки ConcurrentModificationException при множестве потоков

.gitea/workflows/build.yaml

@@ -1,4 +1,5 @@
 name: Build rosetta-wss
+run-name: Build and Deploy WSS
 on:
   push:
     branches:

src/main/java/im/rosetta/client/ClientManager.java

@@ -1,7 +1,7 @@
 package im.rosetta.client;
 
-import java.util.HashSet;
 import java.util.List;
+import java.util.Set;
 
 import im.rosetta.client.tags.ECIAuthentificate;
 
@@ -33,7 +33,7 @@ public class ClientManager {
     }
 
     public boolean isClientConnected(String publicKey) {
-        HashSet<Client> clients = this.clientIndexer.getClients(ECIAuthentificate.class, "publicKey", publicKey);
+        Set<Client> clients = this.clientIndexer.getClients(ECIAuthentificate.class, "publicKey", publicKey);
         if(clients == null){
             /**
              * Нет клиентов с таким публичным ключом
@@ -59,7 +59,7 @@ public class ClientManager {
      * @throws ProtocolException если произошла ошибка при отправке пакета клиенту
      */
     public void sendPacketToAuthorizedPK(String publicKey, Packet packet) throws ProtocolException {
-        HashSet<Client> clients = this.clientIndexer.getClients(ECIAuthentificate.class, "publicKey", publicKey);
+        Set<Client> clients = this.clientIndexer.getClients(ECIAuthentificate.class, "publicKey", publicKey);
         if(clients == null){
             /**
              * Нет клиентов с таким публичным ключом, значит отправлять некому
@@ -91,7 +91,7 @@ public class ClientManager {
      */
     public void retranslate(Client client, Packet packet) throws ProtocolException{
         ECIAuthentificate eciAuthentificate = client.getTag(ECIAuthentificate.class);
-        HashSet<Client> clients = this.clientIndexer
+        Set<Client> clients = this.clientIndexer
             .getClients(ECIAuthentificate.class, "publicKey", eciAuthentificate.getPublicKey());
         if(clients == null){
             /**
@@ -129,7 +129,7 @@ public class ClientManager {
      * @return список клиентов с таким публичным ключом, может быть пустым, если клиентов с таким публичным ключом нет
      */
     public List<Client> getPKClients(String publicKey) {
-        HashSet<Client> clients = this.clientIndexer.getClients(ECIAuthentificate.class, "publicKey", publicKey);
+        Set<Client> clients = this.clientIndexer.getClients(ECIAuthentificate.class, "publicKey", publicKey);
         if(clients == null){
             /**
              * Нет клиентов с таким публичным ключом

src/main/java/io/orprotocol/index/ClientIndexer.java

@@ -1,7 +1,7 @@
 package io.orprotocol.index;
 
-import java.util.HashSet;
 import java.util.Map;
+import java.util.Set;
 import java.util.concurrent.ConcurrentHashMap;
 
 import io.orprotocol.client.Client;
@@ -19,14 +19,8 @@ public class ClientIndexer {
      * Ключ третьего уровня - значение поля.
      * Значение - клиент.
      * Структура: tagClass -> indexName -> indexValue -> Client
-     * 
-     * 
-     * В качестве клиента используем HashSet потому что он очень хорош для операции contains,
-     * в отличие от ArrayList, который при поиске перебирает элементы один за другим, 
-     * HashSet использует хэш таблицу и находит нужный объект почти 
-     * мгновенно, независимо от того, 10 там элементов или миллион.
      */
-    private final Map<Class<?>, Map<String, Map<Object, HashSet<Client>>>> indices 
+    private final ConcurrentHashMap<Class<?>, ConcurrentHashMap<String, ConcurrentHashMap<Object, Set<Client>>>> indices 
         = new ConcurrentHashMap<>();
 
     public <T extends ECITag> void indexTag(Client client, Class<T> tagClass, T tag) {
@@ -53,7 +47,7 @@ public class ClientIndexer {
          * Если использовать putIfAbsent, то он вернет null если значение там уже есть,
          * что не подходит
          */
-        Map<String, Map<Object, HashSet<Client>>> tagIndices = this.indices.computeIfAbsent(
+        ConcurrentHashMap<String, ConcurrentHashMap<Object, Set<Client>>> tagIndices = this.indices.computeIfAbsent(
             tagClass, 
             k -> new ConcurrentHashMap<>());
         
@@ -76,7 +70,7 @@ public class ClientIndexer {
             /**
              * Инициализируем имя индекса, почему compute, а не put - написано выше
              */
-            Map<Object, HashSet<Client>> index = tagIndices.computeIfAbsent(
+            Map<Object, Set<Client>> index = tagIndices.computeIfAbsent(
                 indexName,
                 k -> new ConcurrentHashMap<>()
             );
@@ -86,7 +80,7 @@ public class ClientIndexer {
              * с одинковыми индексами, хотя такого лучше избегать, желательно чтобы индексы
              * были уникальными, тогда обработка будет быстрее всего
              */
-            HashSet<Client> clients = index.computeIfAbsent(indexValue, k -> new HashSet<>());
+            Set<Client> clients = index.computeIfAbsent(indexValue, k -> ConcurrentHashMap.newKeySet());
             /**
              * Добавляем клиента в инициализиованный индекс
              */
@@ -117,7 +111,7 @@ public class ClientIndexer {
      * @internal
      */
     public <T extends ECITag> void removeTagIndex(Client client, Class<T> tagClass) {
-        Map<String, Map<Object, HashSet<Client>>> tagIndices = indices.get(tagClass);
+        ConcurrentHashMap<String, ConcurrentHashMap<Object, Set<Client>>> tagIndices = indices.get(tagClass);
         if (tagIndices == null) {
             /**
              * Индекса и так не было, удалять нечего
@@ -128,12 +122,12 @@ public class ClientIndexer {
         /**
          * Удаляем все ключи indexName по tagClass если Client == client
          */
-        for (Map<Object, HashSet<Client>> index : tagIndices.values()) {
+        for (ConcurrentHashMap<Object, Set<Client>> index : tagIndices.values()) {
             /**
              * contains всегда использует переопределенный equals, по этому
              * обьекты клиентов сравниваются нормально
              */
-            for(HashSet<Client> clients : index.values()){
+            for(Set<Client> clients : index.values()){
                 if(!clients.contains(client)){
                     continue;
                 }
@@ -147,9 +141,9 @@ public class ClientIndexer {
      * @internal
      */
     public void removeClientFromIndex(Client client) {
-        for(Map<String, Map<Object, HashSet<Client>>> tagIndices : this.indices.values()){
+        for(Map<String, ConcurrentHashMap<Object, Set<Client>>> tagIndices : this.indices.values()){
-            for(Map<Object, HashSet<Client>> index : tagIndices.values()){
+            for(Map<Object, Set<Client>> index : tagIndices.values()){
-                for(HashSet<Client> clients : index.values()){
+                for(Set<Client> clients : index.values()){
                     /**
                      * Этот тройной цикл не такой страшный, так как мы всего лишь
                      * проходим по всем тегам (их немного), дальше идем по всем значениям в тегах
@@ -173,7 +167,7 @@ public class ClientIndexer {
      * @param indexValue значение в теге
      * @return список клиентов с заданными значениями
      */
-    public <T extends ECITag> HashSet<Client> getClients(Class<T> tagClass, String indexName, Object indexValue) {
+    public <T extends ECITag> Set<Client> getClients(Class<T> tagClass, String indexName, Object indexValue) {
         if(indexName == null || indexValue == null){
             return null;
         }
@@ -181,12 +175,12 @@ public class ClientIndexer {
         /**
          * Получение по индексу простое, так как каждое из заданных значений и есть ключ
          */
-        Map<String, Map<Object, HashSet<Client>>> tagIndices = indices.get(tagClass);
+        ConcurrentHashMap<String, ConcurrentHashMap<Object, Set<Client>>> tagIndices = indices.get(tagClass);
         if (tagIndices == null) {
             return null;
         }
 
-        Map<Object, HashSet<Client>> index = tagIndices.get(indexName);
+        ConcurrentHashMap<Object, Set<Client>> index = tagIndices.get(indexName);
         if (index == null) {
             return null;
         }