changeset ed977ed34b94 in modules/account_payment_stripe:4.4
details: 
https://hg.tryton.org/modules/account_payment_stripe?cmd=changeset;node=ed977ed34b94
description:
        Lock records when processing

        issue8012
        review60511002
        (grafted from d404a49a74d2d906ce2b2d3bf14978b5d5c51427)
diffstat:

 payment.py |  20 ++++++++++++++++++++
 1 files changed, 20 insertions(+), 0 deletions(-)

diffs (37 lines):

diff -r 4e1ee61cf359 -r ed977ed34b94 payment.py
--- a/payment.py        Fri Jan 05 00:06:49 2018 +0100
+++ b/payment.py        Wed Feb 06 22:51:18 2019 +0100
@@ -192,6 +192,7 @@
                     ('stripe_charge_id', '=', None),
                     ])
         for payment in payments:
+            cls.__lock([payment])
             try:
                 charge = stripe.Charge.create(**payment._charge_parameters())
             except (stripe.error.RateLimitError,
@@ -235,6 +236,25 @@
             'idempotency_key': self.stripe_checkout_id,
             }
 
+    @classmethod
+    def __lock(cls, records):
+        from trytond.tools import grouped_slice, reduce_ids
+        from sql import Literal, For
+        transaction = Transaction()
+        database = transaction.database
+        connection = transaction.connection
+        table = cls.__table__()
+
+        if database.has_select_for():
+            for sub_records in grouped_slice(records):
+                where = reduce_ids(table.id, sub_records)
+                query = table.select(
+                    Literal(1), where=where, for_=For('UPDATE', nowait=True))
+                with connection.cursor() as cursor:
+                    cursor.execute(*query)
+        else:
+            database.lock(connection, cls._table)
+
 
 class Account(ModelSQL, ModelView):
     "Stripe Account"

Reply via email to