@Override public Object getRawResponse() { return response.getRawResponse(); }
@Override public String getStringResponse() { return response.getStringResponse(); }
@Override public void onSuccess(WriteResponse writeResponse) { try { attempt.ackable.ack(); AsyncWriterManager.this.recordsSuccess.mark(); if (writeResponse.bytesWritten() > 0) { AsyncWriterManager.this.bytesWritten.mark(writeResponse.bytesWritten()); } if (AsyncWriterManager.this.dataWriterTimer.isPresent()) { AsyncWriterManager.this.dataWriterTimer.get() .update(System.nanoTime() - attempt.getPrevAttemptTimestampNanos(), TimeUnit.NANOSECONDS); } } finally { AsyncWriterManager.this.writePermits.release(); } }
@Override public Object getRawResponse() { return response.getRawResponse(); }
@Override public String getStringResponse() { return response.getStringResponse(); }
@Override public void onSuccess(WriteResponse writeResponse) { try { attempt.ackable.ack(); AsyncWriterManager.this.recordsSuccess.mark(); if (writeResponse.bytesWritten() > 0) { AsyncWriterManager.this.bytesWritten.mark(writeResponse.bytesWritten()); } if (AsyncWriterManager.this.dataWriterTimer.isPresent()) { AsyncWriterManager.this.dataWriterTimer.get() .update(System.nanoTime() - attempt.getPrevAttemptTimestampNanos(), TimeUnit.NANOSECONDS); } } finally { AsyncWriterManager.this.writePermits.release(); } }
@Test public void testSingleRecord() throws IOException { // mock eventhub data writer Properties props = new Properties(); EventhubDataWriter eventhubDataWriter = Mockito.spy(new EventhubDataWriter(props, mockHttpClient)); Mockito.doNothing().when(eventhubDataWriter).refreshSignature(); String record = "abcdefgh"; WriteResponse<Integer> writeResponse = eventhubDataWriter.write(record); int returnCode = writeResponse.getRawResponse(); Assert.assertEquals(returnCode, 201); } }
@Test public void testStringSerialization() throws IOException, InterruptedException, ExecutionException { String topic = "testStringSerialization08"; _kafkaTestHelper.provisionTopic(topic); Properties props = new Properties(); props.setProperty(KafkaWriterConfigurationKeys.KAFKA_TOPIC, topic); props.setProperty(KafkaWriterConfigurationKeys.KAFKA_PRODUCER_CONFIG_PREFIX+"bootstrap.servers", "localhost:" + _kafkaTestHelper.getKafkaServerPort()); props.setProperty(KafkaWriterConfigurationKeys.KAFKA_PRODUCER_CONFIG_PREFIX+"value.serializer", "org.apache.kafka.common.serialization.StringSerializer"); Kafka09DataWriter<String> kafka09DataWriter = new Kafka09DataWriter<String>(props); String messageString = "foobar"; WriteCallback callback = mock(WriteCallback.class); Future<WriteResponse> future; try { future = kafka09DataWriter.write(messageString, callback); kafka09DataWriter.flush(); verify(callback, times(1)).onSuccess(isA(WriteResponse.class)); verify(callback, never()).onFailure(isA(Exception.class)); Assert.assertTrue(future.isDone(), "Future should be done"); System.out.println(future.get().getStringResponse()); byte[] message = _kafkaTestHelper.getIteratorForTopic(topic).next().message(); String messageReceived = new String(message); Assert.assertEquals(messageReceived, messageString); } finally { kafka09DataWriter.close(); } }
@Override public Object getRawResponse() { return response.getRawResponse(); }
@Override public String getStringResponse() { return response.getStringResponse(); }
@Override public Object getRawResponse() { return response.getRawResponse(); }
@Override public String getStringResponse() { return response.getStringResponse(); }