dan/sql2csv
1
0
Fork 0
mirror of https://github.com/danbulant/sql2csv synced 2026-06-20 06:51:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add task counter, stop process when all tasks done

Browse source
This commit is contained in:
danbulant 2019-09-28 18:39:14 +02:00
parent 4158f07c53
commit 6abad395ce
1 changed files with 24 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
index.js
View file

@ -102,13 +102,27 @@ con.connect(function(err) {
log("\x1b[32mMySQL Connected!"); log("\x1b[32mMySQL Connected!");
s2c.setConnection(con); s2c.setConnection(con);
var running = {
aInternal: 0,
aListener: function(val) {},
set a(val) {
this.aInternal = val;
this.aListener(val);
},
get a() {
return this.aInternal;
},
registerListener: function(listener) {
this.aListener = listener;
}
}
fs.readdirSync('./input/').forEach(file => { fs.readdirSync('./input/').forEach(file => {
var fileName = file; var fileName = file;
file = "./input/" + file; file = "./input/" + file;
if(fs.statSync(file).isDirectory()) return;//skip directories if(fs.statSync(file).isDirectory()) return;//skip directories
var query = fs.readFileSync(file); var query = fs.readFileSync(file);
log("Running " + query + " from file " + file); log("Running " + query + " from file " + file);
running.a++;
s2c.query(query + '') s2c.query(query + '')
.then(result => { .then(result => {
var loc = "/output/" + fileName; var loc = "/output/" + fileName;
@ -119,8 +133,16 @@ con.connect(function(err) {
loc += ".csv"; loc += ".csv";
fs.writeFile(loc, result.csv, { flag: "w"}, (err)=>{if(err)error(err)}); fs.writeFile(loc, result.csv, { flag: "w"}, (err)=>{if(err)error(err)});
success(`Task from file ${file} done in ` + Math.round(result.end - result.start) + "ms, saved to "+loc); success(`Task from file ${file} done in ` + Math.round(result.end - result.start) + "ms, saved to "+loc);
running.a--;
}) })
.catch(err => {error(err); process.exit(0)}) .catch(err => {error(err); process.exit(0)});
running.registerListener(function(val) {
if(val == 0){
success("Everything done, exiting");
process.exit(0);
}
log(val + " tasks running");
});
}); });
}); });