- protected TransactionCommitFailedException mapException(final Exception e) {
- Preconditions.checkArgument(e instanceof TransactionCommitFailedException);
- return (TransactionCommitFailedException) e;
+ protected TransactionCommitFailedException mapException(final Exception exception) {
+ final Throwable cause = exception.getCause();
+ if (cause instanceof TransactionCommitFailedException) {
+ return (TransactionCommitFailedException) cause;
+ } else {
+ return new TransactionCommitFailedException(exception.getMessage(), cause);
+ }