Fix rendering regression from the introduction of bignumber (#6937)

In superset-ui 0.8.0, we used bignumber.js to transform numbers in chartProps' payload from plain 64-bit floats to BigNumber instances. This causes a number of charts to render incorrectly when comparison functions in the rendering algorithms operate on BigNumber objects instead of floats. This PR uses the preTransformProps step in SuperChart to transform BigNumber instances back to floats so charts can render properly.

(cherry picked from commit 73cdb37)

superset/assets/package.json

@@ -62,6 +62,7 @@
     "@vx/responsive": "0.0.172",
     "@vx/scale": "^0.0.165",
     "abortcontroller-polyfill": "^1.1.9",
+    "bignumber.js": "^8.1.0",
     "bootstrap": "^3.3.6",
     "bootstrap-slider": "^10.0.0",
     "brace": "^0.11.1",

superset/assets/spec/javascripts/chart/transformBigNumber_spec.js

@@ -0,0 +1,53 @@
+/**
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+import BigNumber from 'bignumber.js';
+import transform from 'src/chart/transformBigNumber';
+
+describe('transformBigNumber', () => {
+  it('should transform BigNumber on its own', () => {
+    expect(transform(new BigNumber(123.456))).toBe(123.456);
+  });
+
+  it('should transform BigNumber in objects', () => {
+    expect(transform({
+      foo: new BigNumber(123),
+      bar: 456,
+      baz: null,
+    })).toEqual({ foo: 123, bar: 456, baz: null });
+  });
+
+  it('should transform BigNumber in arrays', () => {
+    expect(transform([
+      { foo: new BigNumber(123) },
+      { bar: 456 },
+    ])).toEqual([{ foo: 123 }, { bar: 456 }]);
+  });
+
+  it('should transform BigNumber in nested structures', () => {
+    expect(transform([{
+      x: new BigNumber(123),
+      y: [{ foo: new BigNumber(456) }, { bar: 'str' }],
+      z: { some: [new BigNumber(789)] },
+    }])).toEqual([{
+      x: 123,
+      y: [{ foo: 456 }, { bar: 'str' }],
+      z: { some: [789] },
+    }]);
+  });
+});

superset/assets/src/chart/ChartRenderer.jsx

@@ -24,6 +24,7 @@ import { ChartProps } from '@superset-ui/chart';
 import { Tooltip } from 'react-bootstrap';
 import { Logger, LOG_ACTIONS_RENDER_CHART } from '../logger';
 import SuperChart from '../visualizations/core/components/SuperChart';
+import transformBigNumber from './transformBigNumber';
 
 const propTypes = {
   annotationData: PropTypes.object,
@@ -67,6 +68,7 @@ class ChartRenderer extends React.Component {
     this.handleAddFilter = this.handleAddFilter.bind(this);
     this.handleRenderSuccess = this.handleRenderSuccess.bind(this);
     this.handleRenderFailure = this.handleRenderFailure.bind(this);
+    this.preTransformProps = this.preTransformProps.bind(this);
   }
 
   shouldComponentUpdate(nextProps, nextState) {
@@ -150,6 +152,18 @@ class ChartRenderer extends React.Component {
     });
   }
 
+  preTransformProps(chartProps) {
+    const payload = chartProps.payload;
+    const data = transformBigNumber(payload.data);
+    return new ChartProps({
+      ...chartProps,
+      payload: {
+        ...payload,
+        data,
+      },
+    });
+  }
+
   renderTooltip() {
     const { tooltip } = this.state;
     if (tooltip && tooltip.content) {
@@ -193,6 +207,7 @@ class ChartRenderer extends React.Component {
           className={`${snakeCase(vizType)}`}
           chartType={vizType}
           chartProps={skipChartRendering ? null : this.prepareChartProps()}
+          preTransformProps={this.preTransformProps}
           onRenderSuccess={this.handleRenderSuccess}
           onRenderFailure={this.handleRenderFailure}
         />

superset/assets/src/chart/transformBigNumber.js

@@ -0,0 +1,45 @@
+/**
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+// This method transforms any BigNumber object in the given payload to its
+// 64-bit float representation. It is a temporary fix so charts receive
+// floats instead of BigNumber instances in their props to properly render.
+import BigNumber from 'bignumber.js';
+
+export default function transform(payload) {
+  if (!payload) {
+    return payload;
+  } else if (BigNumber.isBigNumber(payload)) {
+    return payload.toNumber();
+  } else if (payload.constructor === Object) {
+    for (const key in payload) {
+      if (payload.hasOwnProperty(key)) {
+        // Modify in place to prevent creating large payloads
+        // eslint-disable-next-line no-param-reassign
+        payload[key] = transform(payload[key]);
+      }
+    }
+  } else if (payload.constructor === Array) {
+    payload.forEach((elem, idx) => {
+      // Modify in place to prevent creating large payloads
+      // eslint-disable-next-line no-param-reassign
+      payload[idx] = transform(elem);
+    });
+  }
+  return payload;
+}