Update spring-data-testapp for transactions. (#1577)

Closed #1576.

Author: mikereiche

spring-data-testapp/src/main/java/com/example/demo/AirlineGates.java

@@ -0,0 +1,72 @@
+/*
+ * Copyright 2012-2022 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.
+ * You may obtain a copy of the License at
+ *
+ *        https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.example.demo;
+
+import org.springframework.data.annotation.Id;
+import org.springframework.data.annotation.PersistenceConstructor;
+import org.springframework.data.annotation.Version;
+import org.springframework.data.couchbase.core.index.QueryIndexed;
+import org.springframework.data.couchbase.core.mapping.Document;
+
+@Document
+/**
+ * @author Michael Reiche
+ */
+public class AirlineGates {
+	@Id String id;
+	@Version Long version;
+
+	@QueryIndexed String name;
+	String iata;
+	Long gates;
+
+	@PersistenceConstructor
+	public AirlineGates(String id, String name, String iata, Long gates) {
+		this.id = id;
+		this.name = name;
+		this.iata = iata;
+		this.gates = gates;
+	}
+
+	public String getId() {
+		return id;
+	}
+
+	public String getName() {
+		return name;
+	}
+
+	public String getIata() {
+		return iata;
+	}
+
+	public Long getGates() {
+		return gates;
+	}
+
+	public String toString(){
+		StringBuffer sb=new StringBuffer();
+		sb.append("{");
+		sb.append("\"id\":"+id);
+		sb.append(", \"name\":"+name);
+		sb.append(", \"iata\":"+iata);
+		sb.append(", \"gates\":"+gates);
+		sb.append("}");
+
+		return sb.toString();
+	}
+
+}

spring-data-testapp/src/main/java/com/example/demo/AirlineGatesRepository.java

@@ -0,0 +1,30 @@
+/*
+ * Copyright 2017-2022 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.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.example.demo;
+
+import org.springframework.data.couchbase.repository.CouchbaseRepository;
+import org.springframework.data.couchbase.repository.DynamicProxyable;
+import org.springframework.stereotype.Repository;
+
+/**
+ * @author Michael Reiche
+ */
+@Repository
+public interface AirlineGatesRepository
+		extends CouchbaseRepository<AirlineGates, String>, DynamicProxyable<AirlineGatesRepository> {
+
+}

spring-data-testapp/src/main/java/com/example/demo/AirlineGatesService.java

@@ -0,0 +1,91 @@
+/*
+ * Copyright 2022 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.
+ * You may obtain a copy of the License at
+ *
+ *        https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.example.demo;
+
+import org.springframework.data.couchbase.core.CouchbaseTemplate;
+import org.springframework.data.couchbase.core.ReactiveCouchbaseOperations;
+import org.springframework.data.couchbase.core.ReactiveCouchbaseTemplate;
+import org.springframework.data.couchbase.core.TransactionalSupport;
+import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+import reactor.core.publisher.Mono;
+
+/**
+ * @author Michael Reiche
+ */
+@Service
+public class AirlineGatesService {
+	CouchbaseTemplate template;
+	ReactiveCouchbaseTemplate reactiveTemplate;
+	public AirlineGatesService(CouchbaseTemplate template) {
+		this.template = template;
+		this.reactiveTemplate = template.reactive();
+	}
+
+
+	// The @Transactional annotation results in the method of the proxy for the service executing this in a transaction
+	@Transactional
+	public void transferGates(String fromId, String toId, int gatesToTransfer, RuntimeException exceptionToThrow) {
+		// May wish to include this check to confirm this is actually in a transaction.
+		TransactionalSupport.checkForTransactionInThreadLocalStorage().map((h) -> {
+			if (!h.isPresent())
+				throw new RuntimeException("not in transaction!");
+			return h;
+		});
+
+		AirlineGates fromAirlineGates = template.findById(AirlineGates.class).one(fromId);
+		AirlineGates toAirlineGates = template.findById(AirlineGates.class).one(toId);
+		toAirlineGates.gates += gatesToTransfer;
+		fromAirlineGates.gates -= gatesToTransfer;
+		template.save(fromAirlineGates);
+		if(exceptionToThrow != null){
+			throw exceptionToThrow;
+		}
+		template.save(toAirlineGates);
+	}
+	// The @Transactional annotation results in the method of the proxy for the service executing this in a transaction
+	@Transactional
+	public Mono<Void> transferGatesReactive(String fromId, String toId, int gatesToTransfer, RuntimeException exceptionToThrow) {
+		return Mono.deferContextual(ctx -> {
+		// May wish to include this check to confirm this is actually in a transaction.
+		TransactionalSupport.checkForTransactionInThreadLocalStorage().map((h) -> {
+			if (!h.isPresent())
+				throw new RuntimeException("not in transaction!");
+			return h;
+		});
+
+		AirlineGates fromAirlineGates = template.findById(AirlineGates.class).one(fromId);
+		AirlineGates toAirlineGates = template.findById(AirlineGates.class).one(toId);
+		toAirlineGates.gates += gatesToTransfer;
+		fromAirlineGates.gates -= gatesToTransfer;
+		template.save(fromAirlineGates);
+		if(exceptionToThrow != null){
+			throw exceptionToThrow;
+		}
+		return reactiveTemplate.save(toAirlineGates).then();
+		});
+	}
+
+	// This does not have the @Transactional annotation therefore is not executed in a transaction
+	public AirlineGates save(AirlineGates airlineGates) {
+		return template.save(airlineGates);
+	}
+
+	// This does not have the @Transactional annotation therefore is not executed in a transaction
+	public AirlineGates findById(String id) {
+		return template.findById(AirlineGates.class).one(id);
+	}
+}