Merge pull request #1 from sdemurjian/master

Write properties from recordset to csvStream
adnan-kamili · Sep 21, 2016 · 777d6dc · 777d6dc
2 parents 29c7208 + bf96d8e
commit 777d6dc
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -27,7 +27,8 @@ Node.js module to export MS SQL database to CSV
         ignoreList: ["sysdiagrams"], // tables to ignore
         tables: [],                  // empty to export all the tables
         outputDirectory: 'somedir',
-        log: true
+        log: true,
+        header: false                // true to export column names as csv header
     };
 
     mssqlExport(dbconfig, options).then(function(){

diff --git a/lib/util.js b/lib/util.js
@@ -6,7 +6,8 @@ var options = {
     ignoreList: ["sysdiagrams"],
     tables: [],
     outputDirectory: '.',
-    log: true
+    log: true,
+    header: false
 };
 
 module.exports = {
@@ -35,6 +36,18 @@ module.exports = {
             var request = new sql.Request();
             request.stream = true;
             request.query('select * from ' + tableName);
+
+            if (options.header) {
+                request.on('recordset', function (columns) {
+                    var keys = [];
+                    for (var key in columns) {
+                        if (columns.hasOwnProperty(key)) {
+                            keys.push(key);
+                        }
+                    }
+                    csvStream.write(keys);
+                });
+            }
             request.on('row', function (row) {
                 // fix for null values getting replaced with empty string
                 for (var key in row) {
@@ -68,6 +81,9 @@ module.exports = {
         if(opts.log !== 'undefined') {
             options.log = opts.log;
         }
+        if(opts.header !== 'undefined') {
+            options.header = opts.header;
+        }
         return options;
     }
 }