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.predicate;
021
022import ca.uhn.fhir.interceptor.model.RequestPartitionId;
023import ca.uhn.fhir.jpa.model.config.PartitionSettings;
024import ca.uhn.fhir.jpa.model.entity.SearchParamPresentEntity;
025import ca.uhn.fhir.jpa.search.builder.sql.SearchQueryBuilder;
026import com.healthmarketscience.sqlbuilder.BinaryCondition;
027import com.healthmarketscience.sqlbuilder.Condition;
028import com.healthmarketscience.sqlbuilder.dbspec.basic.DbColumn;
029import org.springframework.beans.factory.annotation.Autowired;
030
031public class SearchParamPresentPredicateBuilder extends BaseJoiningPredicateBuilder {
032
033        private final DbColumn myColumnResourceId;
034        private final DbColumn myColumnHashPresence;
035
036        @Autowired
037        private PartitionSettings myPartitionSettings;
038
039        /**
040         * Constructor
041         */
042        public SearchParamPresentPredicateBuilder(SearchQueryBuilder theSearchSqlBuilder) {
043                super(theSearchSqlBuilder, theSearchSqlBuilder.addTable("HFJ_RES_PARAM_PRESENT"));
044                myColumnResourceId = getTable().addColumn("RES_ID");
045                myColumnHashPresence = getTable().addColumn("HASH_PRESENCE");
046        }
047
048        @Override
049        public DbColumn getResourceIdColumn() {
050                return myColumnResourceId;
051        }
052
053        public Condition createPredicateParamMissingForReference(
054                        String theResourceName, String theParamName, boolean theMissing, RequestPartitionId theRequestPartitionId) {
055                Long hash = SearchParamPresentEntity.calculateHashPresence(
056                                myPartitionSettings, theRequestPartitionId, theResourceName, theParamName, !theMissing);
057                BinaryCondition predicate = BinaryCondition.equalTo(myColumnHashPresence, generatePlaceholder(hash));
058                return combineWithRequestPartitionIdPredicate(theRequestPartitionId, predicate);
059        }
060}