DATACOUCH-1066 - N1qlExpression as QueryCriteria key.

This is a merge of mmonti's changeset into master.  The changeset was made
on top of 4.1.x instead of master so it has a little catching up to do.

Merge branch 'gh-1066-n1qlexpression-as-querycriteria-key' of github.com:mmonti/spring-data-couchbase into datacouch_1066_1072_mmonti_n1ql_query_creator_expression

Original pull request #1076

Author: mikereiche

src/main/java/org/springframework/data/couchbase/core/ReactiveReplaceByIdOperationSupport.java

@@ -22,6 +22,7 @@
 import java.util.Collection;
 
 import org.springframework.data.couchbase.core.mapping.CouchbaseDocument;
+import org.springframework.data.couchbase.core.mapping.Document;
 import org.springframework.util.Assert;
 
 import com.couchbase.client.core.msg.kv.DurabilityLevel;

src/main/java/org/springframework/data/couchbase/core/query/QueryCriteria.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2012-2020 the original author or authors
+ * Copyright 2012-2021 the original author or authors
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -27,13 +27,16 @@
 import org.springframework.data.couchbase.core.convert.CouchbaseConverter;
 import org.springframework.lang.Nullable;
 
+import static org.springframework.data.couchbase.core.query.N1QLExpression.*;
+
 /**
  * @author Michael Nitschinger
  * @author Michael Reiche
+ * @author Mauro Monti
  */
 public class QueryCriteria implements QueryCriteriaDefinition {
 
-	private final String key;
+	private final N1QLExpression key;
 	/**
 	 * Holds the chain itself, the current operator being always the last one.
 	 */
@@ -46,15 +49,15 @@ public class QueryCriteria implements QueryCriteriaDefinition {
 	private Object[] value;
 	private String format;
 
-	QueryCriteria(List<QueryCriteria> chain, String key, Object[] value, ChainOperator chainOperator) {
+	QueryCriteria(List<QueryCriteria> chain, N1QLExpression key, Object[] value, ChainOperator chainOperator) {
 		this(chain, key, value, chainOperator, null, null);
 	}
 
-	QueryCriteria(List<QueryCriteria> chain, String key, Object value, ChainOperator chainOperator) {
+	QueryCriteria(List<QueryCriteria> chain, N1QLExpression key, Object value, ChainOperator chainOperator) {
 		this(chain, key, new Object[] { value }, chainOperator, null, null);
 	}
 
-	QueryCriteria(List<QueryCriteria> chain, String key, Object[] value, ChainOperator chainOperator, String operator,
+	QueryCriteria(List<QueryCriteria> chain, N1QLExpression key, Object[] value, ChainOperator chainOperator, String operator,
 			String format) {
 		this.criteriaChain = chain;
 		criteriaChain.add(this);
@@ -70,9 +73,16 @@ Object[] getValue() {
 	}
 
 	/**
-	 * Static factory method to create a Criteria using the provided key.
+	 * Static factory method to create a Criteria using the provided String key.
 	 */
 	public static QueryCriteria where(String key) {
+		return where(x(key));
+	}
+
+	/**
+	 * Static factory method to create a Criteria using the provided N1QLExpression key.
+	 */
+	public static QueryCriteria where(N1QLExpression key) {
 		return new QueryCriteria(new ArrayList<>(), key, null, null);
 	}
 
@@ -83,21 +93,29 @@ private static QueryCriteria wrap(QueryCriteria criteria) {
 	}
 
 	public QueryCriteria and(String key) {
+		return and(x(key));
+	}
+
+	public QueryCriteria and(N1QLExpression key) {
 		return new QueryCriteria(this.criteriaChain, key, null, ChainOperator.AND);
 	}
 
 	public QueryCriteria and(QueryCriteria criteria) {
 		return new QueryCriteria(this.criteriaChain, null, criteria, ChainOperator.AND);
 	}
 
-	public QueryCriteria or(QueryCriteria criteria) {
-		return new QueryCriteria(this.criteriaChain, null, criteria, ChainOperator.OR);
+	public QueryCriteria or(String key) {
+		return or(x(key));
 	}
 
-	public QueryCriteria or(String key) {
+	public QueryCriteria or(N1QLExpression key) {
 		return new QueryCriteria(this.criteriaChain, key, null, ChainOperator.OR);
 	}
 
+	public QueryCriteria or(QueryCriteria criteria) {
+		return new QueryCriteria(this.criteriaChain, null, criteria, ChainOperator.OR);
+	}
+
 	public QueryCriteria eq(@Nullable Object o) {
 		return is(o);
 	}
@@ -343,7 +361,7 @@ public String export() { // used only by tests
 	 */
 	private StringBuilder exportSingle(StringBuilder sb, int[] paramIndexPtr, JsonValue parameters,
 			CouchbaseConverter converter) {
-		String fieldName = maybeBackTic(key);
+		String fieldName = key == null ? null : key.toString(); // maybeBackTic(key);
 		int valueLen = value == null ? 0 : value.length;
 		Object[] v = new Object[valueLen + 2];
 		v[0] = fieldName;
@@ -377,7 +395,7 @@ private StringBuilder exportSingle(StringBuilder sb, int[] paramIndexPtr, JsonVa
 	 * @param parameters - parameters of the query. If operands are parameterized, their values are added to parameters
 	 * @return string containing part of N1QL query
 	 */
-	private String maybeWrapValue(String key, Object value, int[] paramIndexPtr, JsonValue parameters,
+	private String maybeWrapValue(N1QLExpression key, Object value, int[] paramIndexPtr, JsonValue parameters,
 			CouchbaseConverter converter) {
 		if (paramIndexPtr != null) {
 			if (paramIndexPtr[0] >= 0) {
@@ -397,10 +415,10 @@ private String maybeWrapValue(String key, Object value, int[] paramIndexPtr, Jso
 				JsonObject params = (JsonObject) parameters;
 				// from StringBasedN1qlQueryParser.getNamedPlaceholderValues()
 				try {
-					params.put(key, convert(converter, value));
+					params.put(key.toString(), convert(converter, value));
 				} catch (InvalidArgumentException iae) {
 					if (value instanceof Object[]) {
-						params.put(key, JsonArray.from((Object[]) value));
+						params.put(key.toString(), JsonArray.from((Object[]) value));
 					} else {
 						throw iae;
 					}
@@ -416,7 +434,7 @@ private String maybeWrapValue(String key, Object value, int[] paramIndexPtr, Jso
 		} else if (value == null) {
 			return "null";
 		} else if (value instanceof Object[]) { // convert array into sequence of comma-separated values
-			StringBuffer l = new StringBuffer();
+			StringBuilder l = new StringBuilder();
 			l.append("[");
 			Object[] array = (Object[]) value;
 			for (int i = 0; i < array.length; i++) {

src/main/java/org/springframework/data/couchbase/repository/query/CouchbaseRepositoryQuery.java

@@ -118,6 +118,11 @@ public Pageable first() {
 			return delegate.first();
 		}
 
+		@Override
+		public Pageable withPage(int i) {
+			return new CountPageable(delegate.withPage(i));
+		}
+
 		public boolean hasPrevious() {
 			return delegate.hasPrevious();
 		}

src/main/java/org/springframework/data/couchbase/repository/query/N1qlCountQueryCreator.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2012-2020 the original author or authors
+ * Copyright 2012-2021 the original author or authors
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -16,12 +16,15 @@
 package org.springframework.data.couchbase.repository.query;
 
 import static org.springframework.data.couchbase.core.query.QueryCriteria.where;
+import static org.springframework.data.couchbase.core.query.N1QLExpression.*;
+import static org.springframework.data.couchbase.core.query.QueryCriteria.*;
 
 import java.util.Iterator;
 
 import org.springframework.core.convert.converter.Converter;
 import org.springframework.data.couchbase.core.convert.CouchbaseConverter;
 import org.springframework.data.couchbase.core.mapping.CouchbasePersistentProperty;
+import org.springframework.data.couchbase.core.query.N1QLExpression;
 import org.springframework.data.couchbase.core.query.Query;
 import org.springframework.data.couchbase.core.query.QueryCriteria;
 import org.springframework.data.domain.Sort;
@@ -36,32 +39,38 @@
 /**
  * @author Michael Nitschinger
  * @author Michael Reiche
+ * @author Mauro Monti
  */
 public class N1qlQueryCreator extends AbstractQueryCreator<Query, QueryCriteria> {
 
+	private static final String META_ID_PROPERTY = "id";
+	private static final String META_CAS_PROPERTY = "cas";
+
 	private final ParameterAccessor accessor;
 	private final MappingContext<?, CouchbasePersistentProperty> context;
 	private final QueryMethod queryMethod;
 	private final CouchbaseConverter converter;
+	private final String bucketName;
 
-	public N1qlQueryCreator(final PartTree tree, final ParameterAccessor accessor, QueryMethod queryMethod,
-			CouchbaseConverter converter) {
+	public N1qlQueryCreator(final PartTree tree, final ParameterAccessor accessor, final QueryMethod queryMethod,
+			final CouchbaseConverter converter, final String bucketName) {
 		super(tree, accessor);
 		this.accessor = accessor;
 		this.context = converter.getMappingContext();
 		this.queryMethod = queryMethod;
 		this.converter = converter;
+		this.bucketName = bucketName;
 	}
 
 	@Override
 	protected QueryCriteria create(final Part part, final Iterator<Object> iterator) {
 		PersistentPropertyPath<CouchbasePersistentProperty> path = context.getPersistentPropertyPath(part.getProperty());
 		CouchbasePersistentProperty property = path.getLeafProperty();
-		return from(part, property, where(path.toDotPath(cvtr)), iterator);
+		return from(part, property, where(addMetaIfRequired(path, property)), iterator);
 	}
 
 	static Converter<? super CouchbasePersistentProperty, String> cvtr = (
-			source) -> new StringBuilder(source.getName().length() + 2).append('`').append(source.getName()).append('`')
+			source) -> new StringBuilder(source.getFieldName().length() + 2).append('`').append(source.getFieldName()).append('`')
 					.toString();
 
 	@Override
@@ -73,7 +82,7 @@ protected QueryCriteria and(final Part part, final QueryCriteria base, final Ite
 		PersistentPropertyPath<CouchbasePersistentProperty> path = context.getPersistentPropertyPath(part.getProperty());
 		CouchbasePersistentProperty property = path.getLeafProperty();
 
-		return from(part, property, base.and(path.toDotPath()), iterator);
+		return from(part, property, base.and(addMetaIfRequired(path, property)), iterator);
 	}
 
 	@Override
@@ -149,4 +158,15 @@ private QueryCriteria from(final Part part, final CouchbasePersistentProperty pr
 		}
 	}
 
+	private N1QLExpression addMetaIfRequired(final PersistentPropertyPath<CouchbasePersistentProperty> persistentPropertyPath,
+											 final CouchbasePersistentProperty property) {
+		if (property.isIdProperty()) {
+			return path(meta(i(bucketName)), i(META_ID_PROPERTY));
+		}
+		if (property.isVersionProperty()) {
+			return path(meta(i(bucketName)), i(META_CAS_PROPERTY));
+		}
+		return x(persistentPropertyPath.toDotPath(cvtr));
+	}
+
 }