Support future calls to self

benlangfeld · Sep 3, 2012 · 768594f · 768594f
1 parent 1e4c6d9
commit 768594f
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lib/celluloid.rb b/lib/celluloid.rb
@@ -381,8 +381,12 @@ def async(meth = nil, *args, &block)
   end
 
   # Handle calls to future within an actor itself
-  def future(meth, *args, &block)
-    Actor.future Thread.current[:actor].mailbox, meth, *args, &block
+  def future(meth = nil, *args, &block)
+    if meth
+      Actor.future Thread.current[:actor].mailbox, meth, *args, &block
+    else
+      Thread.current[:actor].proxy.future
+    end
   end
 end