Examples with PartitionToOffsetSum org.apache.kafka.streams.internals.generated.SubscriptionInfoData.PartitionToOffsetSum used on opensource projects

Example 1 with PartitionToOffsetSum

use of org.apache.kafka.streams.internals.generated.SubscriptionInfoData.PartitionToOffsetSum in project kafka by apache.

the class SubscriptionInfo method taskOffsetSums.

public Map<TaskId, Long> taskOffsetSums() {
    if (taskOffsetSumsCache == null) {
        taskOffsetSumsCache = new HashMap<>();
        if (data.version() >= MIN_VERSION_OFFSET_SUM_SUBSCRIPTION) {
            for (final TaskOffsetSum taskOffsetSum : data.taskOffsetSums()) {
                if (data.version() >= MIN_NAMED_TOPOLOGY_VERSION) {
                    taskOffsetSumsCache.put(new TaskId(taskOffsetSum.topicGroupId(), taskOffsetSum.partition(), taskOffsetSum.namedTopology()), taskOffsetSum.offsetSum());
                } else {
                    for (final PartitionToOffsetSum partitionOffsetSum : taskOffsetSum.partitionToOffsetSum()) {
                        taskOffsetSumsCache.put(new TaskId(taskOffsetSum.topicGroupId(), partitionOffsetSum.partition()), partitionOffsetSum.offsetSum());
                    }
                }
            }
        } else {
            prevTasks().forEach(taskId -> taskOffsetSumsCache.put(taskId, Task.LATEST_OFFSET));
            standbyTasks().forEach(taskId -> taskOffsetSumsCache.put(taskId, UNKNOWN_OFFSET_SUM));
        }
    }
    return taskOffsetSumsCache;
}