ca.uhn.fhir.jpa.searchparam.registry.RuntimeSearchParamCache Maven / Gradle / Ivy

Go to download
/*-
 * #%L
 * HAPI FHIR JPA - Search Parameters
 * %%
 * 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.searchparam.registry;

import ca.uhn.fhir.context.RuntimeSearchParam;
import ca.uhn.fhir.rest.server.util.ResourceSearchParams;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.Map;
import java.util.Set;

import static org.apache.commons.lang3.StringUtils.isNotBlank;

public class RuntimeSearchParamCache extends ReadOnlySearchParamCache {
	private static final Logger ourLog = LoggerFactory.getLogger(RuntimeSearchParamCache.class);

	protected RuntimeSearchParamCache() {}

	public void add(String theResourceName, String theName, RuntimeSearchParam theSearchParam) {
		ResourceSearchParams resourceSearchParams = getSearchParamMap(theResourceName);
		resourceSearchParams.put(theName, theSearchParam);
		String uri = theSearchParam.getUri();
		if (isNotBlank(uri)) {
			RuntimeSearchParam existingForUrl = myUrlToParam.get(uri);
			if (existingForUrl == theSearchParam) {
				// This is expected, since the same SP can span multiple resource types
				// so it may get added more than once by this method
				ourLog.trace("Search param was previously registered for url: {}", uri);
			} else if (existingForUrl != null) {
				ourLog.debug("Multiple search parameters have URL: {}", uri);
			} else {
				myUrlToParam.put(uri, theSearchParam);
			}
		}
		if (theSearchParam.getId() != null && theSearchParam.getId().hasIdPart()) {
			String value = theSearchParam.getId().toUnqualifiedVersionless().getValue();
			myUrlToParam.put(value, theSearchParam);
		}
	}

	public void remove(String theResourceName, String theName) {
		if (!myResourceNameToSpNameToSp.containsKey(theResourceName)) {
			return;
		}
		myResourceNameToSpNameToSp.get(theResourceName).remove(theName);
	}

	private void putAll(ReadOnlySearchParamCache theReadOnlySearchParamCache) {
		Set> builtInSps =
				theReadOnlySearchParamCache.myResourceNameToSpNameToSp.entrySet();
		for (Map.Entry nextBuiltInEntry : builtInSps) {
			for (RuntimeSearchParam nextParam : nextBuiltInEntry.getValue().values()) {
				String nextResourceName = nextBuiltInEntry.getKey();
				String nextParamName = nextParam.getName();
				add(nextResourceName, nextParamName, nextParam);
			}

			ourLog.trace(
					"Have {} built-in SPs for: {}", nextBuiltInEntry.getValue().size(), nextBuiltInEntry.getKey());
		}
	}

	public RuntimeSearchParam get(String theResourceName, String theParamName) {
		RuntimeSearchParam retVal = null;
		ResourceSearchParams params = myResourceNameToSpNameToSp.get(theResourceName);
		if (params != null) {
			retVal = params.get(theParamName);
		}
		return retVal;
	}

	public Set getResourceNameKeys() {
		return myResourceNameToSpNameToSp.keySet();
	}

	@Override
	protected ResourceSearchParams getSearchParamMap(String theResourceName) {
		return myResourceNameToSpNameToSp.computeIfAbsent(
				theResourceName, k -> new ResourceSearchParams(theResourceName));
	}

	public static RuntimeSearchParamCache fromReadOnlySearchParamCache(
			ReadOnlySearchParamCache theBuiltInSearchParams) {
		RuntimeSearchParamCache retVal = new RuntimeSearchParamCache();
		retVal.putAll(theBuiltInSearchParams);
		return retVal;
	}
}