Merge pull request #161 from TaoVonQi/main

Add support for dashboard query variables.

src/components/VariableQueryEditor.tsx

@@ -0,0 +1,36 @@
+import React, { useState } from 'react';
+import { SQLQuery } from 'types';
+
+interface VariableQueryProps {
+  query: SQLQuery;
+  onChange: (query: SQLQuery, definition: string) => void;
+}
+
+export const VariableQueryEditor = ({ onChange, query }: VariableQueryProps) => {
+  const [state, setState] = useState(query);
+
+  const saveQuery = () => {
+    onChange(state, `${state.queryText}`);
+  };
+
+  const handleChange = (event: React.FormEvent<HTMLInputElement>) =>
+    setState({
+      ...state,
+      [event.currentTarget.name]: event.currentTarget.value,
+    });
+
+  return (
+    <>
+      <div className="gf-form">
+        <span className="gf-form-label width-10">Query</span>
+        <input
+          name="rawQuery"
+          className="gf-form-input"
+          onBlur={saveQuery}
+          onChange={handleChange}
+          value={state.queryText}
+        />
+      </div>
+    </>
+  );
+};

src/datasource.ts

@@ -1,12 +1,37 @@
-import {DataSourceInstanceSettings, CoreApp, ScopedVars, VariableWithMultiSupport} from '@grafana/data'
-import {DataSourceWithBackend, getTemplateSrv} from '@grafana/runtime'
+import {DataQueryResponse, MetricFindValue, DataSourceInstanceSettings, CoreApp, ScopedVars, VariableWithMultiSupport} from '@grafana/data'
+import {frameToMetricFindValue, DataSourceWithBackend, getTemplateSrv} from '@grafana/runtime'
 import {SQLQuery, FlightSQLDataSourceOptions, DEFAULT_QUERY} from './types'
+import { lastValueFrom } from 'rxjs';
 
 export class FlightSQLDataSource extends DataSourceWithBackend<SQLQuery, FlightSQLDataSourceOptions> {
   constructor(instanceSettings: DataSourceInstanceSettings<FlightSQLDataSourceOptions>) {
     super(instanceSettings)
   }
 
+async metricFindQuery(queryText: string, options?: any): Promise<MetricFindValue[]> {
+      const target: SQLQuery = {
+        refId: 'metricFindQuery',
+        queryText,
+        rawEditor: true,
+        format: 'table'
+      };
+      return lastValueFrom(
+        super.query({
+          ...(options ?? {}), // includes 'range'
+          targets: [target],
+        })
+      ).then(this.toMetricFindValue);
+  }
+
+  toMetricFindValue(rsp: DataQueryResponse): MetricFindValue[] {
+      const data = rsp.data ?? [];
+      // Create MetricFindValue object for all frames
+      const values = data.map((d) => frameToMetricFindValue(d)).flat();
+      // Filter out duplicate elements
+      return values.filter((elm, idx, self) => idx === self.findIndex((t) => t.text === elm.text));
+
+  }
+
   getDefaultQuery(_: CoreApp): Partial<SQLQuery> {
     return DEFAULT_QUERY
   }