43 lines
1.7 KiB
Java
43 lines
1.7 KiB
Java
package com.zdjizhi.utils.kafka;
|
|
|
|
import com.zdjizhi.common.StreamAggregateConfig;
|
|
import org.apache.flink.api.common.serialization.SimpleStringSchema;
|
|
import org.apache.flink.streaming.connectors.kafka.FlinkKafkaConsumer;
|
|
import org.apache.kafka.clients.consumer.ConsumerConfig;
|
|
import org.apache.kafka.common.config.SslConfigs;
|
|
|
|
import java.util.Properties;
|
|
|
|
/**
|
|
* @author qidaijie
|
|
* @Package com.zdjizhi.utils.kafka
|
|
* @Description:
|
|
* @date 2021/6/813:54
|
|
*/
|
|
public class KafkaConsumer {
|
|
private static Properties createConsumerConfig() {
|
|
Properties properties = new Properties();
|
|
properties.put("bootstrap.servers", StreamAggregateConfig.SOURCE_KAFKA_SERVERS);
|
|
properties.put("group.id", StreamAggregateConfig.GROUP_ID);
|
|
properties.put("session.timeout.ms", StreamAggregateConfig.SESSION_TIMEOUT_MS);
|
|
properties.put("max.poll.records", StreamAggregateConfig.MAX_POLL_RECORDS);
|
|
properties.put("max.partition.fetch.bytes", StreamAggregateConfig.MAX_PARTITION_FETCH_BYTES);
|
|
properties.put("key.deserializer", "org.apache.kafka.common.serialization.StringDeserializer");
|
|
properties.put("value.deserializer", "org.apache.kafka.common.serialization.StringDeserializer");
|
|
|
|
CertUtils.chooseCert(StreamAggregateConfig.SOURCE_KAFKA_SERVERS, properties);
|
|
|
|
return properties;
|
|
}
|
|
|
|
public static FlinkKafkaConsumer<String> getKafkaConsumer() {
|
|
FlinkKafkaConsumer<String> kafkaConsumer = new FlinkKafkaConsumer<>(StreamAggregateConfig.SOURCE_KAFKA_TOPIC,
|
|
new SimpleStringSchema(), createConsumerConfig());
|
|
|
|
kafkaConsumer.setCommitOffsetsOnCheckpoints(true);
|
|
kafkaConsumer.setStartFromGroupOffsets();
|
|
|
|
return kafkaConsumer;
|
|
}
|
|
}
|