chore(entities): include new query fields for tagsWithMetadata

bpeckNR · Sep 10, 2020 · 94a0504 · 94a0504
1 parent 7f3c33e
commit 94a0504
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 11 deletions.
diff --git a/pkg/entities/entity.go b/pkg/entities/entity.go
@@ -216,6 +216,19 @@ const (
 					reporting
 					type
 `
+	graphqlEntityStructTagsFields = `
+					tagsWithMetadata {
+						key
+						values {
+							mutable
+							value
+						}
+					}
+					tags {
+						key
+						values
+					}
+`
 
 	graphqlApmApplicationEntityFields = `
 					... on ApmApplicationEntity {
@@ -275,13 +288,15 @@ const (
 
 	getEntitiesQuery = `query($guids: [String!]!) { actor { entities(guids: $guids)  {` +
 		graphqlEntityStructFields +
+		graphqlEntityStructTagsFields +
 		graphqlApmApplicationEntityFields +
 		graphqlBrowserApplicationEntityFields +
 		graphqlMobileApplicationEntityFields +
 		` } } }`
 
 	getEntityQuery = `query($guid: String!) { actor { entity(guid: $guid)  {` +
 		graphqlEntityStructFields +
+		graphqlEntityStructTagsFields +
 		graphqlApmApplicationEntityFields +
 		graphqlBrowserApplicationEntityFields +
 		graphqlMobileApplicationEntityFields +

diff --git a/pkg/entities/tags.go b/pkg/entities/tags.go
@@ -122,17 +122,9 @@ func parseTagMutationErrors(errors []tagMutationError) string {
 }
 
 var listTagsQuery = `
-	query($guid:EntityGuid!) {
-		actor {
-			entity(guid: $guid)  {
-				tags {
-					values
-					key
-				}
-			}
-		}
-	}
-`
+	query($guid:EntityGuid!) { actor { entity(guid: $guid)  {` +
+	graphqlEntityStructTagsFields +
+	` } } }`
 
 type listTagsResponse struct {
 	Actor struct {