feat: add tag flag in profiling computation

Kanaries · Jun 26, 2024 · ba79f59 · ba79f59
1 parent bf58f50
commit ba79f59
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 5 deletions.
diff --git a/packages/graphic-walker/src/components/dataTable/profiling.tsx b/packages/graphic-walker/src/components/dataTable/profiling.tsx
@@ -1,15 +1,14 @@
 import { ComponentType, useCallback, useContext, useEffect, useMemo, useRef, useState } from 'react';
 import { IComputationFunction, ISemanticType } from '../../interfaces';
-import { profileNonmialField, profileQuantitativeField } from '../../computation';
+import { profileNonmialField, profileQuantitativeField, wrapComputationWithTag } from '../../computation';
 import React from 'react';
 import { formatDate, isNotEmpty } from '../../utils';
 import Tooltip from '../tooltip';
-import { uiThemeContext, themeContext, vegaThemeContext } from '../../store/theme';
+import { themeContext, vegaThemeContext } from '../../store/theme';
 import { parsedOffsetDate } from '../../lib/op/offset';
 import embed, { VisualizationSpec } from 'vega-embed';
 import { format } from 'd3-format';
 import { getTheme } from '../../utils/useTheme';
-import { parseColorToHSL } from '@/utils/colors';
 
 export interface FieldProfilingProps {
  field: string;
@@ -19,7 +18,7 @@ export interface FieldProfilingProps {
 function NominalProfiling({ computation, field, valueRenderer = (s) => `${s}` }: FieldProfilingProps & { valueRenderer?: (v: string | number) => string }) {
  const [stat, setStat] = useState<Awaited<ReturnType<typeof profileNonmialField>>>();
  useEffect(() => {
- profileNonmialField(computation, field).then(setStat);
+ profileNonmialField(wrapComputationWithTag(computation, "profiling"), field).then(setStat);
  }, [computation, field]);
 
  if (!isNotEmpty(stat)) {
@@ -85,7 +84,7 @@ const formatter = format('~s');
 function QuantitativeProfiling({ computation, field }: FieldProfilingProps) {
  const [stat, setStat] = useState<Awaited<ReturnType<typeof profileQuantitativeField>>>();
  useEffect(() => {
- profileQuantitativeField(computation, field).then(setStat);
+ profileQuantitativeField(wrapComputationWithTag(computation, "profiling"), field).then(setStat);
  }, [computation, field]);
  if (!isNotEmpty(stat)) {
  return <div className="h-24 flex items-center justify-center">Loading...</div>;

diff --git a/packages/graphic-walker/src/computation/index.ts b/packages/graphic-walker/src/computation/index.ts
@@ -640,3 +640,12 @@ export async function profileQuantitativeField(service: IComputationFunction, fi
  binValues,
  };
 }
+
+export function wrapComputationWithTag(service: IComputationFunction, tag: string) {
+ return (payload: IDataQueryPayload) => {
+ return service({
+ ...payload,
+ tag,
+ });
+ };
+}
diff --git a/packages/graphic-walker/src/interfaces.ts b/packages/graphic-walker/src/interfaces.ts
@@ -631,6 +631,7 @@ export type IDataQueryWorkflowStep = IFilterWorkflowStep | ITransformWorkflowSte
 
 export interface IDataQueryPayload {
  workflow: IDataQueryWorkflowStep[];
+ tag?: string;
  limit?: number;
  offset?: number;
 }