Esempio n. 1
0
  def _help_tasklet_along(self, gen, val=None, exc=None, tb=None):
    # XXX Docstring
    info = utils.gen_info(gen)
    __ndb_debug__ = info
    try:
      save_context = get_context()
      try:
        set_context(self._context)
        if exc is not None:
          logging.debug('Throwing %s(%s) into %s',
                        exc.__class__.__name__, exc, info)
          value = gen.throw(exc.__class__, exc, tb)
        else:
          logging.debug('Sending %r to %s', val, info)
          value = gen.send(val)
          self._context = get_context()
      finally:
        set_context(save_context)

    except StopIteration, err:
      result = get_return_value(err)
      logging.debug('%s returned %r', info, result)
      self.set_result(result)
      return
Esempio n. 2
0
 if isinstance(value, datastore_rpc.MultiRpc):
   # TODO: Tail recursion if the RPC is already complete.
   if len(value.rpcs) == 1:
     value = value.rpcs[0]
     # Fall through to next isinstance test.
   else:
     assert False  # TODO: Support MultiRpc using MultiFuture.
 if isinstance(value, UserRPC):
   # TODO: Tail recursion if the RPC is already complete.
   eventloop.queue_rpc(value, self._on_rpc_completion, value, gen)
   return
 if isinstance(value, Future):
   # TODO: Tail recursion if the Future is already done.
   assert not self._next, self._next
   self._next = value
   self._geninfo = utils.gen_info(gen)
   logging.debug('%s is now blocked waiting for %s', self, value)
   value.add_callback(self._on_future_completion, value, gen)
   return
 if isinstance(value, (tuple, list)):
   # Arrange for yield to return a list of results (not Futures).
   info = 'multi-yield from ' + utils.gen_info(gen)
   mfut = MultiFuture(info)
   for subfuture in value:
     mfut.add_dependent(subfuture)
   mfut.complete()
   mfut.add_callback(self._on_future_completion, mfut, gen)
   return
 if is_generator(value):
   assert False  # TODO: emulate PEP 380 here?
 assert False  # A tasklet shouldn't yield plain values.