com.taosdata.jdbc.tmq.ConsumerRecords Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of taos-jdbcdriver Show documentation

TDengine JDBC Driver

There is a newer version: 3.4.0

package com.taosdata.jdbc.tmq;

import java.util.*;

public class ConsumerRecords implements Iterable> {

    public static final ConsumerRecords EMPTY = new ConsumerRecords<>(Collections.emptyMap());

    private long offset;
    private final Map>> records;

    public ConsumerRecords(Map>> records) {
        this.records = records;
    }

    public ConsumerRecords(long offset) {
        this.records = new HashMap<>();
        this.offset = offset;
    }

    protected long getOffset() {
        return offset;
    }

    public void put(TopicPartition tp, ConsumerRecord r) {
        if (records.containsKey(tp)) {
            records.get(tp).add(r);
        } else {
            ArrayList> list = new ArrayList<>();
            list.add(r);
            records.put(tp, list);
        }
    }

    public List> get(TopicPartition partition) {
        List> recs = this.records.get(partition);
        if (recs == null) return Collections.emptyList();
        else return Collections.unmodifiableList(recs);
    }

    @Override
    public Iterator> iterator() {
        return new Iterator>() {
            final Iterator>> iters = records.values().iterator();
            Iterator> current;

            @Override
            public boolean hasNext() {
                while (current == null || !current.hasNext()) {
                    if (iters.hasNext()) current = iters.next().iterator();
                    else return false;
                }
                return true;
            }

            @Override
            public ConsumerRecord next() {
                return current.next();
            }
        };
    }

    public boolean isEmpty() {
        return records.isEmpty();
    }

    @SuppressWarnings("unchecked")
    public static  ConsumerRecords emptyRecord() {
        return (ConsumerRecords) EMPTY;
    }
}