Add Ability to Lookup AMP Workspace by Alias

.changelog/28568.txt

@@ -0,0 +1,3 @@
+```release-note:enhancement
+datasource/aws_prometheus_workspace: Add ability to query by workspace alias
+```

internal/service/amp/find.go

@@ -101,6 +101,43 @@ func FindWorkspaceByID(ctx context.Context, conn *prometheusservice.PrometheusSe
 	return output.Workspace, nil
 }
 
+func FindWorkspaceByAlias(ctx context.Context, conn *prometheusservice.PrometheusService, alias string) (*prometheusservice.WorkspaceDescription, error) {
+	input := &prometheusservice.ListWorkspacesInput{
+		Alias: aws.String(alias),
+	}
+
+	var result *prometheusservice.WorkspaceSummary
+	err := conn.ListWorkspacesPages(input, func(page *prometheusservice.ListWorkspacesOutput, lastPage bool) bool {
+		if page == nil {
+			return !lastPage
+		}
+
+		for _, w := range page.Workspaces {
+			if w == nil {
+				continue
+			}
+
+			// ListWorkspaces returns all workspaces that begin with the given alias, so
+			// we need to return the result that matches the alias exactly.
+			if aws.StringValue(w.Alias) == alias {
+				result = w
+				return false
+			}
+		}
+		return !lastPage
+	})
+
+	if err != nil {
+		return nil, err
+	}
+	if result == nil {
+		return nil, tfresource.NewEmptyResultError(nil)
+	}
+
+	// Call FindWorkspaceByID so that we can return a WorkspaceDescription instead of a WorkspaceSummary.
+	return FindWorkspaceByID(ctx, conn, aws.StringValue(result.WorkspaceId))
+}
+
 func FindLoggingConfigurationByWorkspaceID(ctx context.Context, conn *prometheusservice.PrometheusService, id string) (*prometheusservice.LoggingConfigurationMetadata, error) {
 	input := &prometheusservice.DescribeLoggingConfigurationInput{
 		WorkspaceId: aws.String(id),

internal/service/amp/workspace_data_source.go

@@ -4,6 +4,8 @@ import (
 	"context"
 	"time"
 
+	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/service/prometheusservice"
 	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
 	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
 	"github.com/hashicorp/terraform-provider-aws/internal/conns"
@@ -16,8 +18,10 @@ func DataSourceWorkspace() *schema.Resource {
 
 		Schema: map[string]*schema.Schema{
 			"alias": {
-				Type:     schema.TypeString,
-				Computed: true,
+				Type:          schema.TypeString,
+				Optional:      true,
+				Computed:      true,
+				ConflictsWith: []string{"workspace_id"},
 			},
 			"arn": {
 				Type:     schema.TypeString,
@@ -37,8 +41,10 @@ func DataSourceWorkspace() *schema.Resource {
 			},
 			"tags": tftags.TagsSchemaComputed(),
 			"workspace_id": {
-				Type:     schema.TypeString,
-				Required: true,
+				Type:          schema.TypeString,
+				Optional:      true,
+				Computed:      true,
+				ConflictsWith: []string{"alias"},
 			},
 		},
 	}
@@ -49,13 +55,26 @@ func dataSourceWorkspaceRead(ctx context.Context, d *schema.ResourceData, meta i
 	ignoreTagsConfig := meta.(*conns.AWSClient).IgnoreTagsConfig
 
 	workspaceID := d.Get("workspace_id").(string)
-	workspace, err := FindWorkspaceByID(ctx, conn, workspaceID)
+	workspaceAlias := d.Get("alias").(string)
+
+	var workspace *prometheusservice.WorkspaceDescription
+	var err error
+
+	if workspaceID != "" {
+		workspace, err = FindWorkspaceByID(ctx, conn, workspaceID)
+	} else if workspaceAlias != "" {
+		workspace, err = FindWorkspaceByAlias(ctx, conn, workspaceAlias)
+	}
 
 	if err != nil {
-		return diag.Errorf("reading AMP Workspace (%s): %s", workspaceID, err)
+		if workspaceID != "" {
+			return diag.Errorf("reading AMP Workspace by ID (%s): %s", workspaceID, err)
+		} else {
+			return diag.Errorf("reading AMP Workspace by alias (%s): %s", workspaceAlias, err)
+		}
 	}
 
-	d.SetId(workspaceID)
+	d.SetId(aws.StringValue(workspace.WorkspaceId))
 
 	d.Set("alias", workspace.Alias)
 	d.Set("arn", workspace.Arn)

internal/service/amp/workspace_data_source_test.go

@@ -10,7 +10,33 @@ import (
 	"github.com/hashicorp/terraform-provider-aws/internal/acctest"
 )
 
-func TestAccAMPWorkspaceDataSource_basic(t *testing.T) {
+func TestAccAMPWorkspaceDataSource_workspace_id(t *testing.T) {
+	rName := sdkacctest.RandomWithPrefix(acctest.ResourcePrefix)
+	resourceName := "aws_prometheus_workspace.test"
+	dataSourceName := "data.aws_prometheus_workspace.test"
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:                 func() { acctest.PreCheck(t); acctest.PreCheckPartitionHasService(prometheusservice.EndpointsID, t) },
+		ErrorCheck:               acctest.ErrorCheck(t, prometheusservice.EndpointsID),
+		CheckDestroy:             nil,
+		ProtoV5ProviderFactories: acctest.ProtoV5ProviderFactories,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccWorkspaceDataSourceConfig_workspace_id(rName),
+				Check: resource.ComposeTestCheckFunc(
+					resource.TestCheckResourceAttrPair(resourceName, "alias", dataSourceName, "alias"),
+					resource.TestCheckResourceAttrPair(resourceName, "arn", dataSourceName, "arn"),
+					resource.TestCheckResourceAttrSet(dataSourceName, "created_date"),
+					resource.TestCheckResourceAttrPair(resourceName, "prometheus_endpoint", dataSourceName, "prometheus_endpoint"),
+					resource.TestCheckResourceAttrSet(dataSourceName, "status"),
+					resource.TestCheckResourceAttrPair(resourceName, "tags.%", dataSourceName, "tags.%"),
+				),
+			},
+		},
+	})
+}
+
+func TestAccAMPWorkspaceDataSource_alias(t *testing.T) {
 	rName := sdkacctest.RandomWithPrefix(acctest.ResourcePrefix)
 	resourceName := "aws_prometheus_workspace.test"
 	dataSourceName := "data.aws_prometheus_workspace.test"
@@ -36,14 +62,28 @@ func TestAccAMPWorkspaceDataSource_basic(t *testing.T) {
 	})
 }
 
-func testAccWorkspaceDataSourceConfig_alias(rName string) string {
+func testAccWorkspaceBaseDataSourceConfig(rName string) string {
 	return fmt.Sprintf(`
 resource "aws_prometheus_workspace" "test" {
   alias = %[1]q
 }
+`, rName)
+}
 
+func testAccWorkspaceDataSourceConfig_workspace_id(rName string) string {
+	return acctest.ConfigCompose(
+		testAccWorkspaceBaseDataSourceConfig(rName), `
 data "aws_prometheus_workspace" "test" {
   workspace_id = aws_prometheus_workspace.test.id
 }
-`, rName)
+`)
+}
+
+func testAccWorkspaceDataSourceConfig_alias(rName string) string {
+	return acctest.ConfigCompose(
+		testAccWorkspaceBaseDataSourceConfig(rName), `
+data "aws_prometheus_workspace" "test" {
+  alias = aws_prometheus_workspace.test.alias
+}
+`)
 }