在GAE中,如何使汇款交易具有幂等性?

时间:2017-09-23 16:12:39

标签: google-app-engine transactions google-cloud-datastore

GAE文档警告:

  

尽可能使数据存储区事务处于幂等状态,这样,如果重复某个事务,最终结果将是相同的。

假设我想在两个人之间转移一笔款项:

class User(ndb.Model):
    balance = ndb.IntegerProperty(default=0)

@ndb.transactional(xg=True)
def transfer(from_key, to_key, amount)
    from = from_key.get()
    to = to_key.get()
    from.balance -= amount
    to.balance += amount
    ndb.put_multi([from, to])

由于这不是幂等的,因此可能不止一次发生并导致问题。我想重构这一点以确保交易是幂等的。

answer提出了一个解决方案:

  

要解决此问题,您可以通过创建“事务密钥”并将该密钥作为事务的一部分记录在新实体中来使事务具有幂等性。第二个事务可以检查该事务密钥,如果找到,将不执行任何操作。一旦您对交易完成感到满意,或者您放弃重试,就可以删除交易密钥。

但我不明白如何实施它。

有人可以解释如何使此交易具有幂等性吗?

1 个答案:

答案 0 :(得分:1)

您可以根据交易详情创建密钥,例如:

import datetime
import hashlib

>>> txn = {
    'from_account': '100123',
    'to_account': '200456',
    'amount': 123456,
    'timestamp': datetime.datetime(2017, 9, 23, 10, 11, 12, 123456)
}

# Combine the values into a string
>>> raw_key = u''.join([unicode(v) for k, v in sorted(txn.items())])

>>> print raw_key
1234561001232017-09-23 10:11:12.123456200456

# hash the key so exposing it in logs etc. doesn't expose transaction data
>>> key = hashlib.sha256().hexdigest()
>>> print key
261c516faa580d6604850967c5804f3fce5f323aae90e36debdb84aa0b950dcb

您可以将散列密钥存储在数据存储区中,或者将其作为事务模型的计算属性(如果有),并在尝试创建新事务之前对其进行查询。

class TransactionKeys(ndb.model):
        pass


class TransactionHandler(webapp2.RequestHandler):

   def post(self):
       txn = {
            'from_account': self.request.POST['from'],
            'to_account': self.request.POST['to'],
            'value': self.request.POST['value']
            'timestamp': datetime.datetime.now()
       }
       raw_key = u''.join([unicode(v) for k, v in sorted(txn.items())])
       txn_key = hashlib.sha256().hexdigest()
       ...
       transfer(from_key, to_key, amount, txn_key)


@ndb.transactional(xg=True)
def transfer(from_key, to_key, amount, txn_key)
    already_exists = TransactionKeys.get_by_id(txn_key)
    if already_exists:
        raise DuplicateTransactionError('Duplicate transaction!')
    else:
        transaction_key = TransactionKey(id=txn_key)
    from = from_key.get()
    to = to_key.get()
    from.balance -= amount
    to.balance += amount
    ndb.put_multi([from, to, txn_key])

这种方法并不完美 - 例如,如果两个相同的事务以相同的微秒到达,则会失败。您可以添加其他数据以使密钥更加独特,例如App Engine实例ID或请求ID。

最后,强制性免责声明:我不是安全专家,如果你用真钱做这件事,你应该进行适当的尽职调查并考虑专业赔偿/公共责任保险。