python - 在 Google App Engine 上处理 db.Timeout

Question

我正在测试我的应用程序(在 Google App Engine 实时服务器上)和我编写它的方式我的代码中有大约 40 个 db.GqlQuery() 语句(大部分是类的一部分)。

虽然我一直收到 db.Timeout 非常频繁。

我该如何处理？我打算用这样非常残酷的代码包围我的所有查询:

  querySucceeded = False
  while not querySucceeded :
    try :
      result = db.GqlQuery( """xxx""" ).get()
      querySucceeded = True #only get here if above line doesn't raise exc
    except :
      querySucceeded = False

这样可以吗？你同意？处理 db.Timeouts 的更好方法是什么？

编辑:

我现在将它用于任何 get 查询

""" Query gets single result """
def queryGet( gql ) :
  querySucceeded = False
  while not querySucceeded :
    try :
      result = db.GqlQuery( gql ).get()
      querySucceeded = True #only get here if above line doesn't raise
    except :
      querySucceeded = False
  
  return result

我有类似的获取和计数功能。

Answer 1

这是一个重试 db.Timeout 的装饰器，改编自 Kay 框架:

import logging, time
from google.appengine.ext import db

def retry_on_timeout(retries=3, interval=1.0, exponent=2.0):
    """A decorator to retry a given function performing db operations."""
    def _decorator(func):
        def _wrapper(*args, **kwargs):
            count = 0
            while True:
                try:
                    return func(*args, **kwargs)
                except db.Timeout, e:
                    logging.debug(e)
                    if count >= retries:
                        raise e
                    else:
                        sleep_time = (exponent ** count) * interval
                        logging.warning("Retrying function %r in %d secs" %
                            (func, sleep_time))
                        time.sleep(sleep_time)
                        count += 1

        return _wrapper

    return _decorator

要使用它，只需装饰任何执行数据库操作的函数，并且你想重试:

@retry_on_timeout()
def do_the_stuff(models):
    return db.put(models)