ca.uhn.fhir.model.api.PagingIterator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of hapi-fhir-base Show documentation

Builds the hapi fhir base. Requires Common lang, commons-text, and guava be built first.

There is a newer version: 7.0.2-r12

/*-
 * #%L
 * HAPI FHIR - Core Library
 * %%
 * Copyright (C) 2014 - 2024 Smile CDR, Inc.
 * %%
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * #L%
 */
package ca.uhn.fhir.model.api;

import ca.uhn.fhir.i18n.Msg;

import java.util.Iterator;
import java.util.LinkedList;
import java.util.NoSuchElementException;
import java.util.function.Consumer;

public class PagingIterator implements Iterator {

	public interface PageFetcher {
		void fetchNextPage(int thePageIndex, int theBatchSize, Consumer theConsumer);
	}

	static final int PAGE_SIZE = 100;

	private int myPage;

	private boolean myIsFinished;

	private final LinkedList myCurrentBatch = new LinkedList<>();

	private final PageFetcher myFetcher;

	public PagingIterator(PageFetcher theFetcher) {
		myFetcher = theFetcher;
	}

	@Override
	public boolean hasNext() {
		fetchNextBatch();

		return !myCurrentBatch.isEmpty();
	}

	@Override
	public T next() {
		fetchNextBatch();

		if (myCurrentBatch.isEmpty()) {
			throw new NoSuchElementException(Msg.code(2098) + " Nothing to fetch");
		}

		return myCurrentBatch.remove(0);
	}

	private void fetchNextBatch() {
		if (!myIsFinished && myCurrentBatch.isEmpty()) {
			myFetcher.fetchNextPage(myPage, PAGE_SIZE, myCurrentBatch::add);
			myPage++;
			myIsFinished = myCurrentBatch.size() < PAGE_SIZE;
		}
	}
}