added the option to configure an existing logger (e.g. Rails.logger)

maxdemarzi · Jan 22, 2014 · d0e75a6 · d0e75a6
1 parent 8737a75
commit d0e75a6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/lib/neography/config.rb b/lib/neography/config.rb
@@ -3,7 +3,7 @@ class Config
 
     attr_accessor :protocol, :server, :port, :directory,
       :cypher_path, :gremlin_path,
-      :log_file, :log_enabled, :slow_log_threshold,
+      :log_file, :log_enabled, :logger, :slow_log_threshold,
       :max_threads,
       :authentication, :username, :password,
       :parser, :max_execution_time
@@ -22,6 +22,7 @@ def to_hash
         :gremlin_path       => @gremlin_path,
         :log_file           => @log_file,
         :log_enabled        => @log_enabled,
+        :logger             => @logger,
         :slow_log_threshold => @slow_log_threshold,
         :max_threads        => @max_threads,
         :authentication     => @authentication,

diff --git a/lib/neography/connection.rb b/lib/neography/connection.rb
@@ -89,6 +89,7 @@ def save_local_configuration(config)
       @slow_log_threshold = config[:slow_log_threshold]
       @max_threads        = config[:max_threads]
       @parser             = config[:parser]
+      @logger             = config[:logger]
 
       @max_execution_time = { 'max-execution-time' => config[:max_execution_time] }
       @user_agent     = { "User-Agent" => USER_AGENT }
@@ -104,7 +105,7 @@ def save_local_configuration(config)
       end
 
       if @log_enabled
-        @logger = Logger.new(@log_file)
+        @logger ||= Logger.new(@log_file)
       end
     end