From 74bd22923eb16284cfb618794c84946ffaf52e63 Mon Sep 17 00:00:00 2001
From: Krzysztof Kiewicz <krzysztof@quesma.com>
Date: Fri, 27 Sep 2024 13:17:16 +0200
Subject: [PATCH] Done

---
 quesma/queryparser/query_translator.go | 52 +++++++-------------------
 1 file changed, 13 insertions(+), 39 deletions(-)

diff --git a/quesma/queryparser/query_translator.go b/quesma/queryparser/query_translator.go
index 46ba48e9a..85bed85ac 100644
--- a/quesma/queryparser/query_translator.go
+++ b/quesma/queryparser/query_translator.go
@@ -12,7 +12,6 @@ import (
 	"quesma/quesma/config"
 	"quesma/schema"
 	"quesma/util"
-	"strings"
 )
 
 type JsonMap = map[string]interface{}
@@ -178,44 +177,19 @@ func (cw *ClickhouseQueryTranslator) makeTotalCount(queries []*model.Query, resu
 				if len(results[queryIdx]) == 0 {
 					continue
 				}
-				totalCount = 0
-				for rowIdx, row := range results[queryIdx] {
-					// sum over all rows
-					if len(row.Cols) == 0 {
-						continue
-					}
-
-					// if group by key exists + it's the same as last, we have already counted it and need to continue
-					newKey := true
-					if rowIdx != 0 { // for first row we always have a new key
-						for colIdx, cell := range row.Cols {
-							// find first group by key
-							if strings.HasSuffix(cell.ColName, "__key_0") {
-								if row.Cols[colIdx].Value == results[queryIdx][rowIdx-1].Cols[colIdx].Value {
-									newKey = false
-								}
-								break
-							}
-						}
-					}
-					if !newKey {
-						continue
-					}
-
-					// find the count column
-					for _, cell := range row.Cols {
-						// FIXME THIS is hardcoded for now, as we don't have a way to get the name of the column
-						if cell.ColName == "metric____quesma_total_count_col_0" {
-							switch v := cell.Value.(type) {
-							case uint64:
-								totalCount += int(v)
-							case int:
-								totalCount += v
-							case int64:
-								totalCount += int(v)
-							default:
-								logger.ErrorWithCtx(cw.Ctx).Msgf("Unknown type of count %v %t", v, v)
-							}
+				firstRow := results[queryIdx][0]
+				for _, cell := range firstRow.Cols {
+					// FIXME THIS is hardcoded for now, as we don't have a way to get the name of the column
+					if cell.ColName == "metric____quesma_total_count_col_0" {
+						switch v := cell.Value.(type) {
+						case uint64:
+							totalCount = int(v)
+						case int:
+							totalCount = v
+						case int64:
+							totalCount = int(v)
+						default:
+							logger.ErrorWithCtx(cw.Ctx).Msgf("Unknown type of count %v %t", v, v)
 						}
 					}
 				}