Merge pull request #481 from neo4jrb/migration

fix max per batch value
neo4jrb · Sep 21, 2014 · 39f405c · 39f405c
2 parents 123f45f + c61c757
commit 39f405c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lib/neo4j/migration.rb b/lib/neo4j/migration.rb
@@ -33,7 +33,7 @@ def setup
       def add_ids_to(model)
         require 'benchmark'
 
-        max_per_batch = (ENV['MAX_PER_BATCH'] || max_per_batch).to_i
+        max_per_batch = (ENV['MAX_PER_BATCH'] || default_max_per_batch).to_i
 
         label = model.mapped_label_name
         property = model.primary_key
@@ -52,8 +52,8 @@ def add_ids_to(model)
 
           return if nodes_left == 0
           to_set = [nodes_left, max_per_batch].min
-          new_ids = to_set.times.map { new_id_for(model) }
 
+          new_ids = to_set.times.map { new_id_for(model) }
           begin
             last_time_taken = Benchmark.realtime do
               Neo4j::Session.query("MATCH (n:`#{label}`) WHERE NOT has(n.#{property})
@@ -72,7 +72,7 @@ def add_ids_to(model)
         end
       end
 
-      def max_per_batch
+      def default_max_per_batch
         900
       end