From 9d326f5a8be14f064d20fcdccfdc6b646d4ca449 Mon Sep 17 00:00:00 2001
From: Andrea Cavalli <andrea@cavallium.it>
Date: Wed, 4 Aug 2021 01:16:17 +0200
Subject: [PATCH] Human readable representation of total hits count

---
 src/main/data-generator/lucene-query.yaml                | 1 +
 .../java/it/cavallium/dbengine/lucene/LuceneUtils.java   | 9 +++++++++
 2 files changed, 10 insertions(+)

diff --git a/src/main/data-generator/lucene-query.yaml b/src/main/data-generator/lucene-query.yaml
index 08c215e..16297eb 100644
--- a/src/main/data-generator/lucene-query.yaml
+++ b/src/main/data-generator/lucene-query.yaml
@@ -197,6 +197,7 @@ versions:
           onlyTopScores: boolean
           computeScores: boolean
       TotalHitsCount:
+        stringRepresenter: "it.cavallium.dbengine.lucene.LuceneUtils.toHumanReadableString"
         data:
           value: long
           exact: boolean
diff --git a/src/main/java/it/cavallium/dbengine/lucene/LuceneUtils.java b/src/main/java/it/cavallium/dbengine/lucene/LuceneUtils.java
index 70a7487..fd06fd4 100644
--- a/src/main/java/it/cavallium/dbengine/lucene/LuceneUtils.java
+++ b/src/main/java/it/cavallium/dbengine/lucene/LuceneUtils.java
@@ -478,4 +478,13 @@ public class LuceneUtils {
 				totalHitsCount.exact() && totalHitsCount1.exact()
 		);
 	}
+
+	@SuppressWarnings("unused")
+	public static String toHumanReadableString(TotalHitsCount totalHitsCount) {
+		if (totalHitsCount.exact()) {
+			return Long.toString(totalHitsCount.value());
+		} else {
+			return totalHitsCount.value() + "+";
+		}
+	}
 }