Fix Warmup for Neo4j 3.2

neo4j-contrib · Aug 2, 2017 · d3c45be · d3c45be
1 parent 1a41bcb
commit d3c45be
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/src/main/java/apoc/warmup/Warmup.java b/src/main/java/apoc/warmup/Warmup.java
@@ -9,10 +9,7 @@
 import org.neo4j.kernel.impl.store.format.standard.NodeRecordFormat;
 import org.neo4j.kernel.impl.store.format.standard.PropertyRecordFormat;
 import org.neo4j.kernel.impl.store.format.standard.RelationshipRecordFormat;
-import org.neo4j.kernel.impl.store.record.AbstractBaseRecord;
-import org.neo4j.kernel.impl.store.record.NodeRecord;
-import org.neo4j.kernel.impl.store.record.PropertyRecord;
-import org.neo4j.kernel.impl.store.record.RelationshipRecord;
+import org.neo4j.kernel.impl.store.record.*;
 import org.neo4j.kernel.internal.GraphDatabaseAPI;
 import org.neo4j.procedure.Context;
 import org.neo4j.procedure.Description;
@@ -88,7 +85,11 @@ public <R extends AbstractBaseRecord> long loadRecords(int recordsPerPage, long
             pages++;
             record.setId(id);
             record.clear();
-            recordStore.ensureHeavy(record);
+            try {
+                recordStore.getRecord(id, record, RecordLoad.NORMAL);
+            } catch(Exception ignore) {
+                // ignore
+            }
         }
         return pages;
     }