How to use

stop

method

in

org.apache.samza.system.kafka.KafkaSystemConsumer

@Override
public Map<SystemStreamPartition, List<IncomingMessageEnvelope>> poll(
  Set<SystemStreamPartition> systemStreamPartitions, long timeout) throws InterruptedException {
 // check if the proxy is running
 if (!proxy.isRunning()) {
  LOG.info("{}: KafkaConsumerProxy is not running. Stopping the consumer.", this);
  stop();
  String message = String.format("%s: KafkaConsumerProxy has stopped.", this);
  throw new SamzaException(message, proxy.getFailureCause());
 }
 return super.poll(systemStreamPartitions, timeout);
}

@Override
public Map<SystemStreamPartition, List<IncomingMessageEnvelope>> poll(
  Set<SystemStreamPartition> systemStreamPartitions, long timeout) throws InterruptedException {
 // check if the proxy is running
 if (!proxy.isRunning()) {
  stop();
  String message = String.format("%s: KafkaConsumerProxy has stopped.", this);
  throw new SamzaException(message, proxy.getFailureCause());
 }
 return super.poll(systemStreamPartitions, timeout);
}

@Override
public Map<SystemStreamPartition, List<IncomingMessageEnvelope>> poll(
  Set<SystemStreamPartition> systemStreamPartitions, long timeout) throws InterruptedException {
 // check if the proxy is running
 if (!proxy.isRunning()) {
  stop();
  String message = String.format("%s: KafkaConsumerProxy has stopped.", this);
  throw new SamzaException(message, proxy.getFailureCause());
 }
 return super.poll(systemStreamPartitions, timeout);
}

@Test
public void testFetchThresholdShouldDivideEvenlyAmongPartitions() {
 final KafkaSystemConsumer consumer = createConsumer(FETCH_THRESHOLD_MSGS, FETCH_THRESHOLD_BYTES);
 final int partitionsNum = 50;
 for (int i = 0; i < partitionsNum; i++) {
  consumer.register(new SystemStreamPartition(TEST_SYSTEM, TEST_STREAM, new Partition(i)), "0");
 }
 consumer.start();
 Assert.assertEquals(Long.valueOf(FETCH_THRESHOLD_MSGS) / partitionsNum, consumer.perPartitionFetchThreshold);
 Assert.assertEquals(Long.valueOf(FETCH_THRESHOLD_BYTES) / 2 / partitionsNum,
   consumer.perPartitionFetchThresholdBytes);
 consumer.stop();
}

Assert.assertEquals(ime1Size + ime11Size, consumer.getMessagesSizeInQueue(ssp1));
consumer.stop();

Assert.assertEquals(ime1Size + ime11Size, consumer.getMessagesSizeInQueue(ssp1));
consumer.stop();