001package ca.uhn.fhir.jpa.bulk.export.job;
002
003/*-
004 * #%L
005 * HAPI FHIR JPA Server
006 * %%
007 * Copyright (C) 2014 - 2021 Smile CDR, Inc.
008 * %%
009 * Licensed under the Apache License, Version 2.0 (the "License");
010 * you may not use this file except in compliance with the License.
011 * You may obtain a copy of the License at
012 *
013 *      http://www.apache.org/licenses/LICENSE-2.0
014 *
015 * Unless required by applicable law or agreed to in writing, software
016 * distributed under the License is distributed on an "AS IS" BASIS,
017 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
018 * See the License for the specific language governing permissions and
019 * limitations under the License.
020 * #L%
021 */
022
023import ca.uhn.fhir.interceptor.model.RequestPartitionId;
024import ca.uhn.fhir.jpa.batch.log.Logs;
025import ca.uhn.fhir.jpa.dao.IResultIterator;
026import ca.uhn.fhir.jpa.dao.ISearchBuilder;
027import ca.uhn.fhir.jpa.model.search.SearchRuntimeDetails;
028import ca.uhn.fhir.jpa.searchparam.SearchParameterMap;
029import ca.uhn.fhir.rest.api.server.storage.ResourcePersistentId;
030import org.slf4j.Logger;
031
032import java.util.HashSet;
033import java.util.Iterator;
034import java.util.List;
035import java.util.Set;
036
037/**
038 * Basic Bulk Export implementation which simply reads all type filters and applies them, along with the _since param
039 * on a given resource type.
040 */
041public class BulkItemReader extends BaseJpaBulkItemReader {
042        private static final Logger ourLog = Logs.getBatchTroubleshootingLog();
043
044        @Override
045        protected Iterator<ResourcePersistentId> getResourcePidIterator() {
046                ourLog.info("Bulk export assembling export of type {} for job {}", myResourceType, myJobUUID);
047                Set<ResourcePersistentId> myReadPids = new HashSet<>();
048
049                List<SearchParameterMap> map = createSearchParameterMapsForResourceType();
050                ISearchBuilder sb = getSearchBuilderForLocalResourceType();
051
052                for (SearchParameterMap spMap: map) {
053                        ourLog.debug("About to evaluate query {}", spMap.toNormalizedQueryString(myContext));
054                        IResultIterator myResultIterator = sb.createQuery(spMap, new SearchRuntimeDetails(null, myJobUUID), null, RequestPartitionId.allPartitions());
055                        while (myResultIterator.hasNext()) {
056                                myReadPids.add(myResultIterator.next());
057                        }
058                }
059                return myReadPids.iterator();
060        }
061
062}