-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathcsvAPI.js
More file actions
121 lines (117 loc) · 3.89 KB
/
csvAPI.js
File metadata and controls
121 lines (117 loc) · 3.89 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
const parse = require('csv-parse');
const util = require('util');
const fs = require('fs');
const path = require('path');
const mysql = require('mysql');
const async = require('async');
const co = require('co');
const csvHeaders = require('csv-headers');
const leftpad = require('leftpad');
const dbhost = 'localhost';
const dbuser = 'root';
const dbpass = '';
const dbname = 'Management';
const tblnm = 'test';
const csvfn = '.\\files\\Devices.csv';
new Promise((resolve, reject) => {
csvHeaders({
file : csvfn,
delimiter : ','
}, function(err, headers) {
if (err) reject(err);
else resolve({ headers });
});
})
.then(context => {
return new Promise((resolve, reject) => {
context.db = mysql.createConnection({
host : dbhost,
user : dbuser,
password : dbpass,
database : dbname
});
context.db.connect((err) => {
if (err) {
console.error('error connecting: ' + err.stack);
reject(err);
} else {
resolve(context);
}
});
})
})
.then(context => {
return new Promise((resolve, reject) => {
context.db.query(`DROP TABLE IF EXISTS ${tblnm}`,
[ ],
err => {
if (err) reject(err);
else resolve(context);
})
});
})
.then(context => {
return new Promise((resolve, reject) => {
var fields = '';
var fieldnms = '';
var qs = '';
context.headers.forEach(hdr => {
hdr = hdr.replace(' ', '_');
if (fields !== '') fields += ',';
if (fieldnms !== '') fieldnms += ','
if (qs !== '') qs += ',';
fields += ` ${hdr} TEXT`;
fieldnms += ` ${hdr}`;
qs += ' ?';
});
context.qs = qs;
context.fieldnms = fieldnms;
console.log(`about to create CREATE TABLE IF NOT EXISTS ${tblnm} ( ${fields} )`);
context.db.query(`CREATE TABLE IF NOT EXISTS ${tblnm} ( ${fields} )`,
[ ],
err => {
if (err) reject(err);
else resolve(context);
})
});
})
.then(context => {
return new Promise((resolve, reject) => {
fs.createReadStream(csvfn).pipe(parse({
delimiter: ',',
columns: true,
relax_column_count: true
}, (err, data) => {
if (err) return reject(err);
async.eachSeries(data, (datum, next) => {
// console.log(`about to run INSERT INTO ${tblnm} ( ${context.fieldnms} ) VALUES ( ${context.qs} )`);
var d = [];
try {
context.headers.forEach(hdr => {
// In some cases the data fields have embedded blanks,
// which must be trimmed off
let tp = datum[hdr].trim();
// For a field with an empty string, send NULL instead
d.push(tp === '' ? null : tp);
});
} catch (e) {
console.error(e.stack);
}
// console.log(`${d.length}: ${util.inspect(d)}`);
if (d.length > 0) {
context.db.query(`INSERT INTO ${tblnm} ( ${context.fieldnms} ) VALUES ( ${context.qs} )`, d,
err => {
if (err) { console.error(err); next(err); }
else setTimeout(() => { next(); });
});
} else { console.log(`empty row ${util.inspect(datum)} ${util.inspect(d)}`); next(); }
},
err => {
if (err) reject(err);
else resolve(context);
});
}));
});
})
.then(context => { context.db.end(); })
.catch(err => { console.error(err.stack); });