feat(aws): Fetch tags for aws_elasticache_clusters (#5911)

disq · web-flow · commit 65b5093e65b9 · 2022-12-22T12:38:42.000Z
Implements #5899
diff --git a/plugins/source/aws/codegen/recipes/elasticache.go b/plugins/source/aws/codegen/recipes/elasticache.go
@@ -22,6 +22,11 @@ func ElastiCacheResources() []*Resource {
 						Resolver: `schema.PathResolver("ARN")`,
 						Options:  schema.ColumnCreationOptions{PrimaryKey: true},
 					},
+					{
+						Name:     "tags",
+						Type:     schema.TypeJSON,
+						Resolver: `resolveClusterTags`,
+					},
 				}...),
 		},
 		{
diff --git a/plugins/source/aws/docs/tables/aws_elasticache_clusters.md b/plugins/source/aws/docs/tables/aws_elasticache_clusters.md
@@ -15,6 +15,7 @@ The primary key for this table is **arn**.
 |account_id|String|
 |region|String|
 |arn (PK)|String|
+|tags|JSON|
 |at_rest_encryption_enabled|Bool|
 |auth_token_enabled|Bool|
 |auth_token_last_modified_date|Timestamp|
diff --git a/plugins/source/aws/resources/services/elasticache/clusters.go b/plugins/source/aws/resources/services/elasticache/clusters.go
diff --git a/plugins/source/aws/resources/services/elasticache/clusters_fetch.go b/plugins/source/aws/resources/services/elasticache/clusters_fetch.go
@@ -5,6 +5,7 @@ import (
 
 	"github.com/aws/aws-sdk-go-v2/aws"
 	"github.com/aws/aws-sdk-go-v2/service/elasticache"
+	"github.com/aws/aws-sdk-go-v2/service/elasticache/types"
 	"github.com/cloudquery/cloudquery/plugins/source/aws/client"
 	"github.com/cloudquery/plugin-sdk/schema"
 )
@@ -23,3 +24,16 @@ func fetchElasticacheClusters(ctx context.Context, meta schema.ClientMeta, paren
 	}
 	return nil
 }
+
+func resolveClusterTags(ctx context.Context, meta schema.ClientMeta, resource *schema.Resource, c schema.Column) error {
+	cluster := resource.Item.(types.CacheCluster)
+
+	svc := meta.(*client.Client).Services().Elasticache
+	response, err := svc.ListTagsForResource(ctx, &elasticache.ListTagsForResourceInput{
+		ResourceName: cluster.ARN,
+	})
+	if err != nil {
+		return err
+	}
+	return resource.Set(c.Name, client.TagsToMap(response.TagList))
+}
diff --git a/plugins/source/aws/resources/services/elasticache/clusters_mock_test.go b/plugins/source/aws/resources/services/elasticache/clusters_mock_test.go
@@ -4,6 +4,7 @@ import (
 	"testing"
 
 	"github.com/aws/aws-sdk-go-v2/service/elasticache"
+	"github.com/aws/aws-sdk-go-v2/service/elasticache/types"
 	"github.com/cloudquery/cloudquery/plugins/source/aws/client"
 	"github.com/cloudquery/cloudquery/plugins/source/aws/client/mocks"
 	"github.com/cloudquery/plugin-sdk/faker"
@@ -19,7 +20,13 @@ func buildElasticacheClusters(t *testing.T, ctrl *gomock.Controller) client.Serv
 		t.Fatal(err)
 	}
 
+	var ta types.Tag
+	if err = faker.FakeObject(&ta); err != nil {
+		t.Fatal(err)
+	}
+
 	mockElasticache.EXPECT().DescribeCacheClusters(gomock.Any(), gomock.Any(), gomock.Any()).Return(&output, nil)
+	mockElasticache.EXPECT().ListTagsForResource(gomock.Any(), &elasticache.ListTagsForResourceInput{ResourceName: output.CacheClusters[0].ARN}, gomock.Any()).Return(&elasticache.ListTagsForResourceOutput{TagList: []types.Tag{ta}}, nil)
 
 	return client.Services{
 		Elasticache: mockElasticache,
