rusackas commented on code in PR #42296:
URL: https://github.com/apache/superset/pull/42296#discussion_r3627524905
##########
superset-frontend/plugins/plugin-chart-echarts/src/Timeseries/EchartsTimeseries.tsx:
##########
@@ -343,6 +345,55 @@ export default function EchartsTimeseries({
});
});
+ // Filters for the clicked x-axis value, so Drill By can subset the
+ // drilled data to the clicked bar/point rather than only the series
+ const xAxisFilters: BinaryQueryObjectFilterClause[] = [];
+ const xAxisCol =
+ // if the xAxis is '__timestamp', granularity_sqla will be the
column of filter
+ xAxis.label === DTTM_ALIAS ? formData.granularitySqla : xAxis.label;
+ if (data && xAxis.type === AxisType.Time && xAxisCol) {
+ // For horizontal orientation the [x, value] pair is swapped
+ const xValue = Array.isArray(data)
+ ? data[categoryAxisValueIndex]
+ : data;
+ const xAxisFilter = getTemporalXAxisDrillByFilter(
+ xAxisCol,
+ xValue,
+ formData.timeGrainSqla,
+ String(xValueFormatter(xValue as number)),
+ );
+ if (xAxisFilter) {
+ xAxisFilters.push(xAxisFilter);
+ }
+ } else if (xAxis.type === AxisType.Category && xAxisCol) {
+ const categoryAxisValue = getCategoryAxisValue(
+ data,
+ eventParams.name,
+ );
+ if (categoryAxisValue !== undefined) {
+ // A category axis can still sit on a temporal column when the
+ // axis is forced categorical; filter by time bucket in that case
+ const xAxisFilter =
+ coltypeMapping?.[getColumnLabel(xAxis.label)] ===
+ GenericDataType.Temporal
Review Comment:
I think Codeant has this backwards. `coltypeMapping` is keyed by the result
column names, so when the axis is `__timestamp` that is the key holding the
temporal type. `xAxisCol` is the source `granularitySqla`, which isn't in the
map. Same lookup `transformProps` uses for `xAxisDataType`.
##########
superset-frontend/src/components/Chart/ChartContextMenu/ChartContextMenu.tsx:
##########
@@ -155,17 +156,29 @@ const ChartContextMenu = (
const [drillModalIsOpen, setDrillModalIsOpen] = useState(false);
const [drillByColumn, setDrillByColumn] = useState<Column>();
+ // Drill by config as selected in the submenu (e.g. with the chosen
+ // x-axis/series filter scope applied), used over the raw context filters
+ const [selectedDrillByConfig, setSelectedDrillByConfig] =
+ useState<ContextMenuFilters['drillBy']>();
Review Comment:
The submenu-selected config path is already covered in
`DrillBySubmenu.test.tsx`, where the scope selector drives the `drillByConfig`
handed to `onDrillBy`. The `??` fallback here is trivial glue over that, so I
do not think a separate test earns its keep.
##########
superset-frontend/plugins/plugin-chart-echarts/src/utils/xAxisDrillByFilter.ts:
##########
@@ -0,0 +1,137 @@
+/**
+ * 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 {
+ BinaryQueryObjectFilterClause,
+ QueryFormColumn,
+ TimeGranularity,
+} from '@superset-ui/core';
+
+const DAY_MS = 24 * 60 * 60 * 1000;
+
+/**
+ * Format a Date as a naive ISO datetime string (UTC, no timezone suffix),
+ * the format Superset's time range parser expects, e.g. "2021-01-01T00:00:00".
+ */
+export const formatNaiveDateTime = (date: Date): string =>
+ date.toISOString().slice(0, 19);
+
+/**
+ * Given the start (label) of a time bucket and its time grain, return the
+ * [since, until) range covering the bucket, using calendar-aware UTC
+ * arithmetic. Week-ending grains are labeled by the last day of the bucket,
+ * so their range extends backwards from the label. Returns undefined for
+ * unknown grains.
+ */
+export const getTimeBucketRange = (
+ bucketLabel: Date,
+ grain: TimeGranularity,
+): { since: Date; until: Date } | undefined => {
+ const until = new Date(bucketLabel.getTime());
+ switch (grain) {
+ case TimeGranularity.SECOND:
+ until.setUTCSeconds(until.getUTCSeconds() + 1);
+ break;
+ case TimeGranularity.MINUTE:
+ until.setUTCMinutes(until.getUTCMinutes() + 1);
+ break;
+ case TimeGranularity.FIVE_MINUTES:
+ until.setUTCMinutes(until.getUTCMinutes() + 5);
+ break;
+ case TimeGranularity.TEN_MINUTES:
+ until.setUTCMinutes(until.getUTCMinutes() + 10);
+ break;
+ case TimeGranularity.FIFTEEN_MINUTES:
+ until.setUTCMinutes(until.getUTCMinutes() + 15);
+ break;
+ case TimeGranularity.THIRTY_MINUTES:
+ until.setUTCMinutes(until.getUTCMinutes() + 30);
+ break;
+ case TimeGranularity.HOUR:
+ until.setUTCHours(until.getUTCHours() + 1);
+ break;
+ case TimeGranularity.DATE:
+ case TimeGranularity.DAY:
+ until.setUTCDate(until.getUTCDate() + 1);
+ break;
+ case TimeGranularity.WEEK:
+ case TimeGranularity.WEEK_STARTING_SUNDAY:
+ case TimeGranularity.WEEK_STARTING_MONDAY:
+ until.setUTCDate(until.getUTCDate() + 7);
+ break;
+ case TimeGranularity.WEEK_ENDING_SATURDAY:
+ case TimeGranularity.WEEK_ENDING_SUNDAY:
+ // These buckets are labeled with their last day: the bucket spans
+ // the 6 days before the label plus the label day itself.
+ until.setUTCDate(until.getUTCDate() + 1);
+ return { since: new Date(bucketLabel.getTime() - 6 * DAY_MS), until };
+ case TimeGranularity.MONTH:
+ until.setUTCMonth(until.getUTCMonth() + 1);
+ break;
+ case TimeGranularity.QUARTER:
+ until.setUTCMonth(until.getUTCMonth() + 3);
+ break;
+ case TimeGranularity.YEAR:
+ until.setUTCFullYear(until.getUTCFullYear() + 1);
+ break;
+ default:
+ return undefined;
+ }
+ return { since: new Date(bucketLabel.getTime()), until };
+};
+
+/**
+ * Build a drill-by filter clause matching the clicked value on a temporal
+ * x-axis. When a known time grain is active, the clause is a TEMPORAL_RANGE
+ * covering the clicked bucket; otherwise it falls back to an exact match on
+ * the timestamp.
+ */
+export const getTemporalXAxisDrillByFilter = (
+ col: QueryFormColumn,
+ value: unknown,
+ grain?: TimeGranularity,
+ formattedVal?: string,
+): BinaryQueryObjectFilterClause | undefined => {
+ if (
+ !col ||
+ (typeof value !== 'number' &&
+ typeof value !== 'string' &&
+ !(value instanceof Date))
+ ) {
+ return undefined;
+ }
+ const bucketLabel = value instanceof Date ? value : new Date(value);
+ if (Number.isNaN(bucketLabel.getTime())) {
+ return undefined;
+ }
+ const range = grain ? getTimeBucketRange(bucketLabel, grain) : undefined;
+ if (!range) {
+ return {
+ col,
+ op: '==',
+ val: formatNaiveDateTime(bucketLabel),
+ formattedVal,
+ };
Review Comment:
This only fires on the no-grain fallback, and the second-precision format is
what Superset's time parser expects. I am not sure a fractional-second `==`
would round-trip on the backend anyway, and a grain-less temporal axis
bucketing to sub-second is a corner I would rather not chase. Leaving as-is.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
