Merge pull request #4 from joseEnrique/master

added DIRECTORY var into export and import
masumsoft · Jan 10, 2018 · 56209d5 · 56209d5
2 parents 657a11f + d28caa6
commit 56209d5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/export.js b/export.js
@@ -14,7 +14,7 @@ if (!KEYSPACE) {
 
 var USER = process.env.USER;
 var PASSWORD = process.env.PASSWORD;
-
+var DIRECTORY = process.env.DIRECTORY || "./data";
 var authProvider;
 
 if (USER && PASSWORD) {
@@ -28,7 +28,7 @@ function processTableExport(table) {
     console.log('==================================================');
     console.log('Reading table: ' + table);
     return new Promise(function(resolve, reject) {
-        var jsonfile = fs.createWriteStream('data/' + table + '.json');
+        var jsonfile = fs.createWriteStream(DIRECTORY +"/" + table + '.json');
         jsonfile.on('error', function (err) {
             reject(err);
         });

diff --git a/import.js b/import.js
@@ -15,6 +15,7 @@ if (!KEYSPACE) {
 
 var USER = process.env.USER;
 var PASSWORD = process.env.PASSWORD;
+var DIRECTORY = process.env.DIRECTORY || "./data";
 
 var authProvider;
 
@@ -82,7 +83,7 @@ function processTableImport(table) {
                 }
 
                 console.log('Creating read stream from: ' + table + '.json');
-                var jsonfile = fs.createReadStream('data/' + table + '.json', {encoding: 'utf8'});
+                var jsonfile = fs.createReadStream(DIRECTORY + '/' + table + '.json', {encoding: 'utf8'});
                 var readStream = jsonfile.pipe(jsonStream.parse('*'));
                 var queryPromises = [];
                 var processed = 0;