@Test public void shouldHandleControlRequest() { // given final int writtenLength = writeControlRequestToBuffer(buffer); when(mockControlMessageDispatcher.claim(any(ClaimedFragment.class), anyInt())) .thenAnswer(claimFragment(0)); // when final boolean isHandled = messageHandler.onRequest( serverOutput, DEFAULT_ADDRESS, buffer, 0, writtenLength, REQUEST_ID); // then assertThat(isHandled).isTrue(); verify(mockControlMessageDispatcher).claim(any(ClaimedFragment.class), anyInt()); int offset = fragmentOffset; controlMessageRequestHeaderDescriptor.wrap(sendBuffer, offset); assertThat(controlMessageRequestHeaderDescriptor.streamId()) .isEqualTo(DEFAULT_ADDRESS.getStreamId()); assertThat(controlMessageRequestHeaderDescriptor.requestId()).isEqualTo(REQUEST_ID); offset += ControlMessageRequestHeaderDescriptor.headerLength(); headerEncoder.wrap(sendBuffer, offset); offset += headerEncoder.encodedLength(); controlRequestDecoder.wrap( sendBuffer, offset, controlRequestDecoder.sbeBlockLength(), controlRequestDecoder.sbeSchemaVersion()); final byte[] requestData = readBytes(controlRequestDecoder::getData, controlRequestDecoder::dataLength); assertThat(requestData).isEqualTo(JOB_EVENT); }
@Test public void shouldHandleControlRequest() { // given final int writtenLength = writeControlRequestToBuffer(buffer); when(mockControlMessageDispatcher.claim(any(ClaimedFragment.class), anyInt())) .thenAnswer(claimFragment(0)); // when final boolean isHandled = messageHandler.onRequest( serverOutput, DEFAULT_ADDRESS, buffer, 0, writtenLength, REQUEST_ID); // then assertThat(isHandled).isTrue(); verify(mockControlMessageDispatcher).claim(any(ClaimedFragment.class), anyInt()); int offset = fragmentOffset; controlMessageRequestHeaderDescriptor.wrap(sendBuffer, offset); assertThat(controlMessageRequestHeaderDescriptor.streamId()) .isEqualTo(DEFAULT_ADDRESS.getStreamId()); assertThat(controlMessageRequestHeaderDescriptor.requestId()).isEqualTo(REQUEST_ID); offset += ControlMessageRequestHeaderDescriptor.headerLength(); headerEncoder.wrap(sendBuffer, offset); offset += headerEncoder.encodedLength(); controlRequestDecoder.wrap( sendBuffer, offset, controlRequestDecoder.sbeBlockLength(), controlRequestDecoder.sbeSchemaVersion()); final byte[] requestData = readBytes(controlRequestDecoder::getData, controlRequestDecoder::dataLength); assertThat(requestData).isEqualTo(JOB_EVENT); }
@Override public int onFragment( DirectBuffer buffer, int offset, int length, int streamId, boolean isMarkedFailed) { requestHeaderDescriptor.wrap(buffer, offset); final long requestId = requestHeaderDescriptor.requestId(); final int requestStreamId = requestHeaderDescriptor.streamId(); offset += ControlMessageRequestHeaderDescriptor.headerLength(); messageHeaderDecoder.wrap(requestBuffer, 0); offset += messageHeaderDecoder.encodedLength(); requestDecoder.wrap( buffer, offset, requestDecoder.sbeBlockLength(), requestDecoder.sbeSchemaVersion()); final ControlMessageType messageType = requestDecoder.messageType(); lastRequestMessageType(messageType); final int partitionId = requestDecoder.partitionId(); ensureBufferCapacity(requestDecoder.dataLength()); requestDecoder.getData(requestBuffer, 0, requestDecoder.dataLength()); final ControlMessageHandler handler = handlersByTypeId.get(messageType.value()); if (handler != null) { handler.handle(actor, partitionId, requestBuffer, requestId, requestStreamId); } else { sendResponse( actor, () -> errorResponseWriter .unsupportedMessage( getLastRequestMessageType().name(), handlersByTypeId.keySet().toArray()) .tryWriteResponseOrLogFailure(requestStreamId, requestId)); } return FragmentHandler.CONSUME_FRAGMENT_RESULT; }
@Override public int onFragment( DirectBuffer buffer, int offset, int length, int streamId, boolean isMarkedFailed) { requestHeaderDescriptor.wrap(buffer, offset); final long requestId = requestHeaderDescriptor.requestId(); final int requestStreamId = requestHeaderDescriptor.streamId(); offset += ControlMessageRequestHeaderDescriptor.headerLength(); messageHeaderDecoder.wrap(requestBuffer, 0); offset += messageHeaderDecoder.encodedLength(); requestDecoder.wrap( buffer, offset, requestDecoder.sbeBlockLength(), requestDecoder.sbeSchemaVersion()); final ControlMessageType messageType = requestDecoder.messageType(); lastRequestMessageType(messageType); final int partitionId = requestDecoder.partitionId(); ensureBufferCapacity(requestDecoder.dataLength()); requestDecoder.getData(requestBuffer, 0, requestDecoder.dataLength()); final ControlMessageHandler handler = handlersByTypeId.get(messageType.value()); if (handler != null) { handler.handle(actor, partitionId, requestBuffer, requestId, requestStreamId); } else { sendResponse( actor, () -> errorResponseWriter .unsupportedMessage( getLastRequestMessageType().name(), handlersByTypeId.keySet().toArray()) .tryWriteResponseOrLogFailure(requestStreamId, requestId)); } return FragmentHandler.CONSUME_FRAGMENT_RESULT; }