org.springframework.data.cassandra.repository.query.StringBasedCassandraQuery Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of spring-data-cassandra Show documentation

Cassandra support for Spring Data

There is a newer version: 4.3.4

Show newest version

package org.springframework.data.cassandra.repository.query;

import java.util.Date;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.cassandra.core.cql.CqlStringUtils;
import org.springframework.data.cassandra.core.CassandraOperations;

public class StringBasedCassandraQuery extends AbstractCassandraQuery {

	private static final Pattern PLACEHOLDER = Pattern.compile("\\?(\\d+)");
	private static final Logger LOG = LoggerFactory.getLogger(StringBasedCassandraQuery.class);

	protected String query;

	public StringBasedCassandraQuery(String query, CassandraQueryMethod queryMethod, CassandraOperations operations) {

		super(queryMethod, operations);

		this.query = query;
	}

	public StringBasedCassandraQuery(CassandraQueryMethod queryMethod, CassandraOperations operations) {
		this(queryMethod.getAnnotatedQuery(), queryMethod, operations);
	}

	@Override
	public String createQuery(CassandraParameterAccessor accessor) {
		return replacePlaceholders(query, accessor);
	}

	private String replacePlaceholders(String input, CassandraParameterAccessor accessor) {

		Matcher matcher = PLACEHOLDER.matcher(input);
		String result = input;

		while (matcher.find()) {
			String group = matcher.group();
			int index = Integer.parseInt(matcher.group(1));
			Object value = getParameterWithIndex(accessor, index);
			String stringValue = null;
			CassandraQueryMethod queryMethod = getQueryMethod();

			if (queryMethod.isStringLikeParameter(index)) {
				stringValue = "'" + CqlStringUtils.escapeSingle(value) + "'";
			} else if (queryMethod.isDateParameter(index)) {
				stringValue = "" + ((Date) value).getTime();
			} else {
				stringValue = value.toString();
			}

			result = result.replace(group, stringValue);
		}

		return result;
	}

	private Object getParameterWithIndex(CassandraParameterAccessor accessor, int index) {
		return accessor.getBindableValue(index);
	}
}