Merge pull request #285 from dhanush-2397/dev

Write to CSV function changes
Sunbird-cQube · Oct 12, 2023 · a95f221 · a95f221
2 parents 6fd85bb + b5ee4d5
commit a95f221
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/src/ingestion/services/generic-function.ts b/src/ingestion/services/generic-function.ts
@@ -88,17 +88,22 @@ export class GenericFunction {
             } while (true);
             // get the lock on the file
             this.currentlyLockedFiles[fileName] = true;
-
+            const fileExists = fs.existsSync(fileName);
             const stream = fs.createWriteStream(fileName,{flags:'a'});
-            await csv.write(inputArray, { headers: true})
-                .pipe(stream)
-                .on('finish', () => {
+            if(!fileExists){
+                await csv.write(inputArray, { headers: true }).pipe(stream);
+            }else{
+                await csv.write(inputArray).pipe(stream);
+            }
+            // await csv.write(inputArray, { headers: true})
+            //     .pipe(stream)
+                stream.on('finish', () => {
                     console.log('CSV file has been written successfully');
                     // delete the lock after writing
                     delete this.currentlyLockedFiles[fileName];
                     resolve('done')
                 })
-                .on('error', (err) => {
+                stream.on('error', (err) => {
                     console.error('Error writing CSV file:', err);
                     reject(err)
                 });