dev.dsf.fhir.search.parameters.EndpointAddress Maven / Gradle / Ivy

Go to download
package dev.dsf.fhir.search.parameters;

import java.sql.Array;
import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.util.Objects;

import org.hl7.fhir.r4.model.Endpoint;
import org.hl7.fhir.r4.model.Enumerations.SearchParamType;

import dev.dsf.fhir.function.BiFunctionWithSqlException;
import dev.dsf.fhir.search.SearchQueryParameter.SearchParameterDefinition;
import dev.dsf.fhir.search.parameters.basic.AbstractCanonicalUrlParameter;

@SearchParameterDefinition(name = EndpointAddress.PARAMETER_NAME, definition = "http://hl7.org/fhir/SearchParameter/Endpoint-address", type = SearchParamType.URI, documentation = "The address (url) of the endpoint")
public class EndpointAddress extends AbstractCanonicalUrlParameter
{
	public static final String PARAMETER_NAME = "address";

	public EndpointAddress()
	{
		super(Endpoint.class, PARAMETER_NAME);
	}

	@Override
	public String getFilterQuery()
	{
		return switch (valueAndType.type)
		{
			case PRECISE -> "endpoint->>'address' = ?";
			case BELOW -> "endpoint->>'address' LIKE ?";
		};
	}

	@Override
	public int getSqlParameterCount()
	{
		return 1;
	}

	@Override
	public void modifyStatement(int parameterIndex, int subqueryParameterIndex, PreparedStatement statement,
			BiFunctionWithSqlException arrayCreator) throws SQLException
	{
		switch (valueAndType.type)
		{
			case PRECISE:
				statement.setString(parameterIndex, valueAndType.url);
				return;

			case BELOW:
				statement.setString(parameterIndex, valueAndType.url + "%");
				return;
		}
	}

	@Override
	protected boolean resourceMatches(Endpoint resource)
	{
		return resource.hasAddress() && switch (valueAndType.type)
		{
			case PRECISE -> Objects.equals(resource.getAddress(), valueAndType.url);
			case BELOW -> resource.getAddress().startsWith(valueAndType.url);
		};
	}

	@Override
	protected String getSortSql(String sortDirectionWithSpacePrefix)
	{
		return "endpoint->>'address'" + sortDirectionWithSpacePrefix;
	}
}