ca.uhn.fhir.jpa.dao.IdSubstitutionMap Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of hapi-fhir-storage Show documentation
This project provides services useful for fhir repository storage. Actual repository storage services are not in this project, just the api and helper libraries.
The newest version!
/*-
 * #%L
 * HAPI FHIR Storage api
 * %%
 * Copyright (C) 2014 - 2024 Smile CDR, Inc.
 * %%
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * #L%
 */
package ca.uhn.fhir.jpa.dao;

import com.google.common.collect.Multimap;
import com.google.common.collect.MultimapBuilder;
import org.apache.commons.lang3.tuple.Pair;
import org.hl7.fhir.instance.model.api.IIdType;

import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.stream.Collectors;

public class IdSubstitutionMap {

	private final Map myMap = new HashMap<>();
	private final Multimap myReverseMap =
			MultimapBuilder.hashKeys().arrayListValues().build();

	public boolean containsSource(IIdType theId) {
		if (theId.isLocal()) {
			return false;
		}
		return myMap.containsKey(new Entry(theId));
	}

	public boolean containsSource(String theId) {
		return myMap.containsKey(new Entry(theId));
	}

	public boolean containsTarget(IIdType theId) {
		return myReverseMap.containsKey(new Entry(theId));
	}

	public boolean containsTarget(String theId) {
		return myReverseMap.containsKey(new Entry(theId));
	}

	public IIdType getForSource(IIdType theId) {
		Entry target = myMap.get(new Entry(theId));
		if (target != null) {
			assert target.myId != null;
			return target.myId;
		}
		return null;
	}

	public IIdType getForSource(String theId) {
		Entry target = myMap.get(new Entry(theId));
		if (target != null) {
			assert target.myId != null;
			return target.myId;
		}
		return null;
	}

	public List> entrySet() {
		return myMap.entrySet().stream()
				.map(t -> Pair.of(t.getKey().myId, t.getValue().myId))
				.collect(Collectors.toList());
	}

	public void put(IIdType theSource, IIdType theTarget) {
		myMap.put(new Entry(theSource), new Entry(theTarget));
		myReverseMap.put(new Entry(theTarget), new Entry(theSource));
	}

	public boolean isEmpty() {
		return myMap.isEmpty();
	}

	/**
	 * Updates all targets of the map with a new id value if the input id has
	 * the same ResourceType and IdPart as the target id.
	 */
	public void updateTargets(IIdType theNewId) {
		if (theNewId == null) {
			return;
		}
		String newUnqualifiedVersionLessId = theNewId.toUnqualifiedVersionless().getValue();
		entrySet().stream()
				.map(Pair::getValue)
				.filter(targetId ->
						Objects.equals(targetId.toUnqualifiedVersionless().getValue(), newUnqualifiedVersionLessId))
				.forEach(targetId -> targetId.setValue(theNewId.getValue()));
	}

	private static class Entry {

		private final String myUnversionedId;
		private final IIdType myId;

		private Entry(String theId) {
			myId = null;
			myUnversionedId = theId;
		}

		private Entry(IIdType theId) {
			String unversionedId = toVersionlessValue(theId);
			myUnversionedId = unversionedId;
			myId = theId;
		}

		@Override
		public boolean equals(Object theOther) {
			if (theOther instanceof Entry) {
				String otherUnversionedId = ((Entry) theOther).myUnversionedId;
				if (myUnversionedId.equals(otherUnversionedId)) {
					return true;
				}
			}
			return false;
		}

		@Override
		public int hashCode() {
			return myUnversionedId.hashCode();
		}
	}

	static String toVersionlessValue(IIdType theId) {
		boolean isPlaceholder = theId.getValue().startsWith("urn:");
		String unversionedId;
		if (isPlaceholder || (!theId.hasBaseUrl() && !theId.hasVersionIdPart()) || !theId.hasResourceType()) {
			unversionedId = theId.getValue();
		} else {
			unversionedId = theId.toUnqualifiedVersionless().getValue();
		}
		return unversionedId;
	}
}