How to use

getRequestId

method

in

org.apache.flink.queryablestate.network.messages.MessageSerializer

@Override
public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
  try {
    ByteBuf buf = (ByteBuf) msg;
    MessageType msgType = MessageSerializer.deserializeHeader(buf);
    if (msgType == MessageType.REQUEST_RESULT) {
      long requestId = MessageSerializer.getRequestId(buf);
      RESP result = serializer.deserializeResponse(buf);
      callback.onRequestResult(requestId, result);
    } else if (msgType == MessageType.REQUEST_FAILURE) {
      RequestFailure failure = MessageSerializer.deserializeRequestFailure(buf);
      callback.onRequestFailure(failure.getRequestId(), failure.getCause());
    } else if (msgType == MessageType.SERVER_FAILURE) {
      throw MessageSerializer.deserializeServerFailure(buf);
    } else {
      throw new IllegalStateException("Unexpected response type '" + msgType + "'");
    }
  } catch (Throwable t1) {
    try {
      callback.onFailure(t1);
    } catch (Throwable t2) {
      LOG.error("Failed to notify callback about failure", t2);
    }
  } finally {
    ReferenceCountUtil.release(msg);
  }
}

final MessageType msgType = MessageSerializer.deserializeHeader(buf);
requestId = MessageSerializer.getRequestId(buf);

  @Override
  public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
    ByteBuf buf = (ByteBuf) msg;
    assertEquals(MessageType.REQUEST, MessageSerializer.deserializeHeader(buf));
    long requestId = MessageSerializer.getRequestId(buf);
    KvStateInternalRequest request = serializer.deserializeRequest(buf);
    buf.release();
    KvStateResponse response = new KvStateResponse(serializedResult);
    ByteBuf serResponse = MessageSerializer.serializeResponse(
        ctx.alloc(),
        requestId,
        response);
    ctx.channel().writeAndFlush(serResponse);
  }
});

long requestId = MessageSerializer.getRequestId(buf);
KvStateInternalRequest deserRequest = serializer.deserializeRequest(buf);

assertEquals(requestId, MessageSerializer.getRequestId(buf));
KvStateResponse response = server.getSerializer().deserializeResponse(buf);

/**
 * Tests response serialization with zero-length serialized result.
 */
@Test
public void testResponseSerializationWithZeroLengthSerializedResult() throws Exception {
  byte[] serializedResult = new byte[0];
  final KvStateResponse response = new KvStateResponse(serializedResult);
  final MessageSerializer<KvStateInternalRequest, KvStateResponse> serializer =
      new MessageSerializer<>(new KvStateInternalRequest.KvStateInternalRequestDeserializer(), new KvStateResponse.KvStateResponseDeserializer());
  ByteBuf buf = MessageSerializer.serializeResponse(alloc, 72727278L, response);
  int frameLength = buf.readInt();
  assertEquals(MessageType.REQUEST_RESULT, MessageSerializer.deserializeHeader(buf));
  assertEquals(72727278L, MessageSerializer.getRequestId(buf));
  KvStateResponse responseDeser = serializer.deserializeResponse(buf);
  assertEquals(buf.readerIndex(), frameLength + 4);
  assertArrayEquals(serializedResult, responseDeser.getContent());
}

/**
 * Tests response serialization.
 */
@Test
public void testResponseSerialization() throws Exception {
  long requestId = Integer.MAX_VALUE + 72727278L;
  byte[] serializedResult = randomByteArray(1024);
  final KvStateResponse response = new KvStateResponse(serializedResult);
  final MessageSerializer<KvStateInternalRequest, KvStateResponse> serializer =
      new MessageSerializer<>(new KvStateInternalRequest.KvStateInternalRequestDeserializer(), new KvStateResponse.KvStateResponseDeserializer());
  ByteBuf buf = MessageSerializer.serializeResponse(alloc, requestId, response);
  int frameLength = buf.readInt();
  assertEquals(MessageType.REQUEST_RESULT, MessageSerializer.deserializeHeader(buf));
  assertEquals(requestId, MessageSerializer.getRequestId(buf));
  KvStateResponse responseDeser = serializer.deserializeResponse(buf);
  assertEquals(buf.readerIndex(), frameLength + 4);
  assertArrayEquals(serializedResult, responseDeser.getContent());
}

long deserRequestId = MessageSerializer.getRequestId(buf);
KvStateResponse response = serializer.deserializeResponse(buf);

/**
 * Tests request serialization with zero-length serialized key and namespace.
 */
@Test
public void testRequestSerializationWithZeroLengthKeyAndNamespace() throws Exception {
  long requestId = Integer.MAX_VALUE + 1337L;
  KvStateID kvStateId = new KvStateID();
  byte[] serializedKeyAndNamespace = new byte[0];
  final KvStateInternalRequest request = new KvStateInternalRequest(kvStateId, serializedKeyAndNamespace);
  final MessageSerializer<KvStateInternalRequest, KvStateResponse> serializer =
      new MessageSerializer<>(new KvStateInternalRequest.KvStateInternalRequestDeserializer(), new KvStateResponse.KvStateResponseDeserializer());
  ByteBuf buf = MessageSerializer.serializeRequest(alloc, requestId, request);
  int frameLength = buf.readInt();
  assertEquals(MessageType.REQUEST, MessageSerializer.deserializeHeader(buf));
  assertEquals(requestId, MessageSerializer.getRequestId(buf));
  KvStateInternalRequest requestDeser = serializer.deserializeRequest(buf);
  assertEquals(buf.readerIndex(), frameLength + 4);
  assertEquals(kvStateId, requestDeser.getKvStateId());
  assertArrayEquals(serializedKeyAndNamespace, requestDeser.getSerializedKeyAndNamespace());
}

/**
 * Tests request serialization.
 */
@Test
public void testRequestSerialization() throws Exception {
  long requestId = Integer.MAX_VALUE + 1337L;
  KvStateID kvStateId = new KvStateID();
  byte[] serializedKeyAndNamespace = randomByteArray(1024);
  final KvStateInternalRequest request = new KvStateInternalRequest(kvStateId, serializedKeyAndNamespace);
  final MessageSerializer<KvStateInternalRequest, KvStateResponse> serializer =
      new MessageSerializer<>(new KvStateInternalRequest.KvStateInternalRequestDeserializer(), new KvStateResponse.KvStateResponseDeserializer());
  ByteBuf buf = MessageSerializer.serializeRequest(alloc, requestId, request);
  int frameLength = buf.readInt();
  assertEquals(MessageType.REQUEST, MessageSerializer.deserializeHeader(buf));
  assertEquals(requestId, MessageSerializer.getRequestId(buf));
  KvStateInternalRequest requestDeser = serializer.deserializeRequest(buf);
  assertEquals(buf.readerIndex(), frameLength + 4);
  assertEquals(kvStateId, requestDeser.getKvStateId());
  assertArrayEquals(serializedKeyAndNamespace, requestDeser.getSerializedKeyAndNamespace());
}

@Override
public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
  try {
    ByteBuf buf = (ByteBuf) msg;
    MessageType msgType = MessageSerializer.deserializeHeader(buf);
    if (msgType == MessageType.REQUEST_RESULT) {
      long requestId = MessageSerializer.getRequestId(buf);
      RESP result = serializer.deserializeResponse(buf);
      callback.onRequestResult(requestId, result);
    } else if (msgType == MessageType.REQUEST_FAILURE) {
      RequestFailure failure = MessageSerializer.deserializeRequestFailure(buf);
      callback.onRequestFailure(failure.getRequestId(), failure.getCause());
    } else if (msgType == MessageType.SERVER_FAILURE) {
      throw MessageSerializer.deserializeServerFailure(buf);
    } else {
      throw new IllegalStateException("Unexpected response type '" + msgType + "'");
    }
  } catch (Throwable t1) {
    try {
      callback.onFailure(t1);
    } catch (Throwable t2) {
      LOG.error("Failed to notify callback about failure", t2);
    }
  } finally {
    ReferenceCountUtil.release(msg);
  }
}

@Override
public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
  try {
    ByteBuf buf = (ByteBuf) msg;
    MessageType msgType = MessageSerializer.deserializeHeader(buf);
    if (msgType == MessageType.REQUEST_RESULT) {
      long requestId = MessageSerializer.getRequestId(buf);
      RESP result = serializer.deserializeResponse(buf);
      callback.onRequestResult(requestId, result);
    } else if (msgType == MessageType.REQUEST_FAILURE) {
      RequestFailure failure = MessageSerializer.deserializeRequestFailure(buf);
      callback.onRequestFailure(failure.getRequestId(), failure.getCause());
    } else if (msgType == MessageType.SERVER_FAILURE) {
      throw MessageSerializer.deserializeServerFailure(buf);
    } else {
      throw new IllegalStateException("Unexpected response type '" + msgType + "'");
    }
  } catch (Throwable t1) {
    try {
      callback.onFailure(t1);
    } catch (Throwable t2) {
      LOG.error("Failed to notify callback about failure", t2);
    }
  } finally {
    ReferenceCountUtil.release(msg);
  }
}

final MessageType msgType = MessageSerializer.deserializeHeader(buf);
requestId = MessageSerializer.getRequestId(buf);

final MessageType msgType = MessageSerializer.deserializeHeader(buf);
requestId = MessageSerializer.getRequestId(buf);