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.dao.search;
021
022import ca.uhn.fhir.i18n.Msg;
023import ca.uhn.fhir.model.primitive.IdDt;
024import ca.uhn.fhir.parser.IParser;
025import org.apache.commons.lang3.StringUtils;
026import org.hl7.fhir.instance.model.api.IBaseResource;
027
028/**
029 * Query result when fetching full resources from HSearch.
030 */
031public class ExtendedHSearchResourceProjection {
032        public static final String RESOURCE_NOT_STORED_ERROR = "Resource not stored in search index: ";
033
034        final long myPid;
035        final String myForcedId;
036        final String myResourceString;
037
038        public ExtendedHSearchResourceProjection(long thePid, String theForcedId, String theResourceString) {
039                if (StringUtils.isEmpty(theResourceString)) {
040                        throw new ResourceNotFoundInIndexException(Msg.code(2130) + RESOURCE_NOT_STORED_ERROR + thePid);
041                }
042                myPid = thePid;
043                myForcedId = theForcedId;
044                myResourceString = theResourceString;
045        }
046
047        public IBaseResource toResource(IParser theParser) {
048                IBaseResource result = theParser.parseResource(myResourceString);
049
050                IdDt id;
051                if (myForcedId != null) {
052                        id = new IdDt(myForcedId);
053                } else {
054                        id = new IdDt(myPid);
055                }
056                result.setId(id);
057
058                return result;
059        }
060
061        public long getPid() {
062                return myPid;
063        }
064}