001/*
002 * #%L
003 * HAPI FHIR JPA Server
004 * %%
005 * Copyright (C) 2014 - 2025 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;
021
022import ca.uhn.fhir.jpa.entity.Search;
023import ca.uhn.fhir.rest.api.server.IBundleProvider;
024import ca.uhn.fhir.rest.api.server.RequestDetails;
025import ca.uhn.fhir.rest.server.method.ResponsePage;
026import jakarta.annotation.Nonnull;
027import org.hl7.fhir.instance.model.api.IBaseResource;
028
029import java.util.List;
030import java.util.function.Predicate;
031
032/**
033 * PersistedJpaBundleProvider accepting a resource filtering predicate
034 */
035public class PredicatedPersistedJpaBundleProvider extends PersistedJpaBundleProvider implements IBundleProvider {
036
037        private final Predicate<? super IBaseResource> myPredicate;
038
039        public PredicatedPersistedJpaBundleProvider(
040                        RequestDetails theRequest, Search theSearch, Predicate<? super IBaseResource> thePredicate) {
041                super(theRequest, theSearch);
042                myPredicate = thePredicate;
043        }
044
045        @Override
046        public List<IBaseResource> getResources(
047                        int theFromIndex, int theToIndex, @Nonnull ResponsePage.ResponsePageBuilder theResponsePageBuilder) {
048
049                List<IBaseResource> allResources = super.getResources(theFromIndex, theToIndex, theResponsePageBuilder);
050
051                return allResources.stream().filter(myPredicate).toList();
052        }
053}