<dependency>
<groupId>org.apache.kafka</groupId>
<artifactId>kafka-clients</artifactId>
<version>3.4.0</version>
<exclusions>
<exclusion>
<groupId>org.slf4j</groupId>
<artifactId>slf4j-log4j12</artifactId>
</exclusion>
</exclusions>
</dependency>
package com.orientalsemi.oceanplus.dataetl.util;
import lombok.extern.slf4j.Slf4j;
import org.apache.kafka.clients.consumer.KafkaConsumer;
import org.apache.kafka.clients.consumer.OffsetAndMetadata;
import org.apache.kafka.common.PartitionInfo;
import org.apache.kafka.common.TopicPartition;
import java.util.*;
@Slf4j
public class KafkaUtil {
public static long KafkaConsumeLag(String server, String groupId, String topic){
Map endOffsetMap = new HashMap();
Map commitOffsetMap = new HashMap();
Properties consumeProps = getConsumeProperties(groupId, server);
log.info("consumer properties:" + consumeProps);
KafkaConsumer consumer = new KafkaConsumer(consumeProps);
List topicPartitions = new ArrayList();
List partitionsFor = consumer.partitionsFor(topic);
for (Object map : partitionsFor) {
PartitionInfo partitionInfo = (PartitionInfo) map;
TopicPartition topicPartition = new TopicPartition(partitionInfo.topic(), partitionInfo.partition());
topicPartitions.add(topicPartition);
}
Map endOffsets = consumer.endOffsets(topicPartitions);
for (Object map : endOffsets.keySet()) {
TopicPartition partitionInfo = (TopicPartition) map;
endOffsetMap.put(partitionInfo.partition(), endOffsets.get(partitionInfo));
}
for (Object partitionId : endOffsetMap.keySet()) {
log.info(String.format("at %s, topic:%s, partition:%s, logSize:%s", System.currentTimeMillis(), topic, partitionId, endOffsetMap.get(partitionId)));
}
for (Object map : topicPartitions) {
TopicPartition topicAndPartition = (TopicPartition) map;
OffsetAndMetadata committed = consumer.committed(topicAndPartition);
commitOffsetMap.put(topicAndPartition.partition(), committed.offset());
}
long lagSum = 0l;
if (endOffsetMap.size() == commitOffsetMap.size()) {
for (Object data : endOffsetMap.keySet()) {
long endOffSet = (long) endOffsetMap.get(data);
long commitOffSet = (long) commitOffsetMap.get(data);
long diffOffset = endOffSet - commitOffSet;
lagSum += diffOffset;
log.info("Topic:" + topic + ", groupID:" + groupId + ", partition:" + data + ", endOffset:" + endOffSet + ", commitOffset:" + commitOffSet + ", diffOffset:" + diffOffset);
}
log.info("Topic:" + topic + ", groupID:" + groupId + ", LAG:" + lagSum);
} else {
log.info("this topic partitions lost");
}
consumer.close();
return lagSum;
}
public static Properties getConsumeProperties(String groupID, String bootstrap_server) {
Properties props = new Properties();
props.put("group.id", groupID);
props.put("bootstrap.servers", bootstrap_server);
props.put("key.deserializer", "org.apache.kafka.common.serialization.StringDeserializer");
props.put("value.deserializer", "org.apache.kafka.common.serialization.StringDeserializer");
return props;
}
}