Exemple #1
0
 def set_result(self, result):
   assert not self._done
   self._result = result
   self._done = True
   _state.remove_pending(self)
   for callback, args, kwds  in self._callbacks:
     eventloop.queue_call(None, callback, *args, **kwds)
Exemple #2
0
 def set_result(self, result):
   assert not self._done
   self._result = result
   self._done = True
   logging.debug('_all_pending: remove successful %s', self)
   self._all_pending.remove(self)
   for callback, args, kwds  in self._callbacks:
     eventloop.queue_call(None, callback, *args, **kwds)
Exemple #3
0
 def testRun(self):
   record = []
   def foo(arg):
     record.append(arg)
   eventloop.queue_call(0.2, foo, 42)
   eventloop.queue_call(0.1, foo, arg='hello')
   eventloop.run()
   self.assertEqual(record, ['hello', 42])
Exemple #4
0
def sleep(dt):
  """Public function to sleep some time.

  Example:
    yield tasklets.sleep(0.5)  # Sleep for half a sec.
  """
  fut = Future('sleep(%.3f)' % dt)
  eventloop.queue_call(dt, fut.set_result, None)
  return fut
Exemple #5
0
 def set_exception(self, exc, tb=None):
   assert isinstance(exc, BaseException)
   assert not self._done
   self._exception = exc
   self._traceback = tb
   self._done = True
   _state.remove_pending(self, status='fail')
   for callback, args, kwds in self._callbacks:
     eventloop.queue_call(None, callback, *args, **kwds)
Exemple #6
0
 def add(self, arg):
   fut = tasklets.Future('%s.add(%s)' % (self, arg))
   if not self._todo:  # Schedule the callback
     # We use the fact that regular tasklets are queued at time None,
     # which puts them at absolute time 0 (i.e. ASAP -- still on a
     # FIFO basis).  Callbacks explicitly scheduled with a delay of 0
     # are only run after all immediately runnable tasklets have run.
     eventloop.queue_call(0, self._autobatcher_callback)
   self._todo.append((fut, arg))
   return fut
 def testRunWithRpcs(self):
   record = []
   def foo(arg):
     record.append(arg)
   eventloop.queue_call(0.1, foo, 42)
   config = datastore_rpc.Configuration(on_completion=foo)
   rpc = self.conn.async_get(config, [])
   self.assertEqual(len(rpc.rpcs), 1)
   eventloop.queue_rpc(rpc)
   eventloop.run()
   self.assertEqual(record, [rpc.rpcs[0], 42])
   self.assertEqual(rpc.state, 2)  # TODO: Use apiproxy_rpc.RPC.FINISHING.
Exemple #8
0
 def set_exception(self, exc, tb=None):
   assert isinstance(exc, BaseException)
   assert not self._done
   self._exception = exc
   self._traceback = tb
   self._done = True
   if self in self._all_pending:
     logging.debug('_all_pending: remove failing %s', self)
     self._all_pending.remove(self)
   else:
     logging.debug('_all_pending: not found %s', self)
   for callback, args, kwds in self._callbacks:
     eventloop.queue_call(None, callback, *args, **kwds)
Exemple #9
0
    def testRunWithRpcs(self):
        record = []

        def foo(arg):
            record.append(arg)

        eventloop.queue_call(0.1, foo, 42)
        config = datastore_rpc.Configuration(on_completion=foo)
        rpc = self.conn.async_get(config, [])
        self.assertEqual(len(rpc.rpcs), 1)
        eventloop.queue_rpc(rpc)
        eventloop.run()
        self.assertEqual(record, [rpc.rpcs[0], 42])
        self.assertEqual(rpc.state, 2)  # TODO: Use apiproxy_rpc.RPC.FINISHING.
Exemple #10
0
 def testRunWithRpcs(self):
   apiproxy_stub_map.apiproxy = apiproxy_stub_map.APIProxyStubMap()
   stub = datastore_file_stub.DatastoreFileStub('_', None)
   apiproxy_stub_map.apiproxy.RegisterStub('datastore_v3', stub)
   record = []
   def foo(arg):
     record.append(arg)
   eventloop.queue_call(0.1, foo, 42)
   conn = datastore_rpc.Connection()
   config = datastore_rpc.Configuration(on_completion=foo)
   rpc = conn.async_get(config, [])
   self.assertEqual(len(rpc.rpcs), 1)
   eventloop.queue_rpc(rpc)
   eventloop.run()
   self.assertEqual(record, [rpc.rpcs[0], 42])
   self.assertEqual(rpc.state, 2)  # TODO: Use apiproxy_rpc.RPC.FINISHING.
Exemple #11
0
    def testQueueTasklet(self):
        def f():
            return 1

        def g():
            return 2

        def h():
            return 3

        t_before = time.time()
        eventloop.queue_call(1, f, 42, 'hello', a=1, b=2)
        eventloop.queue_call(3, h, c=3, d=4)
        eventloop.queue_call(2, g, 100, 'abc')
        t_after = time.time()
        self.assertEqual(len(self.ev.queue), 3)
        [(t1, f1, a1, k1), (t2, f2, a2, k2), (t3, f3, a3, k3)] = self.ev.queue
        self.assertTrue(t1 < t2)
        self.assertTrue(t2 < t3)
        self.assertTrue(abs(t1 - (t_before + 1)) < t_after - t_before)
        self.assertTrue(abs(t2 - (t_before + 2)) < t_after - t_before)
        self.assertTrue(abs(t3 - (t_before + 3)) < t_after - t_before)
        self.assertEqual(f1, f)
        self.assertEqual(f2, g)
        self.assertEqual(f3, h)
        self.assertEqual(a1, (42, 'hello'))
        self.assertEqual(a2, (100, 'abc'))
        self.assertEqual(a3, ())
        self.assertEqual(k1, {'a': 1, 'b': 2})
        self.assertEqual(k2, {})
        self.assertEqual(k3, {'c': 3, 'd': 4})
Exemple #12
0
 def testQueueTasklet(self):
   def f(): return 1
   def g(): return 2
   def h(): return 3
   t_before = time.time()
   eventloop.queue_call(1, f, 42, 'hello', a=1, b=2)
   eventloop.queue_call(3, h, c=3, d=4)
   eventloop.queue_call(2, g, 100, 'abc')
   t_after = time.time()
   self.assertEqual(len(self.ev.queue), 3)
   [(t1, f1, a1, k1), (t2, f2, a2, k2), (t3, f3, a3, k3)] = self.ev.queue
   self.assertTrue(t1 < t2)
   self.assertTrue(t2 < t3)
   self.assertTrue(abs(t1 - (t_before + 1)) < t_after - t_before)
   self.assertTrue(abs(t2 - (t_before + 2)) < t_after - t_before)
   self.assertTrue(abs(t3 - (t_before + 3)) < t_after - t_before)
   self.assertEqual(f1, f)
   self.assertEqual(f2, g)
   self.assertEqual(f3, h)
   self.assertEqual(a1, (42, 'hello'))
   self.assertEqual(a2, (100, 'abc'))
   self.assertEqual(a3, ())
   self.assertEqual(k1, {'a': 1, 'b': 2})
   self.assertEqual(k2, {})
   self.assertEqual(k3, {'c': 3, 'd': 4})
Exemple #13
0
def tasklet(func):
  # XXX Docstring

  @utils.wrapping(func)
  def tasklet_wrapper(*args, **kwds):
    # XXX Docstring

    # TODO: make most of this a public function so you can take a bare
    # generator and turn it into a tasklet dynamically.  (Monocle has
    # this I believe.)
    # __ndb_debug__ = utils.func_info(func)
    fut = Future('tasklet %s' % utils.func_info(func))
    fut._context = get_context()
    try:
      result = func(*args, **kwds)
    except StopIteration, err:
      # Just in case the function is not a generator but still uses
      # the "raise Return(...)" idiom, we'll extract the return value.
      result = get_return_value(err)
    if is_generator(result):
      eventloop.queue_call(None, fut._help_tasklet_along, result)
    else:
      fut.set_result(result)
    return fut
Exemple #14
0
 def add_callback(self, callback, *args, **kwds):
   if self._done:
     eventloop.queue_call(None, callback, *args, **kwds)
   else:
     self._callbacks.append((callback, args, kwds))