@Override
public ByteBuf serializeResponseAsBinary(
final ResponseMessage responseMessage, final ByteBufAllocator allocator)
throws SerializationException {
ByteBuf encodedMessage = null;
try {
final Kryo kryo = kryoThreadLocal.get();
try (final OutputStream baos = new ByteArrayOutputStream()) {
final Output output = new Output(baos);
// request id - if present
kryo.writeObjectOrNull(
output,
responseMessage.getRequestId() != null ? responseMessage.getRequestId() : null,
UUID.class);
// status
output.writeShort(responseMessage.getStatus().getCode().getValue());
output.writeString(responseMessage.getStatus().getMessage());
kryo.writeClassAndObject(output, responseMessage.getStatus().getAttributes());
// result
kryo.writeClassAndObject(
output,
serializeToString
? serializeResultToString(responseMessage)
: responseMessage.getResult().getData());
kryo.writeClassAndObject(output, responseMessage.getResult().getMeta());
final long size = output.total();
if (size > Integer.MAX_VALUE)
throw new SerializationException(
String.format("Message size of %s exceeds allocatable space", size));
encodedMessage = allocator.buffer((int) output.total());
encodedMessage.writeBytes(output.toBytes());
}
return encodedMessage;
} catch (Exception ex) {
if (encodedMessage != null) ReferenceCountUtil.release(encodedMessage);
logger.warn(
"Response [{}] could not be serialized by {}.",
responseMessage.toString(),
GryoMessageSerializerV1d0.class.getName());
throw new SerializationException(ex);
}
}
