use of com.tvd12.ezyfox.codec.EzyMessageHeader in project ezyfox-server by youngmonkeys.
the class EzySimpleNioHandlerGroup method handleReceivedMessage.
private void handleReceivedMessage(EzyMessage message) {
try {
EzyMessageHeader header = message.getHeader();
if (header.isRawBytes()) {
boolean sessionStreamingEnable = session.isStreamingEnable();
if (!streamingEnable || !sessionStreamingEnable) {
return;
}
byte[] rawBytes = message.getContent();
EzySocketStream stream = new EzySimpleSocketStream(session, rawBytes);
streamQueue.add(stream);
} else {
Object data = decodeMessage(message);
int dataSize = message.getByteCount();
handleReceivedData(data, dataSize);
}
} catch (Exception e) {
fireExceptionCaught(e);
} finally {
executeAddReadBytes(message.getByteCount());
}
}
use of com.tvd12.ezyfox.codec.EzyMessageHeader in project ezyfox-server by youngmonkeys.
the class EzySimpleNioUdpDataHandler method handleReceivedUdpPacket.
protected void handleReceivedUdpPacket(EzyUdpReceivedPacket packet) {
try {
EzyMessage message = EzyMessageReaders.bytesToMessage(packet.getBytes());
if (message == null) {
return;
}
InetSocketAddress udpAddress = packet.getAddress();
Object socketChannel = handlerGroupManager.getSocketChannel(udpAddress);
if (socketChannel != null) {
SocketAddress tcpAddress = (socketChannel instanceof SocketChannel) ? ((SocketChannel) socketChannel).getRemoteAddress() : ((Session) socketChannel).getRemoteAddress();
if (isOneClient(tcpAddress, udpAddress)) {
socketDataReceiver.udpReceive(socketChannel, message);
}
} else {
EzyMessageHeader header = message.getHeader();
if (header.isUdpHandshake()) {
handleUdpHandshake(packet.getChannel(), udpAddress, message);
}
}
} catch (Exception e) {
logger.warn("handle received udp package: {} error: {}({})", packet, e.getClass().getName(), e.getMessage());
}
}
use of com.tvd12.ezyfox.codec.EzyMessageHeader in project ezyfox-server by youngmonkeys.
the class EzySimpleNioHandlerGroupTest method handleReceivedMessageHeaderIsNotRawBytes.
@Test
public void handleReceivedMessageHeaderIsNotRawBytes() throws Exception {
// given
EzySimpleNioHandlerGroup sut = newHandlerGroup();
EzyMessage message = mock(EzyMessage.class);
EzyMessageHeader header = mock(EzyMessageHeader.class);
when(message.getHeader()).thenReturn(header);
// when
MethodInvoker.create().object(sut).method("handleReceivedMessage").param(EzyMessage.class, message).call();
// then
verify(header, times(1)).isRawBytes();
EzySimpleMessageDataDecoder decoder = FieldUtil.getFieldValue(sut, "decoder");
EzyByteToObjectDecoder d = FieldUtil.getFieldValue(decoder, "decoder");
EzySession session = FieldUtil.getFieldValue(sut, "session");
verify(d, times(1)).decode(message, session.getSessionKey());
}
use of com.tvd12.ezyfox.codec.EzyMessageHeader in project ezyfox-server by youngmonkeys.
the class EzySimpleNioHandlerGroupTest method handleReceivedMessageStreamingNotEnable.
@Test
public void handleReceivedMessageStreamingNotEnable() throws Exception {
// given
EzySimpleNioHandlerGroup sut = newHandlerGroup(false);
EzyMessage message = mock(EzyMessage.class);
EzyMessageHeader header = mock(EzyMessageHeader.class);
when(header.isRawBytes()).thenReturn(true);
when(message.getHeader()).thenReturn(header);
// when
MethodInvoker.create().object(sut).method("handleReceivedMessage").param(EzyMessage.class, message).call();
// then
verify(header, times(1)).isRawBytes();
}
use of com.tvd12.ezyfox.codec.EzyMessageHeader in project ezyfox-server by youngmonkeys.
the class EzySimpleNioHandlerGroupTest method handleReceivedMessageSessionStreamingNotEnable.
@Test
public void handleReceivedMessageSessionStreamingNotEnable() throws Exception {
// given
EzySimpleNioHandlerGroup sut = newHandlerGroup();
EzyMessage message = mock(EzyMessage.class);
EzyMessageHeader header = mock(EzyMessageHeader.class);
when(header.isRawBytes()).thenReturn(true);
when(message.getHeader()).thenReturn(header);
EzySession session = FieldUtil.getFieldValue(sut, "session");
when(session.isStreamingEnable()).thenReturn(false);
// when
MethodInvoker.create().object(sut).method("handleReceivedMessage").param(EzyMessage.class, message).call();
// then
verify(header, times(1)).isRawBytes();
verify(session, times(1)).isStreamingEnable();
}
Aggregations