fix: big number with trendline can't calculate cumsum (#19542)

This commit is contained in:
Yongjie Zhao
2022-04-06 19:23:01 +08:00
committed by GitHub
parent c4baa826d5
commit 2daa071633
2 changed files with 59 additions and 43 deletions

View File

@@ -18,63 +18,33 @@
*/
import {
buildQueryContext,
PostProcessingResample,
DTTM_ALIAS,
QueryFormData,
} from '@superset-ui/core';
import {
flattenOperator,
pivotOperator,
resampleOperator,
rollingWindowOperator,
sortOperator,
} from '@superset-ui/chart-controls';
const TIME_GRAIN_MAP: Record<string, string> = {
PT1S: 'S',
PT1M: 'min',
PT5M: '5min',
PT10M: '10min',
PT15M: '15min',
PT30M: '30min',
PT1H: 'H',
P1D: 'D',
P1M: 'MS',
P3M: 'QS',
P1Y: 'AS',
// TODO: these need to be mapped carefully, as the first day of week
// can vary from engine to engine
// P1W: 'W',
// '1969-12-28T00:00:00Z/P1W': 'W',
// '1969-12-29T00:00:00Z/P1W': 'W',
// 'P1W/1970-01-03T00:00:00Z': 'W',
// 'P1W/1970-01-04T00:00:00Z': 'W',
};
export default function buildQuery(formData: QueryFormData) {
return buildQueryContext(formData, baseQueryObject => {
// todo: move into full advanced analysis section here
const rollingProc = rollingWindowOperator(formData, baseQueryObject);
const { time_grain_sqla } = formData;
let resampleProc: PostProcessingResample;
if (rollingProc && time_grain_sqla) {
const rule = TIME_GRAIN_MAP[time_grain_sqla];
if (rule) {
resampleProc = {
operation: 'resample',
options: {
method: 'asfreq',
rule,
fill_value: null,
},
};
}
}
const { x_axis } = formData;
const is_timeseries = x_axis === DTTM_ALIAS || !x_axis;
return [
{
...baseQueryObject,
is_timeseries: true,
post_processing: [
sortOperator(formData, baseQueryObject),
resampleProc,
rollingProc,
pivotOperator(formData, {
...baseQueryObject,
index: x_axis,
is_timeseries,
}),
rollingWindowOperator(formData, baseQueryObject),
resampleOperator(formData, baseQueryObject),
flattenOperator(formData, baseQueryObject),
],
},