001/*-
002 * #%L
003 * HAPI FHIR JPA Server
004 * %%
005 * Copyright (C) 2014 - 2024 Smile CDR, Inc.
006 * %%
007 * Licensed under the Apache License, Version 2.0 (the "License");
008 * you may not use this file except in compliance with the License.
009 * You may obtain a copy of the License at
010 *
011 *      http://www.apache.org/licenses/LICENSE-2.0
012 *
013 * Unless required by applicable law or agreed to in writing, software
014 * distributed under the License is distributed on an "AS IS" BASIS,
015 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
016 * See the License for the specific language governing permissions and
017 * limitations under the License.
018 * #L%
019 */
020package ca.uhn.fhir.jpa.search.builder;
021
022import ca.uhn.fhir.jpa.model.dao.JpaPid;
023import ca.uhn.fhir.jpa.search.builder.models.ResolvedSearchQueryExecutor;
024import jakarta.annotation.Nonnull;
025import org.apache.commons.lang3.Validate;
026
027import java.util.List;
028
029public class SearchQueryExecutors {
030
031        public static ISearchQueryExecutor limited(ISearchQueryExecutor theExecutor, long theLimit) {
032                Validate.isTrue(theLimit >= 0, "limit must be non-negative");
033
034                return new ISearchQueryExecutor() {
035                        long myCount = 0;
036
037                        @Override
038                        public void close() {
039                                theExecutor.close();
040                        }
041
042                        @Override
043                        public boolean hasNext() {
044                                return theExecutor.hasNext() && myCount < theLimit;
045                        }
046
047                        @Override
048                        public JpaPid next() {
049                                myCount += 1;
050                                return theExecutor.next();
051                        }
052                };
053        }
054
055        @Nonnull
056        public static ISearchQueryExecutor from(List<JpaPid> rawPids) {
057                return new ResolvedSearchQueryExecutor(rawPids);
058        }
059}