Config

Author: Igor Drobiazko

component.json

@@ -41,6 +41,11 @@
     "write_attachment": {
       "main": "./lib/actions/write.js",
       "title": "Write CSV attachment",
+      "fields": {
+        "reader": {
+          "viewClass": "CSVReadView"
+        }
+      },
       "metadata": {
         "in": {},
         "out": {}

lib/actions/write.js

@@ -1,6 +1,7 @@
 const request = require('request-promise');
 const co = require('co');
 const csv = require('csv');
+const _ = require('lodash');
 const messages = require('elasticio-node').messages;
 const client = require('elasticio-rest-node')();
 
@@ -12,12 +13,22 @@ let rowCount = 0;
 exports.init = init;
 exports.process = processAction;
 
-function init() {
+function init(cfg) {
     return co(function* gen() {
 
-        stringifier = csv.stringify({
-            header: true
-        });
+        const delimiter = cfg.reader.delimiter || ',';
+        console.log('Using delimiter: \'%s\'', delimiter);
+        const options = {
+            header: true,
+            delimiter
+        };
+
+        if (cfg.reader.columns) {
+            console.log('Configured columns:', cfg.reader.columns);
+            options.columns = cfg.reader.columns;
+        }
+
+        stringifier = csv.stringify(options);
 
         signedUrl = yield client.resources.storage.createSignedUrl();
 
@@ -28,7 +39,7 @@ function init() {
     });
 }
 
-function processAction(msg) {
+function processAction(msg, cfg) {
     const self = this;
 
     if (timeout) {
@@ -59,7 +70,9 @@ function processAction(msg) {
         });
     }, 10000);
 
-    stringifier.write(msg.body);
+    const columns = cfg.reader.columns;
+    const row = columns ? _.pick(msg.body, columns) : msg.body;
+    stringifier.write(row);
     rowCount++;
     this.emit('end');
 }

package.json

@@ -31,6 +31,7 @@
     "elasticio-node": "0.0.7",
     "elasticio-sailor-nodejs": "2.1.3",
     "elasticio-rest-node": "1.0.2",
+    "lodash": "^4.17.4",
     "moment": "^2.10.6",
     "node-uuid": "^1.4.3",
     "q": "^1.4.1",