How to use

getConsumeAs

method

in

uk.gov.gchq.gaffer.operation.impl.add.AddElementsFromKafka

@Override
public Object doOperation(final AddElementsFromKafka op, final Context context, final Store store) throws OperationException {
  final StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
  if (null != op.getParallelism()) {
    env.setParallelism(op.getParallelism());
  }
  final GafferMapFunction function = new GafferMapFunction(op.getConsumeAs(), op.getElementGenerator());
  final DataStream<Element> builder = env.addSource(
      new FlinkKafkaConsumer010<>(
          op.getTopic(),
          function.getSerialisationType(),
          createFlinkProperties(op)))
      .flatMap(function);
  if (Boolean.parseBoolean(op.getOption(FlinkConstants.SKIP_REBALANCING))) {
    builder.addSink(new GafferSink(op, store));
  } else {
    builder.rebalance().addSink(new GafferSink(op, store));
  }
  try {
    env.execute(op.getClass().getSimpleName() + "-" + op.getGroupId() + "-" + op.getTopic());
  } catch (final Exception e) {
    throw new OperationException("Failed to add elements from Kafka topic: " + op.getTopic(), e);
  }
  return null;
}