From 8b9419222a95d8e6799afe7064084bffa71ac88a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?G=C3=A1bor=20Farkas?= <gabor.farkas@gmail.com>
Date: Mon, 16 May 2022 08:28:41 +0200
Subject: [PATCH] loki: backend-mode: fix experimental stream-mode (#48936)

---
 public/app/plugins/datasource/loki/datasource.ts | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/public/app/plugins/datasource/loki/datasource.ts b/public/app/plugins/datasource/loki/datasource.ts
index 07265aa5c90..81ef6e7e145 100644
--- a/public/app/plugins/datasource/loki/datasource.ts
+++ b/public/app/plugins/datasource/loki/datasource.ts
@@ -175,6 +175,17 @@ export class LokiDatasource
         targets: request.targets.map(getNormalizedLokiQuery),
       };
 
+      const streamQueries = fixedRequest.targets.filter((q) => q.queryType === LokiQueryType.Stream);
+      if (config.featureToggles.lokiLive && streamQueries.length > 0 && fixedRequest.rangeRaw?.to === 'now') {
+        // this is still an in-development feature,
+        // we do not support mixing stream-queries with normal-queries for now.
+        const streamRequest = {
+          ...fixedRequest,
+          targets: streamQueries,
+        };
+        return merge(...streamQueries.map((q) => doLokiChannelStream(q, this, streamRequest)));
+      }
+
       if (fixedRequest.liveStreaming) {
         return this.runLiveQueryThroughBackend(fixedRequest);
       } else {