mirror of
https://github.com/apache/superset.git
synced 2026-07-12 17:55:38 +00:00
87 lines
2.7 KiB
TypeScript
87 lines
2.7 KiB
TypeScript
/**
|
|
* 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 {
|
|
buildQueryContext,
|
|
QueryFormData,
|
|
PostProcessingRule,
|
|
ensureIsArray,
|
|
} from '@superset-ui/core';
|
|
import {
|
|
isTimeComparison,
|
|
timeCompareOperator,
|
|
} from '@superset-ui/chart-controls';
|
|
import { isEmpty } from 'lodash';
|
|
|
|
export default function buildQuery(formData: QueryFormData) {
|
|
const { cols: groupby, extra_form_data } = formData;
|
|
|
|
const queryContextA = buildQueryContext(formData, baseQueryObject => {
|
|
const postProcessing: PostProcessingRule[] = [];
|
|
postProcessing.push(timeCompareOperator(formData, baseQueryObject));
|
|
|
|
const nonCustomNorInheritShifts = ensureIsArray(
|
|
formData.time_compare,
|
|
).filter((shift: string) => shift !== 'custom' && shift !== 'inherit');
|
|
const customOrInheritShifts = ensureIsArray(formData.time_compare).filter(
|
|
(shift: string) => shift === 'custom' || shift === 'inherit',
|
|
);
|
|
|
|
let timeOffsets: string[] = [];
|
|
|
|
// Shifts for non-custom or non inherit time comparison
|
|
if (!isEmpty(nonCustomNorInheritShifts)) {
|
|
timeOffsets = nonCustomNorInheritShifts;
|
|
}
|
|
|
|
// Shifts for custom or inherit time comparison
|
|
if (!isEmpty(customOrInheritShifts)) {
|
|
if (customOrInheritShifts.includes('custom')) {
|
|
timeOffsets = timeOffsets.concat([formData.start_date_offset]);
|
|
}
|
|
if (customOrInheritShifts.includes('inherit')) {
|
|
timeOffsets = timeOffsets.concat(['inherit']);
|
|
}
|
|
}
|
|
|
|
if (
|
|
extra_form_data?.time_compare &&
|
|
!timeOffsets.includes(extra_form_data.time_compare)
|
|
) {
|
|
timeOffsets = [extra_form_data.time_compare];
|
|
}
|
|
|
|
return [
|
|
{
|
|
...baseQueryObject,
|
|
groupby,
|
|
post_processing: postProcessing,
|
|
time_offsets:
|
|
isTimeComparison(formData, baseQueryObject) ||
|
|
extra_form_data?.time_compare
|
|
? ensureIsArray(timeOffsets)
|
|
: [],
|
|
},
|
|
];
|
|
});
|
|
|
|
return {
|
|
...queryContextA,
|
|
};
|
|
}
|