Metric labels do not include hostname when two queries are used

This change adds a custom label input box and allows the user to customize the labels. It corresponding filter is created, these variables are available: * $filter_site: Site name * $filter_host_name: Host name * $filter_host_in_group: Group containing the host * $filter_service: Service name * $filter_service_in_group: Group containing the service $label is always available and contains the original label sent by Checmk. CMK-15138
Checkmk · Nov 27, 2023 · 0fa54aa · 0fa54aa
1 parent 076a845
commit 0fa54aa
Show file tree

Hide file tree

Showing 16 changed files with 248 additions and 31 deletions.
diff --git a/package.json b/package.json
@@ -78,5 +78,5 @@
     "react-dom": "17.0.2",
     "tslib": "2.5.3"
   },
-  "packageManager": "yarn@"
+  "packageManager": "yarn@1.22.21"
 }
diff --git a/src/RequestSpec.ts b/src/RequestSpec.ts
@@ -31,6 +31,8 @@ export interface RequestSpec {
   service_in_group: NegatableOption | undefined;
 
   graph: string | undefined;
+
+  label: string | undefined;
 }
 
 // subset of RequestSpec used with the Filters Component

diff --git a/src/backend/rest.ts b/src/backend/rest.ts
@@ -8,6 +8,7 @@ import {
   FieldType,
   MetricFindValue,
   MutableDataFrame,
+  ScopedVars,
   TimeRange,
   dateTime,
 } from '@grafana/data';
@@ -16,7 +17,7 @@ import { Aggregation, GraphType, MetricFindQuery } from 'RequestSpec';
 import * as process from 'process';
 
 import { CmkQuery } from '../types';
-import { createCmkContext, replaceVariables, toLiveStatusQuery, updateQuery } from '../utils';
+import { createCmkContext, replaceVariables, toLiveStatusQuery, updateMetricTitles, updateQuery } from '../utils';
 import { Backend, DatasourceOptions } from './types';
 import { validateRequestSpec } from './validate';
 
@@ -26,18 +27,20 @@ type RestApiError = {
   title: string;
 };
 
+export type MetricResponse = {
+  color: string;
+  data_points: number[];
+  line_type: string;
+  title: string;
+};
+
 type RestApiGraphResponse = {
   time_range: {
     start: string;
     end: string;
   };
   step: number;
-  metrics: Array<{
-    color: string;
-    data_points: number[];
-    line_type: string;
-    title: string;
-  }>;
+  metrics: MetricResponse[];
 };
 
 type CommonRequest = {
@@ -155,7 +158,7 @@ export default class RestApiBackend implements Backend {
     const promises = request.targets
       .filter((target) => !target.hide)
       .map((target) => {
-        return this.getSingleGraph(request.range, target);
+        return this.getSingleGraph(request.range, target, request.scopedVars);
       });
     return await Promise.all(promises).then((data) => ({ data }));
   }
@@ -250,7 +253,7 @@ export default class RestApiBackend implements Backend {
     return result;
   }
 
-  async getSingleGraph(range: TimeRange, query: CmkQuery): Promise<DataQueryResponseData> {
+  async getSingleGraph(range: TimeRange, query: CmkQuery, scopedVars: ScopedVars = {}): Promise<DataQueryResponseData> {
     // it's not about a single graph line, but a single chart. grafana supports
     // to query multiple graphs in one request, but we have to unwind this, as
     // our api only supports a single chart/query per api call.
@@ -322,6 +325,8 @@ export default class RestApiBackend implements Backend {
 
     const { time_range, step, metrics } = response.data;
 
+    updateMetricTitles(metrics, query, scopedVars);
+
     const timeValues = [];
     let currentTime: DateTime = dateTime(time_range.start);
     const endTime: DateTime = dateTime(time_range.end);

diff --git a/src/backend/web.ts b/src/backend/web.ts
@@ -1,10 +1,17 @@
-import { DataQueryRequest, DataQueryResponse, FieldType, MetricFindValue, MutableDataFrame } from '@grafana/data';
+import {
+  DataQueryRequest,
+  DataQueryResponse,
+  FieldType,
+  MetricFindValue,
+  MutableDataFrame,
+  ScopedVars,
+} from '@grafana/data';
 import { BackendSrvRequest, FetchError, FetchResponse, getBackendSrv } from '@grafana/runtime';
 import { MetricFindQuery } from 'RequestSpec';
 import { defaults, get, isUndefined, zip } from 'lodash';
 
 import { CmkQuery, defaultQuery } from '../types';
-import { updateQuery } from '../utils';
+import { updateMetricTitles, updateQuery } from '../utils';
 import {
   WebAPiGetGraphResult,
   WebApiResponse,
@@ -89,7 +96,7 @@ export default class WebApiBackend implements Backend {
       .map((target) => {
         // TODO: check if the defaults call is still necessary.
         const query = defaults(target, defaultQuery);
-        return this.getGraphQuery([from, to], query);
+        return this.getGraphQuery([from, to], query, options.scopedVars);
       });
     return Promise.all(promises).then((data) => ({ data }));
   }
@@ -145,7 +152,11 @@ export default class WebApiBackend implements Backend {
     }
   }
 
-  async getGraphQuery(range: number[], query: CmkQuery): Promise<MutableDataFrame<unknown>> {
+  async getGraphQuery(
+    range: number[],
+    query: CmkQuery,
+    scopedVars: ScopedVars = {}
+  ): Promise<MutableDataFrame<unknown>> {
     updateQuery(query);
     const graph = get(query, 'requestSpec.graph');
     if (isUndefined(graph) || graph === '') {
@@ -173,8 +184,11 @@ export default class WebApiBackend implements Backend {
     if (response.result_code !== 0) {
       throw new Error(`${response.result}`);
     }
+
     const { start_time, step, curves } = response.result;
 
+    updateMetricTitles(curves, query, scopedVars);
+
     const frame = new MutableDataFrame({
       refId: query.refId,
       fields: [

diff --git a/src/types.ts b/src/types.ts
@@ -78,3 +78,12 @@ export interface SecureJsonData {
 export interface ResponseDataAutocomplete {
   choices: Array<[string, string]>;
 }
+
+export enum label {
+  ORIGINAL = '$label',
+  SITE = '$filter_site',
+  HOSTNAME = '$filter_host_name',
+  HOST_IN_GROUP = '$filter_host_in_group',
+  SERVICE = '$filter_service',
+  SERVICE_IN_GROUP = '$filter_service_in_group',
+}
diff --git a/src/ui/QueryEditor.tsx b/src/ui/QueryEditor.tsx
@@ -1,12 +1,12 @@
 import { QueryEditorProps, SelectableValue } from '@grafana/data';
-import { InlineFieldRow, VerticalGroup } from '@grafana/ui';
+import { Button, Card, Icon, InlineFieldRow, Toggletip, VerticalGroup } from '@grafana/ui';
 import React from 'react';
 
 import { DataSource } from '../DataSource';
 import { Aggregation, GraphType, RequestSpec } from '../RequestSpec';
-import { CmkQuery, DataSourceOptions } from '../types';
+import { CmkQuery, DataSourceOptions, label } from '../types';
 import { aggregationToPresentation, updateQuery } from '../utils';
-import { CheckMkSelect } from './components';
+import { CheckMkSelect, GenericField } from './components';
 import { Filters } from './filters';
 import { labelForRequestSpecKey } from './utils';
 
@@ -19,6 +19,8 @@ export const QueryEditor = (props: Props): JSX.Element => {
   const [qAggregation, setQAggregation] = React.useState(rs.aggregation || 'off');
   const [qGraphType, setQGraphType] = React.useState(rs.graph_type || 'predefined_graph');
   const [qGraph, setQGraph] = React.useState(rs.graph);
+  const [qLabel, setQLabel] = React.useState(rs.label);
+
   const filters: Partial<RequestSpec> = {
     // make sure to only include keys filters should change, otherwise they could
     // overwrite other fields!
@@ -43,6 +45,7 @@ export const QueryEditor = (props: Props): JSX.Element => {
     graph_type: qGraphType,
     graph: qGraph,
     aggregation: qAggregation,
+    label: qLabel,
   };
 
   // TODO: not sure if this is a dirty hack or a great solution:
@@ -109,6 +112,56 @@ export const QueryEditor = (props: Props): JSX.Element => {
     />
   );
 
+  const LabelsTooltip = (
+    <Toggletip
+      content={
+        <Card>
+          <Card.Description>
+            These variables are supported if provided as filters:
+            <ul>
+              {[
+                [label.SITE, 'Site name'],
+                [label.HOSTNAME, 'Host name'],
+                [label.HOST_IN_GROUP, 'Group containing the host'],
+                [label.SERVICE, 'Service name'],
+                [label.SERVICE_IN_GROUP, 'Group containing the service'],
+              ].map((lbl) => (
+                <li key={lbl[0]}>
+                  <strong>{lbl[0]}</strong>: {lbl[1]}
+                </li>
+              ))}
+            </ul>
+            <br />
+            Also, the original label is available as
+            <ul>
+              <li>
+                <strong>{label.ORIGINAL}</strong>
+              </li>
+            </ul>
+          </Card.Description>
+        </Card>
+      }
+      closeButton={false}
+      placement="right-end"
+    >
+      <Button type="button" fill="text" size="sm" tooltip={'Variables support information'}>
+        <Icon name="question-circle" />
+      </Button>
+    </Toggletip>
+  );
+
+  const labelField = (
+    <GenericField
+      requestSpecKey="label"
+      label={labelForRequestSpecKey('label', requestSpec)}
+      value={qLabel}
+      onChange={setQLabel}
+      dataTestId="custom-label-field"
+      placeholder={label.ORIGINAL}
+      suffix={LabelsTooltip}
+    ></GenericField>
+  );
+
   if (editionMode === 'RAW') {
     return (
       <VerticalGroup>
@@ -122,6 +175,7 @@ export const QueryEditor = (props: Props): JSX.Element => {
         />
         {graphTypeSelect}
         {graphSelect}
+        {labelField}
       </VerticalGroup>
     );
   } else {
@@ -139,6 +193,7 @@ export const QueryEditor = (props: Props): JSX.Element => {
         />
         {graphTypeSelect}
         {graphSelect}
+        {labelField}
       </VerticalGroup>
     );
   }

diff --git a/src/ui/components.tsx b/src/ui/components.tsx
@@ -288,6 +288,67 @@ export const Filter = <T extends RequestSpecNegatableOptionKeys>(props: FilterPr
   );
 };
 
+interface GenericFieldProps<Key extends RequestSpecStringKeys> extends CommonProps<RequestSpec[Key]> {
+  requestSpecKey: Key;
+  children?: React.ReactNode;
+  width?: number;
+  tooltip?: string;
+  dataTestId?: string;
+  placeholder?: string;
+  prefix?: React.ReactNode;
+  suffix?: React.ReactNode;
+}
+export const GenericField = <T extends RequestSpecStringKeys>(props: GenericFieldProps<T>) => {
+  const {
+    label,
+    width = 32,
+    onChange,
+    value,
+    requestSpecKey,
+    children = null,
+    tooltip,
+    placeholder = 'none',
+    prefix,
+    suffix,
+  } = props;
+  const { dataTestId = `${requestSpecKey}-filter-input` } = props;
+
+  const [textValue, setTextValue] = React.useState(value !== undefined ? value : '');
+
+  const debouncedOnChange = React.useMemo(
+    () =>
+      debounce((newValue) => {
+        onChange(newValue);
+      }, 1000),
+    [onChange]
+  );
+
+  const onValueChange = (event: React.ChangeEvent<HTMLInputElement>) => {
+    setTextValue(event.target.value); // update text input
+    debouncedOnChange(event.target.value); // only the last debounce call comes through
+  };
+
+  return (
+    <HorizontalGroup>
+      <InlineField label={label} labelWidth={LABEL_WIDTH} tooltip={tooltip}>
+        <>
+          <Input
+            width={width}
+            type="text"
+            value={textValue}
+            onChange={onValueChange}
+            placeholder={placeholder}
+            data-test-id={dataTestId}
+            prefix={prefix}
+            suffix={suffix}
+          />
+          {children}
+        </>
+      </InlineField>
+    </HorizontalGroup>
+  );
+};
+
 const SingleTag = (props: {
   index: number;
   onChange: (newValue: TagValue) => void;

diff --git a/src/ui/utils.ts b/src/ui/utils.ts
@@ -14,6 +14,7 @@ export const labelForRequestSpecKey = (key: keyof RequestSpec, rq: Partial<Reque
     aggregation: 'Aggregation',
     graph_type: 'Graph type',
     graph: rq.graph_type === 'predefined_graph' ? 'Predefined graph' : 'Single metric',
+    label: 'Custom label',
   };
   return table[key];
 };
diff --git a/src/utils.ts b/src/utils.ts
@@ -3,9 +3,10 @@ import { getTemplateSrv } from '@grafana/runtime';
 import { isUndefined } from 'lodash';
 
 import { Aggregation, FiltersRequestSpec, NegatableOption, RequestSpec, TagValue } from './RequestSpec';
-import { CmkQuery } from './types';
+import { MetricResponse } from './backend/rest';
+import { CmkQuery, label } from './types';
 import { Presentation } from './ui/autocomplete';
-import { requestSpecFromLegacy } from './webapi';
+import { WebApiCurve, requestSpecFromLegacy } from './webapi';
 
 export const titleCase = (str: string): string => str[0].toUpperCase() + str.slice(1).toLowerCase();
 
@@ -276,3 +277,25 @@ export function toLiveStatusQuery(filter: Partial<FiltersRequestSpec>, table: 'h
     query: query,
   };
 }
+
+type GrapResponse = WebApiCurve | MetricResponse;
+
+export function updateMetricTitles(metrics: GrapResponse[], query: CmkQuery, scopedVars: ScopedVars = {}) {
+  const titleTemplate = query.requestSpec?.label || label.ORIGINAL;
+  if (titleTemplate !== label.ORIGINAL) {
+    scopedVars = {
+      ...scopedVars,
+      [label.SITE.substring(1)]: { value: query.requestSpec?.site || '' },
+      [label.HOSTNAME.substring(1)]: { value: query.requestSpec?.host_name || '' },
+      [label.HOST_IN_GROUP.substring(1)]: { value: query.requestSpec?.host_in_group?.value || '' },
+      [label.SERVICE.substring(1)]: { value: query.requestSpec?.service || '' },
+      [label.SERVICE_IN_GROUP.substring(1)]: { value: query.requestSpec?.service_in_group?.value || '' },
+      [label.ORIGINAL.substring(1)]: { value: '' },
+    };
+
+    metrics.forEach((metric) => {
+      scopedVars[label.ORIGINAL.substring(1)] = { value: metric.title };
+      metric.title = getTemplateSrv().replace(titleTemplate, scopedVars);
+    });
+  }
+}