tech.ytsaurus.client.rows.ConsumerSource Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ytsaurus-client Show documentation

YTsaurus client

The newest version!

package tech.ytsaurus.client.rows;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.function.Consumer;
import java.util.function.IntFunction;

@FunctionalInterface
public interface ConsumerSource extends Consumer {

    default void setRowCount(int rowCount) {
        // do nothing
    }

    static  ConsumerSource wrap(Consumer consumer) {
        return consumer::accept;
    }


    static  ConsumerSourceRet list() {
        return list(ArrayList::new);
    }


    static  ConsumerSourceRet list(IntFunction> function) {
        return new ConsumerSourceRet() {

            private List list;

            @Override
            public void setRowCount(int rowCount) {
                list = function.apply(rowCount);
            }

            @Override
            public void accept(T t) {
                if (list == null) {
                    list = new ArrayList<>();
                }
                list.add(t);
            }

            @Override
            public List get() {
                return list != null ? list : Collections.emptyList();
            }
        };
    }

}