Set app logger

[arvados.git] / lib / google / api_client / railtie.rb

diff --git a/lib/google/api_client/railtie.rb b/lib/google/api_client/railtie.rb
index 703921b7ef76984ac20ff0121bc2911dda007a3f..51edb33b5153edd7420443e044c36b8276bdbcf6 100644 (file)
--- a/lib/google/api_client/railtie.rb
+++ b/lib/google/api_client/railtie.rb
@@ -1,3 +1,4 @@
+require 'rails/railtie'
 require 'google/api_client/logging'
 
 module Google
@@ -9,7 +10,13 @@ module Google
     #
     class Railtie < Rails::Railtie
       initializer 'google-api-client' do |app|
-        Google::APIClient.logger = Rails.logger
+        _logger = case
+                    when app.respond_to?(:logger)        then app.logger
+                    when app.config.respond_to?(:logger) then app.config.logger
+                    else                                      Rails.logger
+                  end
+                  
+        Google::APIClient.logger = _logger
       end
     end
   end