io.quarkiverse.openfga.client.utils.PaginatedList Maven / Gradle / Ivy

Go to download

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

There is a newer version: 2.1.0

Show newest version

package io.quarkiverse.openfga.client.utils;

import java.util.List;
import java.util.Objects;
import java.util.concurrent.atomic.AtomicReference;
import java.util.function.BiFunction;

import javax.annotation.Nullable;

import io.quarkiverse.openfga.client.model.utils.Preconditions;
import io.smallrye.mutiny.Multi;
import io.smallrye.mutiny.Uni;

public final class PaginatedList {
    private final List items;
    @Nullable
    private final String token;

    public PaginatedList(List items, @Nullable String token) {
        this.items = Preconditions.parameterNonNull(items, "items");
        this.token = token;
    }

    public Boolean isNotLastPage() {
        return token != null && !token.isEmpty();
    }

    public static  Uni> collectAllPages(@Nullable Integer pageSize,
            BiFunction>> listGenerator) {
        return Multi.createBy()
                .repeating().uni(AtomicReference::new, lastToken -> {
                    return listGenerator.apply(pageSize, lastToken.get())
                            .onItem().invoke(list -> lastToken.set(list.getToken()));
                })
                .whilst(PaginatedList::isNotLastPage)
                .onItem().transformToIterable(PaginatedList::getItems)
                .collect().asList();
    }

    public List getItems() {
        return items;
    }

    @Nullable
    public String getToken() {
        return token;
    }

    @Override
    public boolean equals(@Nullable Object obj) {
        if (obj == this)
            return true;
        if (obj == null || obj.getClass() != this.getClass())
            return false;
        var that = (PaginatedList) obj;
        return Objects.equals(this.items, that.items) &&
                Objects.equals(this.token, that.token);
    }

    @Override
    public int hashCode() {
        return Objects.hash(items, token);
    }

    @Override
    public String toString() {
        return "PaginatedList[" +
                "items=" + items + ", " +
                "token=" + token + ']';
    }

}