
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}