feat(stats): add indexSize and usedIndexSize to IndexStats (#988)

Author: krishna3006b

src/main/java/com/meilisearch/sdk/model/IndexStats.java

@@ -1,7 +1,9 @@
 package com.meilisearch.sdk.model;
 
+import com.fasterxml.jackson.annotation.JsonProperty;
 import java.util.Map;
 import lombok.Getter;
+import lombok.Setter;
 
 /**
  * Stats data structure of a Meilisearch Index
@@ -10,15 +12,35 @@
  *     specification</a>
  */
 @Getter
+@Setter
 public class IndexStats {
+    @JsonProperty("numberOfDocuments")
     protected long numberOfDocuments;
+
+    @JsonProperty("isIndexing")
     protected boolean isIndexing;
+
+    @JsonProperty("fieldDistribution")
     protected Map<String, Integer> fieldDistribution;
+
+    @JsonProperty("rawDocumentDbSize")
     protected long rawDocumentDbSize;
+
+    @JsonProperty("avgDocumentSize")
     protected long avgDocumentSize;
+
+    @JsonProperty("numberOfEmbeddedDocuments")
     protected long numberOfEmbeddedDocuments;
+
+    @JsonProperty("numberOfEmbeddings")
     protected long numberOfEmbeddings;
 
+    @JsonProperty("indexSize")
+    protected long indexSize;
+
+    @JsonProperty("usedIndexSize")
+    protected long usedIndexSize;
+
     public IndexStats() {}
 
     public IndexStats(
@@ -29,12 +51,36 @@ public IndexStats(
             long avgDocumentSize,
             long numberOfEmbeddedDocuments,
             long numberOfEmbeddings) {
+        this(
+                numberOfDocuments,
+                isIndexing,
+                fieldDistribution,
+                rawDocumentDbSize,
+                avgDocumentSize,
+                numberOfEmbeddedDocuments,
+                numberOfEmbeddings,
+                0L,
+                0L);
+    }
+
+    public IndexStats(
+            long numberOfDocuments,
+            boolean isIndexing,
+            Map<String, Integer> fieldDistribution,
+            long rawDocumentDbSize,
+            long avgDocumentSize,
+            long numberOfEmbeddedDocuments,
+            long numberOfEmbeddings,
+            long indexSize,
+            long usedIndexSize) {
         this.numberOfDocuments = numberOfDocuments;
         this.isIndexing = isIndexing;
         this.fieldDistribution = fieldDistribution;
         this.rawDocumentDbSize = rawDocumentDbSize;
         this.avgDocumentSize = avgDocumentSize;
         this.numberOfEmbeddedDocuments = numberOfEmbeddedDocuments;
         this.numberOfEmbeddings = numberOfEmbeddings;
+        this.indexSize = indexSize;
+        this.usedIndexSize = usedIndexSize;
     }
 }

src/test/java/com/meilisearch/sdk/json/GsonJsonHandlerTest.java

@@ -291,6 +291,20 @@ void granularFilterableAttributesWithNullPatternRoundTrip() throws Exception {
         assertThat(decoded[0].getAttributePatterns(), arrayContaining("director", null, "genres"));
     }
 
+    @Test
+    void decodeIndexStatsWithIndexSizeAndUsedIndexSize() {
+        String json =
+                "{\"numberOfDocuments\":10,\"isIndexing\":false,\"fieldDistribution\":{},\"rawDocumentDbSize\":1024,\"avgDocumentSize\":102,\"numberOfEmbeddedDocuments\":0,\"numberOfEmbeddings\":0,\"indexSize\":2048,\"usedIndexSize\":1500}";
+
+        com.meilisearch.sdk.model.IndexStats indexStats =
+                classToTest.decode(json, com.meilisearch.sdk.model.IndexStats.class);
+
+        assertThat(indexStats, is(notNullValue()));
+        assertThat(indexStats.getNumberOfDocuments(), is(equalTo(10L)));
+        assertThat(indexStats.getIndexSize(), is(equalTo(2048L)));
+        assertThat(indexStats.getUsedIndexSize(), is(equalTo(1500L)));
+    }
+
     @Getter
     private static class Container {
         private Key key;

src/test/java/com/meilisearch/sdk/json/JacksonJsonHandlerTest.java

@@ -178,4 +178,18 @@ void settingsWithAdvancedGranularFilterableAttributesRoundTripWithCustomMapper()
                         .getComparison(),
                 is(false));
     }
+
+    @Test
+    void decodeIndexStatsWithIndexSizeAndUsedIndexSize() throws Exception {
+        String json =
+                "{\"numberOfDocuments\":10,\"isIndexing\":false,\"fieldDistribution\":{},\"rawDocumentDbSize\":1024,\"avgDocumentSize\":102,\"numberOfEmbeddedDocuments\":0,\"numberOfEmbeddings\":0,\"indexSize\":2048,\"usedIndexSize\":1500}";
+
+        com.meilisearch.sdk.model.IndexStats indexStats =
+                classToTest.decode(json, com.meilisearch.sdk.model.IndexStats.class);
+
+        assertThat(indexStats, is(notNullValue()));
+        assertThat(indexStats.getNumberOfDocuments(), is(equalTo(10L)));
+        assertThat(indexStats.getIndexSize(), is(equalTo(2048L)));
+        assertThat(indexStats.getUsedIndexSize(), is(equalTo(1500L)));
+    }
 }