Fix #401: use a Railtie to set Hashie.logger on rails boot.

3 files changed, 13 insertions, 6 deletions

diff --git a/lib/hashie.rb b/lib/hashie.rb
index c0f2caa..19b12bd 100644
--- a/lib/hashie.rb
+++ b/lib/hashie.rb
@@ -56,4 +56,6 @@ module Hashie
     include Hashie::Extensions::StringifyKeys::ClassMethods
     include Hashie::Extensions::SymbolizeKeys::ClassMethods
   end
+
+  require 'hashie/railtie' if defined?(::Rails)
 end
diff --git a/lib/hashie/logger.rb b/lib/hashie/logger.rb
index 4f8fe5a..065738f 100644
--- a/lib/hashie/logger.rb
+++ b/lib/hashie/logger.rb
@@ -5,12 +5,7 @@ module Hashie
   #
   # @return [Logger]
   def self.logger
-    @logger ||=
-      if defined?(::Rails)
-        Rails.logger
-      else
-        Logger.new(STDOUT)
-      end
+    @logger ||= Logger.new(STDOUT)
   end
 
   # Sets the logger that Hashie uses for reporting errors.
diff --git a/lib/hashie/railtie.rb b/lib/hashie/railtie.rb
new file mode 100644
index 0000000..f69a13f
--- /dev/null
+++ b/lib/hashie/railtie.rb
@@ -0,0 +1,10 @@
+require 'rails/railtie'
+
+module Hashie
+  class Railtie < Rails::Railtie
+    # Set the Hashie.logger to use Rails.logger when used with rails.
+    initializer 'hashie.configure_logger', after: 'initialize_logger' do
+      Hashie.logger = Rails.logger
+    end
+  end
+end