Google App Engine:如何使用TaskQueue或Async Urlfetch并行化下载?

时间:2010-08-21 22:17:40

标签: python google-app-engine urlfetch

我的Gae应用程序从第三方站点检索JSON数据;给出一个代表要下载的项目的ID,该站点上的项目数据被组织在多个页面中,因此我的代码必须逐页下载数据块,直到检索到最后一个可用页面的数据为止。 我的简化代码如下所示:

class FetchData(webapp.RequestHandler):
  def get(self):
    ...
    data_list = []
    page = 1
    while True:
      fetched_data= urlfetch.fetch('http://www.foo.com/getdata?id=xxx&result=JSON&page=%s' % page)
      data_chunk = fetched_data["data"] 
      data_list = data_list + data_chunk
      if len(data_list) == int(fetched_data["total_pages"]):
         break
      else:
         page = page +1 
    ...  
    doRender('dataview.htm',{'data_list':data_list} )

data_list结果是一个有序列表,其中第一项具有页码1的数据,最后一项具有最新页面的数据;检索后,此data_list将在视图中呈现。

此方法有效率为99%,但有时,由于Google App Engine强加的 30秒限制,对于包含许多页面的项目,我会感到可怕DeadlineExceededError。 我想知道是否使用TaskQueue | Deferred | AsyncUrlfetch我可以改进此算法以某种方式并行化N urlfetch调用。

2 个答案:

答案 0 :(得分:1)

使用此:http://code.google.com/appengine/docs/python/urlfetch/asynchronousrequests.html

这很简单:

def handle_result(rpc):
    result = rpc.get_result()
    # ... Do something with result...

# Use a helper function to define the scope of the callback.
def create_callback(rpc):
    return lambda: handle_result(rpc)

rpcs = []
for url in urls:
    rpc = urlfetch.create_rpc()
    rpc.callback = create_callback(rpc)
    urlfetch.make_fetch_call(rpc, url)
    rpcs.append(rpc)

# ...

# Finish all RPCs, and let callbacks process the results.
for rpc in rpcs:
    rpc.wait()

答案 1 :(得分:0)

我已经解决了这个问题:

chunks_dict = {}

def handle_result(rpc, page):
    result = rpc.get_result()
    chunks_dict[page] = result["data"]

def create_callback(rpc, page):
    return lambda: handle_result(rpc, page)

rpcs = []
while True:
    rpc = urlfetch.create_rpc(deadline = 10)
    rpc.callback = create_callback(rpc, page)
    urlfetch.make_fetch_call(rpc, 'http://www.foo.com/getdata?id=xxx&result=JSON&page=%s' % page)
    rpcs.append(rpc)
    if page > total_pages:
       break
    else:
       page = page +1   
for rpc in rpcs:
    rpc.wait()

page_keys = chunks_dict.keys()
page_keys.sort()
for key in page_keys:
    data_list= data_list + chunks_dict[key]