From 54efaaf78cd6333f69ff7a9e758cf4bd36592ea6 Mon Sep 17 00:00:00 2001
From: Idriss Neumann <idriss.neumann@comwork.io>
Date: Thu, 23 Nov 2023 16:07:32 +0100
Subject: [PATCH] Issue #22: enable sort on _doc

---
 pkg/quickwit/data_query.go      | 5 ++---
 pkg/quickwit/response_parser.go | 1 +
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/pkg/quickwit/data_query.go b/pkg/quickwit/data_query.go
index 96efbf8..0bfb0be 100644
--- a/pkg/quickwit/data_query.go
+++ b/pkg/quickwit/data_query.go
@@ -335,8 +335,7 @@ func processLogsQuery(q *Query, b *es.SearchRequestBuilder, from, to int64, defa
 		sort = es.SortOrderAsc
 	}
 	b.Sort(sort, defaultTimeField, "boolean")
-	// FIXME: check if sort by _doc is needed.
-	// b.Sort(sort, "_doc", "")
+	b.Sort(sort, "_doc", "")
 	b.Size(stringToIntWithDefaultValue(metric.Settings.Get("limit").MustString(), defaultSize))
 	// TODO when hightlight is supported in quickwit
 	// b.AddHighlight()
@@ -352,7 +351,7 @@ func processLogsQuery(q *Query, b *es.SearchRequestBuilder, from, to int64, defa
 func processDocumentQuery(q *Query, b *es.SearchRequestBuilder, from, to int64, defaultTimeField string) {
 	metric := q.Metrics[0]
 	b.Sort(es.SortOrderDesc, defaultTimeField, "boolean")
-	// b.Sort(es.SortOrderDesc, "_doc", "")
+	b.Sort(es.SortOrderDesc, "_doc", "")
 	// Note: not supported in Quickwit
 	// b.AddDocValueField(defaultTimeField)
 	b.Size(stringToIntWithDefaultValue(metric.Settings.Get("size").MustString(), defaultSize))
diff --git a/pkg/quickwit/response_parser.go b/pkg/quickwit/response_parser.go
index 990ca23..818d0bc 100644
--- a/pkg/quickwit/response_parser.go
+++ b/pkg/quickwit/response_parser.go
@@ -108,6 +108,7 @@ func processLogsResponse(res *es.SearchResponse, target *Query, configuredFields
 
 		doc := map[string]interface{}{
 			"_source": flattened,
+			"sort":    hit["sort"],
 		}
 
 		for k, v := range flattened {