Fixing async task queue sample

Jon Wayne Parrott · Jon Wayne Parrott · commit f81552c6f1ae · 2016-04-15T11:30:31.000-07:00
Change-Id: I7f4deb12a657f4bc9dbfe2c244c0ebd50e490ce4
diff --git a/appengine/taskqueue/counter/application.py b/appengine/taskqueue/counter/application.py
@@ -52,16 +52,24 @@ def post(self):
             'Task {} enqueued, ETA {}.'.format(task.name, task.eta))
 
 
+# AsyncEnqueueTaskHandler behaves the same as EnqueueTaskHandler, but shows
+# how to queue the task using the asyncronous API. This is not wired up by
+# default. To use this, change the MainPageHandler's form action to
+# /enqueue_async
 class AsyncEnqueueTaskHandler(webapp2.RequestHandler):
     def post(self):
         amount = int(self.request.get('amount'))
 
-        future = taskqueue.add_async(
+        queue = taskqueue.Queue(name='default')
+        task = taskqueue.Task(
             url='/update_counter',
             target='worker',
             params={'amount': amount})
 
-        task = future.wait()
+        rpc = queue.add_async(task)
+
+        # Wait for the rpc to complete and return the queued task.
+        task = rpc.get_result()
 
         self.response.write(
             'Task {} enqueued, ETA {}.'.format(task.name, task.eta))
@@ -70,5 +78,5 @@ def post(self):
 app = webapp2.WSGIApplication([
     ('/', MainPageHandler),
     ('/enqueue', EnqueueTaskHandler),
-    ('/enqueue_async', EnqueueTaskHandler)
+    ('/enqueue_async', AsyncEnqueueTaskHandler)
 ], debug=True)
