propert handling of columns

Author: Igor Drobiazko

lib/actions/write.js

@@ -24,8 +24,11 @@ function init(cfg) {
         };
 
         if (cfg.reader.columns) {
-            console.log('Configured columns:', cfg.reader.columns);
-            options.columns = cfg.reader.columns;
+
+            const columns = Object.keys(_.keyBy(cfg.reader.columns, 'property'));
+
+            console.log('Configured column names:', columns);
+            options.columns = columns;
         }
 
         stringifier = csv.stringify(options);
@@ -66,12 +69,15 @@ function processAction(msg, cfg) {
             console.log('Emitting message %j', messageToEmit);
             self.emit('data', messageToEmit);
 
-            yield init();
+            yield init(cfg);
         });
     }, 10000);
 
-    const columns = cfg.reader.columns;
-    const row = columns ? _.pick(msg.body, columns) : msg.body;
+    let row = msg.body;
+    if( cfg.reader.columns) {
+        const columns = Object.keys(_.keyBy(cfg.reader.columns, 'property'));
+        row = _.pick(msg.body, columns);
+    }
     stringifier.write(row);
     rowCount++;
     this.emit('end');