From 25e1238022c9335601207a2867b054413e416ba4 Mon Sep 17 00:00:00 2001
From: David <david.kaltschmidt@gmail.com>
Date: Thu, 14 May 2020 16:27:26 +0200
Subject: [PATCH] Logs: Fix total bytes process calculation (#24691)

- log stats for Loki are per query
- this change tracks the query stats by refId, preventing the summing of
    the same stats across multiple series of the same response.
---
 public/app/core/logs_model.ts | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/public/app/core/logs_model.ts b/public/app/core/logs_model.ts
index f4f9f647b35..3204b4648ec 100644
--- a/public/app/core/logs_model.ts
+++ b/public/app/core/logs_model.ts
@@ -384,13 +384,19 @@ export function logSeriesToLogsModel(logSeries: DataFrame[]): LogsModel | undefi
 
   // Hack to print loki stats in Explore. Should be using proper stats display via drawer in Explore (rework in 7.1)
   let totalBytes = 0;
+  const queriesVisited: { [refId: string]: boolean } = {};
   for (const series of logSeries) {
     const totalBytesKey = series.meta?.custom?.lokiQueryStatKey;
-    if (totalBytesKey && series.meta.stats) {
-      const byteStat = series.meta.stats.find(stat => stat.displayName === totalBytesKey);
-      if (byteStat) {
-        totalBytes += byteStat.value;
+    // Stats are per query, keeping track by refId
+    const { refId } = series;
+    if (refId && !queriesVisited[refId]) {
+      if (totalBytesKey && series.meta.stats) {
+        const byteStat = series.meta.stats.find(stat => stat.displayName === totalBytesKey);
+        if (byteStat) {
+          totalBytes += byteStat.value;
+        }
       }
+      queriesVisited[refId] = true;
     }
   }
   if (totalBytes > 0) {