67 lines
2.2 KiB
Java
67 lines
2.2 KiB
Java
package server.logic;
|
|
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
import server.logic.ws_protocol.binary.handlers.*;
|
|
import server.logic.ws_protocol.WireCodes;
|
|
|
|
import java.nio.ByteBuffer;
|
|
import java.nio.ByteOrder;
|
|
import java.util.Map;
|
|
|
|
/**
|
|
* Обработчик входящих сообщение на сервер.
|
|
* По коду сообщения (первые 4 байта сообщения) находи нужный хэндлер и передаёт в него сообщение
|
|
* Получает и возвращает ответ от хэндлера
|
|
*/
|
|
public final class InboundMessageProcessor {
|
|
private static final Logger log = LoggerFactory.getLogger(InboundMessageProcessor.class);
|
|
|
|
private static final Map<Integer, MessageHandler> HANDLERS = Map.of(
|
|
WireCodes.Op.PING, new PingHandler(),
|
|
// WireCodes.Op.ADD_BLOCK, new AddBlockHandler(),
|
|
WireCodes.Op.GET_BLOCKCHAIN,new GetBlockchainHandler()
|
|
// WireCodes.Op.SEARCH_USERS, new SearchUsersHandler(),
|
|
// WireCodes.Op.GET_LAST_BLOCK_INFO,new GetLastBlockInfoHandler()
|
|
|
|
);
|
|
|
|
private InboundMessageProcessor() {}
|
|
|
|
public static byte[] process(byte[] msg) {
|
|
if (msg == null || msg.length < 4)
|
|
return intTo4Bytes(WireCodes.Status.BAD_REQUEST);
|
|
|
|
int op = first4ToInt(msg);
|
|
MessageHandler h = HANDLERS.get(op);
|
|
if (h == null) {
|
|
log.warn("Неизвестная операция: {}", op);
|
|
return intTo4Bytes(WireCodes.Status.BAD_REQUEST);
|
|
}
|
|
|
|
try {
|
|
return h.handle(msg);
|
|
} catch (Exception e) {
|
|
log.error("Ошибка при обработке операции {}", op, e);
|
|
return intTo4Bytes(WireCodes.Status.INTERNAL_ERROR);
|
|
}
|
|
}
|
|
|
|
private static int first4ToInt(byte[] msg) {
|
|
return ByteBuffer.wrap(msg, 0, 4)
|
|
.order(ByteOrder.BIG_ENDIAN)
|
|
.getInt();
|
|
}
|
|
|
|
public static byte[] intTo4Bytes(int code) {
|
|
return ByteBuffer.allocate(4)
|
|
.order(ByteOrder.BIG_ENDIAN)
|
|
.putInt(code)
|
|
.array();
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|