Merge pull request #5 from mullerpeter/feature/custom-variables

✨ feat: add support for custom variables
mullerpeter · Nov 21, 2022 · 3eda628 · 3eda628
2 parents e0db144 + 56708a3
commit 3eda628
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/package.json b/package.json
@@ -1,7 +1,7 @@
 {
   "name": "mullerpeter-databricks-datasource",
   "private": true,
-  "version": "0.0.6",
+  "version": "0.0.7",
   "description": "Databricks SQL Connector",
   "scripts": {
     "build": "grafana-toolkit plugin:build",

diff --git a/src/datasource.ts b/src/datasource.ts
@@ -1,9 +1,16 @@
 import { DataSourceInstanceSettings } from '@grafana/data';
-import { DataSourceWithBackend } from '@grafana/runtime';
+import { DataSourceWithBackend, getTemplateSrv } from '@grafana/runtime';
 import { MyDataSourceOptions, MyQuery } from './types';
 
 export class DataSource extends DataSourceWithBackend<MyQuery, MyDataSourceOptions> {
   constructor(instanceSettings: DataSourceInstanceSettings<MyDataSourceOptions>) {
     super(instanceSettings);
   }
+  applyTemplateVariables(query: MyQuery) {
+    const templateSrv = getTemplateSrv();
+    return {
+      ...query,
+      rawSqlQuery: query.rawSqlQuery ? templateSrv.replace(query.rawSqlQuery) : ''
+    };
+  }
 }