From: Alexey Shchepin <alexey@process-one.net>
Date: Thu, 7 Mar 2019 19:14:13 +0000 (+0300)
Subject: Fix transaction aborting and restarting in ejabberd_sql
X-Git-Tag: 19.05~135
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=e921b43754ec7b30ad2d23b316651b1b36b05178;p=ejabberd

Fix transaction aborting and restarting in ejabberd_sql
---

diff --git a/src/ejabberd_sql.erl b/src/ejabberd_sql.erl
index c891e7aa8..a02aece4d 100644
--- a/src/ejabberd_sql.erl
+++ b/src/ejabberd_sql.erl
@@ -524,6 +524,7 @@ outer_transaction(F, NRestarts, _Reason) ->
     catch
 	?EX_RULE(throw, {aborted, Reason}, _) when NRestarts > 0 ->
 	    sql_query_internal([<<"rollback;">>]),
+            put(?NESTING_KEY, ?TOP_LEVEL_TXN),
 	    outer_transaction(F, NRestarts - 1, Reason);
 	?EX_RULE(throw, {aborted, Reason}, Stack) when NRestarts =:= 0 ->
 	    ?ERROR_MSG("SQL transaction restarts exceeded~n** "
@@ -610,6 +611,7 @@ sql_query_internal(#sql_query{} = Query) ->
     check_error(Res, Query);
 sql_query_internal(F) when is_function(F) ->
     case catch execute_fun(F) of
+        {aborted, Reason} -> {error, Reason};
         {'EXIT', Reason} -> {error, Reason};
         Res -> Res
     end;