-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmerge-csv.js
45 lines (33 loc) · 1.15 KB
/
merge-csv.js
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
const fs = require("fs");
const path = require("path");
const csv = require("csv-parser");
const fastCsv = require("fast-csv");
const inputFolder = path.join(__dirname, "files");
const outputFile = path.join(__dirname, "combined.csv");
async function mergeCsvFiles() {
const files = fs.readdirSync(inputFolder).filter((file) => file.endsWith(".csv"));
if (files.length === 0) {
console.error("No CSV files found in the folder.");
return;
}
let isHeaderWritten = false;
const writeStream = fs.createWriteStream(outputFile);
for (const file of files) {
const filePath = path.join(inputFolder, file);
const readStream = fs.createReadStream(filePath).pipe(csv());
const rows = [];
for await (const row of readStream) {
rows.push(row);
}
fastCsv
.write(rows, { headers: !isHeaderWritten })
.on("finish", () => {
console.log(`${file} merged successfully!`);
})
.pipe(writeStream, { end: false });
isHeaderWritten = true; // Write headers only once
}
writeStream.end();
console.log(`All CSV files have been merged into ${outputFile}`);
}
mergeCsvFiles().catch(console.error);