feat: return data attribute from es search results (#197)

raystack · Jan 16, 2023 · 7df7acd · 7df7acd
1 parent 00bc9b6
commit 7df7acd
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 7 deletions.
diff --git a/core/asset/discovery.go b/core/asset/discovery.go
@@ -39,13 +39,14 @@ type SearchConfig struct {
 
 // SearchResult represents an item/result in a list of search results
 type SearchResult struct {
-	ID          string            `json:"id"`
-	URN         string            `json:"urn"`
-	Title       string            `json:"title"`
-	Type        string            `json:"type"`
-	Service     string            `json:"service"`
-	Description string            `json:"description"`
-	Labels      map[string]string `json:"labels"`
+	ID          string                 `json:"id"`
+	URN         string                 `json:"urn"`
+	Title       string                 `json:"title"`
+	Type        string                 `json:"type"`
+	Service     string                 `json:"service"`
+	Description string                 `json:"description"`
+	Labels      map[string]string      `json:"labels"`
+	Data        map[string]interface{} `json:"data"`
 }
 
 // ToAsset returns search result as asset
@@ -58,5 +59,6 @@ func (sr SearchResult) ToAsset() Asset {
 		Service:     sr.Service,
 		Description: sr.Description,
 		Labels:      sr.Labels,
+		Data:        sr.Data,
 	}
 }
diff --git a/internal/store/elasticsearch/discovery_search_repository.go b/internal/store/elasticsearch/discovery_search_repository.go
@@ -257,6 +257,7 @@ func (repo *DiscoveryRepository) toSearchResults(hits []searchHit) []asset.Searc
 			Title:       r.Name,
 			Service:     r.Service,
 			Labels:      r.Labels,
+			Data:        r.Data,
 		})
 	}
 	return results