001/*-
002 * #%L
003 * HAPI FHIR Storage api
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.dao.tx;
021
022import ca.uhn.fhir.interceptor.model.RequestPartitionId;
023import jakarta.annotation.Nullable;
024import org.apache.commons.lang3.tuple.Pair;
025import org.springframework.transaction.support.SimpleTransactionStatus;
026import org.springframework.transaction.support.TransactionCallback;
027
028import java.util.HashSet;
029import java.util.Set;
030
031/**
032 * A transaction service implementation that does not actually
033 * wrap any transactions. This is mostly intended for tests but
034 * could be used in non-transactional systems too.
035 */
036public class NonTransactionalHapiTransactionService extends HapiTransactionService {
037
038        private Set<Pair<RequestPartitionId, RequestPartitionId>> myNonCompatiblePartitions = new HashSet<>();
039
040        @Nullable
041        @Override
042        protected <T> T doExecute(ExecutionBuilder theExecutionBuilder, TransactionCallback<T> theCallback) {
043                return theCallback.doInTransaction(new SimpleTransactionStatus());
044        }
045
046        @Override
047        public boolean isCompatiblePartition(
048                        RequestPartitionId theRequestPartitionId, RequestPartitionId theOtherRequestPartitionId) {
049                if (myNonCompatiblePartitions.contains(Pair.of(theRequestPartitionId, theOtherRequestPartitionId))) {
050                        return false;
051                } else return !myNonCompatiblePartitions.contains(Pair.of(theOtherRequestPartitionId, theRequestPartitionId));
052        }
053
054        public void clearNonCompatiblePartitions() {
055                myNonCompatiblePartitions.clear();
056        }
057
058        public void addNonCompatiblePartition(RequestPartitionId thePartitionA, RequestPartitionId thePartitionB) {
059                myNonCompatiblePartitions.add(Pair.of(thePartitionA, thePartitionB));
060        }
061}