From 089ec1d94e91a6dee498683b8f3a0137edeb2599 Mon Sep 17 00:00:00 2001
From: Dominik Prokop <dominik.prokop@grafana.com>
Date: Mon, 12 Dec 2022 06:19:28 -0800
Subject: [PATCH] Scenes: Use DataSourceRef only in the query variable type
 (#60157)

---
 public/app/features/scenes/scenes/queryVariableDemo.tsx       | 2 +-
 .../scenes/variables/variants/query/QueryVariable.tsx         | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/public/app/features/scenes/scenes/queryVariableDemo.tsx b/public/app/features/scenes/scenes/queryVariableDemo.tsx
index ad209a5c449..6b6c3cf9dec 100644
--- a/public/app/features/scenes/scenes/queryVariableDemo.tsx
+++ b/public/app/features/scenes/scenes/queryVariableDemo.tsx
@@ -29,7 +29,7 @@ export function getQueryVariableDemo(standalone: boolean): Scene {
           name: 'instance (using datasource variable)',
           refresh: VariableRefresh.onTimeRangeChanged,
           query: { query: 'label_values(go_gc_duration_seconds, ${metric})' },
-          datasource: '${datasource}',
+          datasource: { uid: '${datasource}' },
         }),
         new QueryVariable({
           name: 'label values (on time range refresh)',
diff --git a/public/app/features/scenes/variables/variants/query/QueryVariable.tsx b/public/app/features/scenes/variables/variants/query/QueryVariable.tsx
index b090851c9ba..39776425d47 100644
--- a/public/app/features/scenes/variables/variants/query/QueryVariable.tsx
+++ b/public/app/features/scenes/variables/variants/query/QueryVariable.tsx
@@ -30,7 +30,7 @@ import { metricNamesToVariableValues } from './utils';
 
 export interface QueryVariableState extends MultiValueVariableState {
   type: 'query';
-  datasource: DataSourceRef | string | null;
+  datasource: DataSourceRef | null;
   query: any;
   regex: string;
   refresh: VariableRefresh;
@@ -126,7 +126,7 @@ export class QueryVariable extends MultiValueVariable<QueryVariableState> {
   }
 
   private async getDataSource(): Promise<DataSourceApi> {
-    return getDataSourceSrv().get(this.state.datasource ?? '', {
+    return getDataSourceSrv().get(this.state.datasource, {
       __sceneObject: { text: '__sceneObject', value: this },
     });
   }