lib/adhearsion/foundation/object.rb in adhearsion-2.0.0 vs lib/adhearsion/foundation/object.rb in adhearsion-2.0.1

@@ -1,31 +1,20 @@
 # encoding: utf-8
 
 require 'adhearsion/logging'
 
 class Object
+  include Adhearsion::Logging::HasLogger
+
   undef :pb_logger
   def pb_logger
     logger
   end
+end
 
-  def logger_id
-    self
-  end
-
-  def method_missing(method_id, *arguments, &block)
-    if method_id == Adhearsion::Logging::METHOD
-      self.class.send :define_method, method_id do
-        Logging.logger[logger_id]
-      end
-      Logging.logger[logger_id]
-    else
-      super
+module Celluloid
+  class ActorProxy
+    def logger
+      Actor.call @mailbox, :logger
     end
-  end
-
-  def respond_to?(method_id, include_private = false)
-    return true if method_id == Adhearsion::Logging::METHOD
-
-    super
   end
 end