Merge pull request #229 from keen/bugfix/extraction-keys

fix issue with extraction missing keys

lib/dataset/utils/create-extraction-keys.js

@@ -0,0 +1,12 @@
+import { flatten } from '../utils/flatten';
+
+export function createExtractionKeys(results) {
+  const keys = new Set();
+  results.forEach((result) => {
+    Object.keys(flatten(result)).map((key) => {
+      keys.add(key);
+    });
+  });
+
+  return keys;
+}

lib/dataset/utils/parsers.js

@@ -3,6 +3,7 @@ let Dataset; /* injected */
 import { each } from '../../utils/each';
 import { flatten } from '../utils/flatten';
 import { valueAtDeepKey } from '../utils/object';
+import { createExtractionKeys } from '../utils/create-extraction-keys';
 
 export default function initialize(lib){
   Dataset = lib;
@@ -162,8 +163,8 @@ function parseExtraction(){
     const datasetExtraction = new Dataset()
       .type('extraction');
 
-    // get keys of the Object, static schema
-    let names = Object.keys(flatten(res.result[0]));
+    const datasetKeys = createExtractionKeys(res.result);
+    let names = Array.from(datasetKeys);
 
     const tableConfig = config && config.table;
     if (tableConfig && tableConfig.schema === 'dynamic') {
@@ -181,13 +182,13 @@ function parseExtraction(){
       });
       names = Object.keys(keys);
     }
-    
+
     let nameI = 0;
     names.forEach(value => {
       datasetExtraction.set([ value, '0' ], value);
       nameI++;
     });
-    
+
     if (Dataset) {
       if (config
         && config.table
@@ -214,7 +215,7 @@ function parseHeatmapAxis() {
   return function(res) {
     const heatmapDataset = new Dataset()
       .type('heatmap');
-    
+
     each(res.result, function(value, i){
       const objKeys = Object.keys(value);
       const x = value[objKeys[0]];