Cleanup resources when commit is aborted (#251)

Follow-up to #250.
GoogleCloudPlatform · Feb 2, 2021 · c4d5fe6 · c4d5fe6
1 parent 110b5c5
commit c4d5fe6
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/...er/src/main/java/com/google/cloud/spring/data/spanner/core/SpannerTransactionManager.java b/...er/src/main/java/com/google/cloud/spring/data/spanner/core/SpannerTransactionManager.java
@@ -231,6 +231,12 @@ protected void doCommit(DefaultTransactionStatus defaultTransactionStatus)
 			}
 		}
 		catch (AbortedException ex) {
+			// The client library will not close transaction resources if state == ABORTED
+			// to allow for retries, but we do not retry aborted transactions.
+			// See: SessionPool.close()
+			if (tx.getTransactionManager() != null) {
+				tx.getTransactionManager().close();
+			}
 			throw new UnexpectedRollbackException("Transaction Got Rolled Back", ex);
 		}
 		catch (SpannerException ex) {