Fix total records response (#113)

* Fix total records response * error handling * change var name to be more readable
similarweb · Jul 26, 2020 · 2c3dc9e · 2c3dc9e
1 parent 7882a97
commit 2c3dc9e
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/api/storage/elasticsearch/elasticsearch.go b/api/storage/elasticsearch/elasticsearch.go
@@ -310,10 +310,21 @@ func (sm *StorageManager) GetResources(resourceType string, executionID string,
 	ResourceNameQ := elastic.NewMatchQuery("ResourceName", resourceType)
 	generalQ := elastic.NewBoolQuery()
 	generalQ = generalQ.Must(componentQ).Must(deploymentQ).Must(ResourceNameQ).Must(dynamicMatchQuery...)
+	searchResultTotalHits, err := sm.client.Search().
+		Query(generalQ).
+		Pretty(true).
+		Size(0).
+		Do(context.Background())
+
+	if err != nil {
+		log.WithError(err).Error("elasticsearch query error")
+		return resources, err
+	}
+
 	searchResult, err := sm.client.Search().
 		Query(generalQ).
 		Pretty(true).
-		Size(100).
+		Size(int(searchResultTotalHits.TotalHits())).
 		Do(context.Background())
 
 	if err != nil {