Take expiry from entity class annotation for getAndTouch. (#1640)

Closes #1634.

Author: mikereiche

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

@@ -63,6 +63,8 @@ static class ReactiveFindByIdSupport<T> implements ReactiveFindById<T> {
 		private final ReactiveTemplateSupport support;
 		private final Duration expiry;
 
+		private Duration expiryToUse;
+
 		ReactiveFindByIdSupport(ReactiveCouchbaseTemplate template, Class<T> domainType, String scope, String collection,
 				CommonOptions<?> options, List<String> fields, Duration expiry, ReactiveTemplateSupport support) {
 			this.template = template;
@@ -86,7 +88,7 @@ public Mono<T> one(final String id) {
 				ReactiveCollection reactive = template.getCouchbaseClientFactory().withScope(pArgs.getScope())
 						.getCollection(pArgs.getCollection()).reactive();
 				if (pArgs.getOptions() instanceof GetAndTouchOptions) {
-					return reactive.getAndTouch(docId, expiryToUse(), (GetAndTouchOptions) pArgs.getOptions());
+					return reactive.getAndTouch(docId, expiryToUse, (GetAndTouchOptions) pArgs.getOptions());
 				} else {
 					return reactive.get(docId, (GetOptions) pArgs.getOptions());
 				}
@@ -142,7 +144,18 @@ public FindByIdWithProjection<T> withExpiry(final Duration expiry) {
 
 		private CommonOptions<?> initGetOptions() {
 			CommonOptions<?> getOptions;
-			if (expiry != null || options instanceof GetAndTouchOptions) {
+			final CouchbasePersistentEntity<?> entity = template.getConverter().getMappingContext()
+					.getRequiredPersistentEntity(domainType);
+			Duration entityExpiryAnnotation = entity.getExpiryDuration();
+			if (expiry != null || entityExpiryAnnotation == null || !entityExpiryAnnotation.isZero()
+					|| options instanceof GetAndTouchOptions) {
+				if (expiry != null) {
+					expiryToUse = expiry;
+				} else if (entityExpiryAnnotation == null || !entityExpiryAnnotation.isZero()) {
+					expiryToUse = entityExpiryAnnotation;
+				} else {
+					expiryToUse = Duration.ZERO;
+				}
 				GetAndTouchOptions gOptions = options != null ? (GetAndTouchOptions) options : getAndTouchOptions();
 				if (gOptions.build().transcoder() == null) {
 					gOptions.transcoder(RawJsonTranscoder.INSTANCE);
@@ -160,18 +173,7 @@ private CommonOptions<?> initGetOptions() {
 			}
 			return getOptions;
 		}
-
-		private Duration expiryToUse() {
-			Duration expiryToUse = expiry;
-			if (expiryToUse != null || options instanceof GetAndTouchOptions) {
-				if (expiryToUse == null) { // GetAndTouchOptions without specifying expiry -> get expiry from annoation
-					final CouchbasePersistentEntity<?> entity = template.getConverter().getMappingContext()
-							.getRequiredPersistentEntity(domainType);
-					expiryToUse = entity.getExpiryDuration();
-				}
-			}
-			return expiryToUse;
-		}
+		
 	}
 
 }

src/test/java/org/springframework/data/couchbase/core/CouchbaseTemplateKeyValueIntegrationTests.java

@@ -112,6 +112,43 @@ void findByIdWithExpiry() {
 
 	}
 
+	@Test
+	void findByIdWithExpiryAnnotation() {
+		try {
+			UserAnnotated user1 = new UserAnnotated(UUID.randomUUID().toString(), "user1", "user1");
+			UserAnnotated user2 = new UserAnnotated(UUID.randomUUID().toString(), "user2", "user2");
+
+			Collection<UserAnnotated> upserts = (Collection<UserAnnotated>) couchbaseTemplate.upsertById(UserAnnotated.class)
+				.all(Arrays.asList(user1, user2));
+
+			// explicitly set expiry to 10 seconds
+			UserAnnotated foundUser1 = couchbaseTemplate.findById(UserAnnotated.class).withExpiry(Duration.ofSeconds(10)).one(user1.getId());
+			user1.setVersion(foundUser1.getVersion());// version will have changed
+			assertEquals(user1, foundUser1);
+			UserAnnotated foundUser2 = couchbaseTemplate.findById(UserAnnotated.class).withExpiry(Duration.ofSeconds(10)).one(user2.getId());
+			user2.setVersion(foundUser2.getVersion());// version will have changed
+			assertEquals(user2, foundUser2);
+
+			// now set user1 expiration back to 1 second with getAndTouch using the @Document(expiry=1) annotation
+			foundUser1 = couchbaseTemplate.findById(UserAnnotated.class).one(user1.getId());
+			user1.setVersion(foundUser1.getVersion());// version will have changed
+			assertEquals(user1, foundUser1);
+
+			// user1 should be gone, user2 should still be there
+			int tries = 0;
+			Collection<User> foundUsers;
+			do {
+				sleepSecs(1);
+				foundUsers = (Collection<User>) couchbaseTemplate.findById(User.class)
+					.all(Arrays.asList(user1.getId(), user2.getId()));
+			} while (tries++ < 7 && foundUsers.size() != 1 && !user2.equals(foundUsers.iterator().next()));
+			assertEquals(1, foundUsers.size(), "should have found exactly 1 user");
+			assertEquals(user2, foundUsers.iterator().next());
+		} finally {
+			couchbaseTemplate.removeByQuery(User.class).withConsistency(QueryScanConsistency.REQUEST_PLUS).all();
+		}
+
+	}
 	@Test
 	void upsertAndFindById() {
 		User user = new User(UUID.randomUUID().toString(), "firstname", "lastname");