global.maplink.geocode.schema.suggestions.SuggestionsResult Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of geocode-schema Show documentation

The newest version!

package global.maplink.geocode.schema.suggestions;

import lombok.*;

import java.util.Iterator;
import java.util.List;
import java.util.Optional;
import java.util.stream.Stream;
import java.util.stream.StreamSupport;

import static java.util.Collections.emptyIterator;
import static java.util.Collections.emptyList;
import static java.util.Comparator.naturalOrder;
import static java.util.Objects.requireNonNull;
import static java.util.Optional.empty;
import static java.util.stream.Collectors.toList;

@SuppressWarnings("ConstantConditions")
@Data
@AllArgsConstructor
@NoArgsConstructor(force = true)
@Builder
public class SuggestionsResult implements Iterable {

    public static final SuggestionsResult EMPTY = new SuggestionsResult(0, emptyList());

    private final int found;

    @Singular
    private final List results;

    public boolean isEmpty() {
        return results == null || results.isEmpty();
    }

    public Suggestion getMostRelevant() {
        if (isEmpty()) return null;
        return results.stream().max(naturalOrder()).orElse(results.get(0));
    }

    public Optional getById(String id) {
        requireNonNull(id);
        if (isEmpty()) return empty();

        return results.stream().filter(r -> id.equals(r.getId())).findAny();
    }

    public SuggestionsResult joinTo(SuggestionsResult anotherResult) {
        return new SuggestionsResult(
                found + anotherResult.found,
                Stream.concat(stream(), anotherResult.stream()).collect(toList())
        );
    }

    @Override
    public Iterator iterator() {
        if (isEmpty()) {
            return emptyIterator();
        }
        return results.iterator();
    }

    public Stream stream() {
        return StreamSupport.stream(this.spliterator(), false);
    }

    public Stream parallelStream() {
        return StreamSupport.stream(this.spliterator(), true);
    }
}