04 12 25 Версия авторификации где сервер выдовал сессион Ид

src/main/java/server/logic/ws_protocol/binary/handlers/AddBlockHandler.java

@@ -20,7 +20,7 @@ import java.util.Arrays;
  * AddBlockHandler — обработчик команды "добавить блок" (ADD_BLOCK)
  * ---------------------------------------------------------------
  * Принимает бинарное сообщение от клиента и добавляет новый блок в цепочку.
- *
+ *.
  * Формат входного сообщения (msg):
  *   [0..3]   — 4 байта: код операции (WireCodes.ADD_BLOCK)
  *   [4..11]  — 8 байт: blockchainId (уникальный идентификатор цепочки)
@@ -33,13 +33,13 @@ import java.util.Arrays;
  *                 ├── M  байт   body (содержимое блока)
  *                 ├── 64 байта  signature (Ed25519)
  *                 └── 32 байта  hash (SHA-256)
- *
+ *.
  * ---------------------------------------------------------------
  * Алгоритм работы:
- *
+ *.
  * 1️⃣ Распаковать BchBlockEntry из msg (т.е. выделить тело блока и подписи).
  * 2️⃣ Найти описание цепочки (BchInfoEntry) по blockchainId.
- *
+ *.
  *   ─ Если описания нет (цепочка ещё не существует):
  *       • принимаем только блок типа 0 (HeaderBody) и номера 0;
  *       • парсим его, создаём новый BchInfoEntry на основе данных заголовка;
@@ -48,16 +48,16 @@ import java.util.Arrays;
  *       • сохраняем блок и создаём новый blockchain-файл;
  *       • добавляем цепочку в менеджер BchInfoManager.
  *       (💡 временное решение: создание цепочки допустимо только через HeaderBody)
- *
+ *.
  *   ─ Если цепочка уже существует:
  *       • проверяем, что номер блока равен (lastBlockNumber + 1);
  *       • проверяем подпись и хэш;
  *       • проверяем тело блока (check);
  *       • добавляем блок в файл цепочки;
  *       • обновляем состояние BchInfoEntry (номер, хэш, размер).
- *
+ *.
  * 3️⃣ Если все проверки пройдены — возвращаем статус OK.
- *
+ *.
  * Таким образом, единственное различие между первым блоком и последующими —
  * момент инициализации описания цепочки (BchInfoEntry).
  * Всё остальное (валидация, подпись, добавление, обновление) выполняется одинаково.

src/main/java/server/ws/BlockchainWsEndpoint.java

@@ -6,6 +6,7 @@ import org.eclipse.jetty.websocket.api.annotations.*;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import server.logic.InboundMessageProcessor;
+import server.logic.ws_protocol.JSON.ActiveConnectionsRegistry;
 import server.logic.ws_protocol.JSON.ConnectionContext;
 import server.logic.ws_protocol.JSON.JsonInboundProcessor;
 
@@ -24,6 +25,8 @@ public class BlockchainWsEndpoint {
     @OnWebSocketConnect
     public void onConnect(Session session) {
         this.session = session;
+        // Привязываем WebSocket-сессию к ConnectionContext
+        connectionContext.setWsSession(session);
         log.info("WS connected: {}", session.getRemoteAddress());
     }
 
@@ -77,6 +80,8 @@ public class BlockchainWsEndpoint {
     @OnWebSocketClose
     public void onClose(int statusCode, String reason) {
         log.info("WS closed: {} {}", statusCode, reason);
+        // Удаляем это подключение из реестра активных соединений
+        ActiveConnectionsRegistry.getInstance().remove(connectionContext);
         // На всякий случай очищаем контекст
         connectionContext.reset();
     }