[incubator-superset] branch master updated: including auto generated avg metrics in druid (#4718)

[gr...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

graceguo pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-superset.git


The following commit(s) were added to refs/heads/master by this push:
     new 1ef856e  including auto generated avg metrics in druid (#4718)
1ef856e is described below

commit 1ef856e57a2987eea318613d285690a42993f865
Author: Gabe Lyons <ga...@airbnb.com>
AuthorDate: Mon Apr 2 16:08:25 2018 -0700

    including auto generated avg metrics in druid (#4718)
---
 .../explore/components/controls/MetricsControl.jsx | 19 ++++++++------
 .../explore/components/MetricsControl_spec.jsx     | 29 +++++++++++++++++++++-
 2 files changed, 39 insertions(+), 9 deletions(-)

diff --git a/superset/assets/javascripts/explore/components/controls/MetricsControl.jsx b/superset/assets/javascripts/explore/components/controls/MetricsControl.jsx
index 9a16f52..ffc2fe2 100644
--- a/superset/assets/javascripts/explore/components/controls/MetricsControl.jsx
+++ b/superset/assets/javascripts/explore/components/controls/MetricsControl.jsx
@@ -74,13 +74,8 @@ function getDefaultAggregateForColumn(column) {
   return null;
 }
 
-const autoGeneratedMetricRegex = /^(LONG|DOUBLE|FLOAT)?(SUM|AVG|MAX|MIN|COUNT)\([A-Z_][A-Z0-9_]*\)$/i;
-function isAutoGeneratedMetric(savedMetric) {
-  return (
-    autoGeneratedMetricRegex.test(savedMetric.expression) ||
-    autoGeneratedMetricRegex.test(savedMetric.verbose_name)
-  );
-}
+const sqlaAutoGeneratedMetricRegex = /^(LONG|DOUBLE|FLOAT)?(SUM|AVG|MAX|MIN|COUNT)\([A-Z_][A-Z0-9_]*\)$/i;
+const druidAutoGeneratedMetricRegex = /^(LONG|DOUBLE|FLOAT)?(SUM|MAX|MIN|COUNT)\([A-Z_][A-Z0-9_]*\)$/i;
 
 export default class MetricsControl extends React.PureComponent {
   constructor(props) {
@@ -90,6 +85,7 @@ export default class MetricsControl extends React.PureComponent {
     this.checkIfAggregateInInput = this.checkIfAggregateInInput.bind(this);
     this.optionsForSelect = this.optionsForSelect.bind(this);
     this.selectFilterOption = this.selectFilterOption.bind(this);
+    this.isAutoGeneratedMetric = this.isAutoGeneratedMetric.bind(this);
     this.optionRenderer = VirtualizedRendererWrap(option => (
       <MetricDefinitionOption option={option} />
     ), { ignoreAutogeneratedMetrics: true });
@@ -209,6 +205,13 @@ export default class MetricsControl extends React.PureComponent {
     });
   }
 
+  isAutoGeneratedMetric(savedMetric) {
+    if (this.props.datasourceType === 'druid') {
+      return druidAutoGeneratedMetricRegex.test(savedMetric.verbose_name);
+    }
+    return sqlaAutoGeneratedMetricRegex.test(savedMetric.expression);
+  }
+
   selectFilterOption(option, filterValue) {
     if (this.state.aggregateInInput) {
       let endIndex = filterValue.length;
@@ -220,7 +223,7 @@ export default class MetricsControl extends React.PureComponent {
         (option.column_name.toLowerCase().indexOf(valueAfterAggregate.toLowerCase()) >= 0);
     }
     return option.optionName &&
-      (!option.metric_name || !isAutoGeneratedMetric(option)) &&
+      (!option.metric_name || !this.isAutoGeneratedMetric(option)) &&
       (option.optionName.toLowerCase().indexOf(filterValue.toLowerCase()) >= 0);
   }
 
diff --git a/superset/assets/spec/javascripts/explore/components/MetricsControl_spec.jsx b/superset/assets/spec/javascripts/explore/components/MetricsControl_spec.jsx
index 346a287..23c8776 100644
--- a/superset/assets/spec/javascripts/explore/components/MetricsControl_spec.jsx
+++ b/superset/assets/spec/javascripts/explore/components/MetricsControl_spec.jsx
@@ -24,6 +24,7 @@ const defaultProps = {
     { metric_name: 'sum__value', expression: 'SUM(energy_usage.value)' },
     { metric_name: 'avg__value', expression: 'AVG(energy_usage.value)' },
   ],
+  datasourceType: 'sqla',
 };
 
 function setup(overrides) {
@@ -188,7 +189,7 @@ describe('MetricsControl', () => {
 
   describe('option filter', () => {
     it('includes user defined metrics', () => {
-      const { wrapper } = setup();
+      const { wrapper } = setup({ datasourceType: 'druid' });
 
       expect(!!wrapper.instance().selectFilterOption(
         {
@@ -200,6 +201,19 @@ describe('MetricsControl', () => {
       )).to.be.true;
     });
 
+    it('includes auto generated avg metrics for druid', () => {
+      const { wrapper } = setup({ datasourceType: 'druid' });
+
+      expect(!!wrapper.instance().selectFilterOption(
+        {
+          metric_name: 'a_metric',
+          optionName: 'a_metric',
+          expression: 'AVG(metric)',
+        },
+        'a',
+      )).to.be.true;
+    });
+
     it('includes columns and aggregates', () => {
       const { wrapper } = setup();
 
@@ -214,6 +228,19 @@ describe('MetricsControl', () => {
       )).to.be.true;
     });
 
+    it('excludes auto generated avg metrics for sqla', () => {
+      const { wrapper } = setup();
+
+      expect(!!wrapper.instance().selectFilterOption(
+        {
+          metric_name: 'a_metric',
+          optionName: 'a_metric',
+          expression: 'AVG(metric)',
+        },
+        'a',
+      )).to.be.false;
+    });
+
     it('excludes auto generated metrics', () => {
       const { wrapper } = setup();
 

-- 
To stop receiving notification emails like this one, please contact
graceguo@apache.org.