DATACOUCH-623 - Add replace() method to CouchbaseRepository for CAS usage.

Author: mikereiche

src/main/java/org/springframework/data/couchbase/repository/CouchbaseRepository.java

@@ -43,4 +43,8 @@ public interface CouchbaseRepository<T, ID> extends PagingAndSortingRepository<T
 	@Override
 	List<T> findAllById(Iterable<ID> iterable);
 
+	<S extends T> S replace(S var1);
+
+	<S extends T> Iterable<S> replaceAll(Iterable<S> var1);
+
 }

src/main/java/org/springframework/data/couchbase/repository/support/SimpleCouchbaseRepository.java

@@ -86,6 +86,21 @@ public <S extends T> Iterable<S> saveAll(final Iterable<S> entities) {
 				.all(Streamable.of(entities).toList());
 	}
 
+	@Override
+	@SuppressWarnings("unchecked")
+	public <S extends T> S replace(final S entity) {
+		Assert.notNull(entity, "Entity must not be null!");
+		return (S) couchbaseOperations.replaceById(entityInformation.getJavaType()).one(entity);
+	}
+
+	@Override
+	@SuppressWarnings("unchecked")
+	public <S extends T> Iterable<S> replaceAll(final Iterable<S> entities) {
+		Assert.notNull(entities, "The given Iterable of entities must not be null!");
+		return (Iterable<S>) couchbaseOperations.replaceById(entityInformation.getJavaType())
+				.all(Streamable.of(entities).toList());
+	}
+
 	@Override
 	public Optional<T> findById(final ID id) {
 		Assert.notNull(id, "The given id must not be null!");

src/test/java/org/springframework/data/couchbase/domain/UserRepository.java

@@ -18,6 +18,8 @@
 
 import java.util.List;
 
+import com.couchbase.client.java.json.JsonArray;
+import org.springframework.data.couchbase.repository.CouchbaseRepository;
 import org.springframework.data.couchbase.repository.Query;
 import org.springframework.data.repository.PagingAndSortingRepository;
 import org.springframework.data.repository.query.Param;
@@ -30,10 +32,14 @@
  * @author Michael Reiche
  */
 @Repository
-public interface UserRepository extends PagingAndSortingRepository<User, String> {
+public interface UserRepository extends CouchbaseRepository<User, String> {
 
 	List<User> findByFirstname(String firstname);
 
+	List<User> findByFirstnameIn(String... firstnames);
+
+	List<User> findByFirstnameIn(JsonArray firstnames);
+
 	List<User> findByFirstnameAndLastname(String firstname, String lastname);
 
 	@Query("#{#n1ql.selectEntity} where #{#n1ql.filter} and firstname = $1 and lastname = $2")

src/test/java/org/springframework/data/couchbase/repository/CouchbaseRepositoryQueryIntegrationTests.java

@@ -15,13 +15,15 @@
  */
 
 package org.springframework.data.couchbase.repository;
-
+import java.util.HashSet;
+import java.util.UUID;
 import static org.junit.jupiter.api.Assertions.*;
 
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.List;
 import java.util.Locale;
+import java.util.Set;
 import java.util.concurrent.Callable;
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.Executors;
@@ -31,10 +33,13 @@
 import org.junit.jupiter.api.Test;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.context.annotation.Configuration;
+import org.springframework.dao.DataIntegrityViolationException;
 import org.springframework.data.couchbase.CouchbaseClientFactory;
 import org.springframework.data.couchbase.config.AbstractCouchbaseConfiguration;
 import org.springframework.data.couchbase.domain.Airport;
 import org.springframework.data.couchbase.domain.AirportRepository;
+import org.springframework.data.couchbase.domain.User;
+import org.springframework.data.couchbase.domain.UserRepository;
 import org.springframework.data.couchbase.repository.config.EnableCouchbaseRepositories;
 import org.springframework.data.couchbase.util.Capabilities;
 import org.springframework.data.couchbase.util.ClusterAwareIntegrationTests;
@@ -58,6 +63,9 @@ public class CouchbaseRepositoryQueryIntegrationTests extends ClusterAwareIntegr
 
 	@Autowired AirportRepository airportRepository;
 
+	@Autowired UserRepository userRepository;
+
+
 	@BeforeEach
 	void beforeEach() {
 		try {
@@ -118,6 +126,17 @@ void findBySimpleProperty() {
 
 	}
 
+	@Test
+	public void testCas() {
+
+		User user = new User("1","Dave","Wilson");
+		userRepository.save(user);
+		user.setVersion(user.getVersion() - 1);
+		assertThrows(DataIntegrityViolationException.class,	() -> userRepository.replace(user));
+		userRepository.delete(user);
+
+	}
+
 	@Test
 	void count() {
 		String[] iatas = { "JFK", "IAD", "SFO", "SJC", "SEA", "LAX", "PHX" };