Examples with KafkaSourceReaderMetrics org.apache.flink.connector.kafka.source.metrics.KafkaSourceReaderMetrics used on opensource projects

Example 1 with KafkaSourceReaderMetrics

use of org.apache.flink.connector.kafka.source.metrics.KafkaSourceReaderMetrics in project flink by apache.

the class KafkaSource method createReader.

@VisibleForTesting
SourceReader<OUT, KafkaPartitionSplit> createReader(SourceReaderContext readerContext, Consumer<Collection<String>> splitFinishedHook) throws Exception {
    FutureCompletingBlockingQueue<RecordsWithSplitIds<ConsumerRecord<byte[], byte[]>>> elementsQueue = new FutureCompletingBlockingQueue<>();
    deserializationSchema.open(new DeserializationSchema.InitializationContext() {

        @Override
        public MetricGroup getMetricGroup() {
            return readerContext.metricGroup().addGroup("deserializer");
        }

        @Override
        public UserCodeClassLoader getUserCodeClassLoader() {
            return readerContext.getUserCodeClassLoader();
        }
    });
    final KafkaSourceReaderMetrics kafkaSourceReaderMetrics = new KafkaSourceReaderMetrics(readerContext.metricGroup());
    Supplier<KafkaPartitionSplitReader> splitReaderSupplier = () -> new KafkaPartitionSplitReader(props, readerContext, kafkaSourceReaderMetrics);
    KafkaRecordEmitter<OUT> recordEmitter = new KafkaRecordEmitter<>(deserializationSchema);
    return new KafkaSourceReader<>(elementsQueue, new KafkaSourceFetcherManager(elementsQueue, splitReaderSupplier::get, splitFinishedHook), recordEmitter, toConfiguration(props), readerContext, kafkaSourceReaderMetrics);
}

Example 2 with KafkaSourceReaderMetrics

use of org.apache.flink.connector.kafka.source.metrics.KafkaSourceReaderMetrics in project flink by apache.

the class KafkaPartitionSplitReaderTest method createReader.

private KafkaPartitionSplitReader createReader(Properties additionalProperties, SourceReaderMetricGroup sourceReaderMetricGroup) {
    Properties props = new Properties();
    props.putAll(KafkaSourceTestEnv.getConsumerProperties(ByteArrayDeserializer.class));
    props.setProperty(ConsumerConfig.AUTO_OFFSET_RESET_CONFIG, "none");
    if (!additionalProperties.isEmpty()) {
        props.putAll(additionalProperties);
    }
    KafkaSourceReaderMetrics kafkaSourceReaderMetrics = new KafkaSourceReaderMetrics(sourceReaderMetricGroup);
    return new KafkaPartitionSplitReader(props, new TestingReaderContext(new Configuration(), sourceReaderMetricGroup), kafkaSourceReaderMetrics);
}