enable running db checkpoints before loads

Author: moxious

src/run-configuration.js

@@ -131,6 +131,7 @@ const generateFromArgs = (args) => {
     const obj = _.merge({
         generator: schema ? new MockData(schema) : null,
         batchSize,
+        runcheckpoint: args.runcheckpoint,
         checkpointFreq: args.checkpoint || process.env.CHECKPOINT_FREQUENCY || 5000,
         failFast,
         phase: 'NOT_STARTED',
@@ -158,6 +159,7 @@ module.exports = {
             .describe('u', 'username')
             .describe('p', 'password')
             .describe('d', 'database')
+            .describe('runcheckpoint', 'whether to run db checkpointing or not')
             .describe('schema', 'batch schema file')
             .describe('batchsize', 'number of records per batch, usable only with schema')
             .describe('n', 'number of hits on the database')
@@ -166,6 +168,7 @@ module.exports = {
             .describe('query', 'Cypher query to run')
             .default('concurrency', 10)
             .default('checkpoint', 5000)
+            .default('runcheckpoint', false)
             .demandOption(['p'])
             .argv;
     },

src/workload.js

@@ -76,7 +76,19 @@ class Workload {
     const runStrategySetupPromises = Object.keys(this.strategyTable)
       .map(stratName => this.strategyTable[stratName].setup(this.driver));
 
-    return Promise.all(runStrategySetupPromises);
+    return Promise.all(runStrategySetupPromises)
+      .then(results => {
+        if (this.runConfig.runcheckpoint) {
+          const opts = {};
+          if (this.runConfig.database) { opts.database = this.runConfig.database; }
+          const session = this.driver.session(opts);
+          console.log('Calling db.checkpoint method to flush prior to run');
+          return session.run('CALL db.checkpoint()')
+            .then(session.close);
+        }
+
+        return results;
+      });
   }
 
   getRunConfiguration() { return this.runConfig; }