Merge pull request #36368 from hunttom/f-aws_timestreamwrite_database…

…-new-datasource

New Datasource for AWS Timestream Database

.changelog/36368.txt

@@ -0,0 +1,3 @@
+```release-note:new-data-source
+aws_timestreamwrite_database
+```

internal/service/timestreamwrite/database_data_source.go

@@ -0,0 +1,105 @@
+// Copyright (c) HashiCorp, Inc.
+// SPDX-License-Identifier: MPL-2.0
+
+package timestreamwrite
+
+import (
+ "context"
+
+ "github.com/aws/aws-sdk-go-v2/service/timestreamwrite"
+ "github.com/hashicorp/terraform-plugin-framework-timetypes/timetypes"
+ "github.com/hashicorp/terraform-plugin-framework-validators/stringvalidator"
+ "github.com/hashicorp/terraform-plugin-framework/datasource"
+ "github.com/hashicorp/terraform-plugin-framework/datasource/schema"
+ "github.com/hashicorp/terraform-plugin-framework/schema/validator"
+ "github.com/hashicorp/terraform-plugin-framework/types"
+ "github.com/hashicorp/terraform-provider-aws/internal/create"
+ "github.com/hashicorp/terraform-provider-aws/internal/framework"
+ "github.com/hashicorp/terraform-provider-aws/internal/framework/flex"
+ "github.com/hashicorp/terraform-provider-aws/names"
+)
+
+// @FrameworkDataSource(name="Database")
+func newDataSourceDatabase(context.Context) (datasource.DataSourceWithConfigure, error) {
+ return &dataSourceDatabase{}, nil
+}
+
+type dataSourceDatabase struct {
+ framework.DataSourceWithConfigure
+}
+
+const (
+ DSNameDatabase = "Database data source"
+)
+
+func (d *dataSourceDatabase) Metadata(_ context.Context, request datasource.MetadataRequest, response *datasource.MetadataResponse) {
+ response.TypeName = "aws_timestreamwrite_database"
+}
+
+func (d *dataSourceDatabase) Schema(ctx context.Context, request datasource.SchemaRequest, response *datasource.SchemaResponse) {
+ response.Schema = schema.Schema{
+ Attributes: map[string]schema.Attribute{
+ names.AttrARN: schema.StringAttribute{
+ Computed: true,
+ Validators: []validator.String{
+ stringvalidator.LengthBetween(3, 256),
+ },
+ },
+ names.AttrName: schema.StringAttribute{
+ Required: true,
+ },
+ names.AttrCreatedTime: schema.StringAttribute{
+ CustomType: timetypes.RFC3339Type{},
+ Computed: true,
+ },
+ names.AttrKMSKeyID: schema.StringAttribute{
+ Computed: true,
+ },
+ "table_count": schema.Int64Attribute{
+ Computed: true,
+ },
+ names.AttrLastUpdatedTime: schema.StringAttribute{
+ CustomType: timetypes.RFC3339Type{},
+ Computed: true,
+ },
+ },
+ }
+}
+
+func (d *dataSourceDatabase) Read(ctx context.Context, req datasource.ReadRequest, resp *datasource.ReadResponse) {
+ conn := d.Meta().TimestreamWriteClient(ctx)
+ var data dsDescribeDatabase
+
+ resp.Diagnostics.Append(req.Config.Get(ctx, &data)...)
+ if resp.Diagnostics.HasError() {
+ return
+ }
+ in := &timestreamwrite.DescribeDatabaseInput{
+ DatabaseName: data.Name.ValueStringPointer(),
+ }
+
+ desc, err := conn.DescribeDatabase(ctx, in)
+
+ if err != nil {
+ resp.Diagnostics.AddError(
+ create.ProblemStandardMessage(names.Transfer, create.ErrActionReading, DSNameDatabase, data.Name.String(), err),
+ err.Error(),
+ )
+ return
+ }
+
+ resp.Diagnostics.Append(flex.Flatten(ctx, desc.Database, &data)...)
+ if resp.Diagnostics.HasError() {
+ return
+ }
+ resp.Diagnostics.Append(resp.State.Set(ctx, &data)...)
+}
+
+type dsDescribeDatabase struct {
+ ARN types.String `tfsdk:"arn"`
+ CreatedTime timetypes.RFC3339 `tfsdk:"created_time"`
+ Name types.String `tfsdk:"name"`
+ KmsKeyID types.String `tfsdk:"kms_key_id"`
+ LastUpdatedTime timetypes.RFC3339 `tfsdk:"last_updated_time"`
+ TableCount types.Int64 `tfsdk:"table_count"`
+}

internal/service/timestreamwrite/database_data_source_test.go

@@ -0,0 +1,157 @@
+// Copyright (c) HashiCorp, Inc.
+// SPDX-License-Identifier: MPL-2.0
+
+package timestreamwrite_test
+
+import (
+ "fmt"
+ "testing"
+
+ "github.com/YakDriver/regexache"
+ sdkacctest "github.com/hashicorp/terraform-plugin-testing/helper/acctest"
+ "github.com/hashicorp/terraform-plugin-testing/helper/resource"
+ "github.com/hashicorp/terraform-provider-aws/internal/acctest"
+ "github.com/hashicorp/terraform-provider-aws/names"
+)
+
+func TestAccTimestreamWriteDatabaseDataSource_basic(t *testing.T) {
+ ctx := acctest.Context(t)
+
+ if testing.Short() {
+ t.Skip("skipping long-running test in short mode")
+ }
+
+ rDatabaseName := sdkacctest.RandomWithPrefix(acctest.ResourcePrefix)
+ resourceName := "aws_timestreamwrite_database.test"
+ dataSourceName := "data.aws_timestreamwrite_database.test"
+
+ resource.ParallelTest(t, resource.TestCase{
+ PreCheck: func() { acctest.PreCheck(ctx, t); testAccPreCheck(ctx, t) },
+ ErrorCheck: acctest.ErrorCheck(t, names.TimestreamWriteServiceID),
+ ProtoV5ProviderFactories: acctest.ProtoV5ProviderFactories,
+ CheckDestroy: testAccCheckDatabaseDestroy(ctx),
+ Steps: []resource.TestStep{
+ {
+ Config: testAccDatabaseDataSourceConfig_basic(rDatabaseName),
+ Check: resource.ComposeTestCheckFunc(
+ testAccCheckDatabaseExists(ctx, resourceName),
+ resource.TestCheckResourceAttrPair(dataSourceName, names.AttrARN, resourceName, names.AttrARN),
+ resource.TestCheckResourceAttrPair(dataSourceName, names.AttrName, resourceName, names.AttrDatabaseName),
+ resource.TestCheckResourceAttrPair(dataSourceName, names.AttrCreatedTime, resourceName, names.AttrCreatedTime),
+ resource.TestCheckResourceAttrPair(dataSourceName, names.AttrKMSKeyID, resourceName, names.AttrKMSKeyID),
+ resource.TestCheckResourceAttrPair(dataSourceName, "table_count", resourceName, "table_count"),
+ ),
+ },
+ },
+ })
+}
+
+func TestAccTimestreamWriteDatabaseDataSource_kmsKey(t *testing.T) {
+ ctx := acctest.Context(t)
+ rDatabaseName := sdkacctest.RandomWithPrefix(acctest.ResourcePrefix)
+ rKmsKeyName := sdkacctest.RandomWithPrefix(acctest.ResourcePrefix)
+ dataSourceName := "data.aws_timestreamwrite_database.test"
+ kmsResourceName := "aws_kms_key.test"
+ resourceName := "aws_timestreamwrite_database.test"
+
+ resource.ParallelTest(t, resource.TestCase{
+ PreCheck: func() { acctest.PreCheck(ctx, t); testAccPreCheck(ctx, t) },
+ ErrorCheck: acctest.ErrorCheck(t, names.TimestreamWriteServiceID),
+ ProtoV5ProviderFactories: acctest.ProtoV5ProviderFactories,
+ CheckDestroy: testAccCheckDatabaseDestroy(ctx),
+ Steps: []resource.TestStep{
+ {
+ Config: testAccDatabaseDataSourceConfig_kmsKey(rDatabaseName, rKmsKeyName),
+ Check: resource.ComposeTestCheckFunc(
+ testAccCheckDatabaseExists(ctx, resourceName),
+ resource.TestCheckResourceAttrPair(dataSourceName, names.AttrName, resourceName, names.AttrDatabaseName),
+ resource.TestCheckResourceAttrPair(dataSourceName, names.AttrKMSKeyID, kmsResourceName, names.AttrARN),
+ ),
+ },
+ },
+ })
+}
+
+func TestAccTimestreamWriteDatabaseDataSource_updateKMSKey(t *testing.T) {
+ ctx := acctest.Context(t)
+ rDatabaseName := sdkacctest.RandomWithPrefix(acctest.ResourcePrefix)
+ rKmsKeyName := sdkacctest.RandomWithPrefix(acctest.ResourcePrefix)
+ resourceName := "aws_timestreamwrite_database.test"
+ dataSourceName := "data.aws_timestreamwrite_database.test"
+ kmsResourceName := "aws_kms_key.test"
+
+ resource.ParallelTest(t, resource.TestCase{
+ PreCheck: func() { acctest.PreCheck(ctx, t); testAccPreCheck(ctx, t) },
+ ErrorCheck: acctest.ErrorCheck(t, names.TimestreamWriteServiceID),
+ ProtoV5ProviderFactories: acctest.ProtoV5ProviderFactories,
+ CheckDestroy: testAccCheckDatabaseDestroy(ctx),
+ Steps: []resource.TestStep{
+ {
+ Config: testAccDatabaseDataSourceConfig_basic(rDatabaseName),
+ Check: resource.ComposeTestCheckFunc(
+ testAccCheckDatabaseExists(ctx, resourceName),
+ acctest.MatchResourceAttrRegionalARN(dataSourceName, names.AttrKMSKeyID, "kms", regexache.MustCompile(`key/.+`)),
+ ),
+ },
+ {
+ Config: testAccDatabaseDataSourceConfig_kmsKey(rDatabaseName, rKmsKeyName),
+ Check: resource.ComposeTestCheckFunc(
+ testAccCheckDatabaseExists(ctx, resourceName),
+ resource.TestCheckResourceAttrPair(dataSourceName, names.AttrKMSKeyID, kmsResourceName, names.AttrARN),
+ ),
+ },
+ {
+ Config: testAccDatabaseDataSourceConfig_basic(rDatabaseName),
+ Check: resource.ComposeTestCheckFunc(
+ testAccCheckDatabaseExists(ctx, resourceName),
+ acctest.MatchResourceAttrRegionalARN(dataSourceName, names.AttrKMSKeyID, "kms", regexache.MustCompile(`key/.+`)),
+ ),
+ },
+ },
+ })
+}
+
+func testAccDatabaseDataSourceConfig_basic(rDatabaseName string) string {
+ return fmt.Sprintf(`
+resource "aws_timestreamwrite_database" "test" {
+ database_name = %[1]q
+}
+
+data "aws_timestreamwrite_database" "test" {
+ name = aws_timestreamwrite_database.test.database_name
+}
+`, rDatabaseName)
+}
+
+func testAccDatabaseDataSourceConfig_kmsKey(rDatabaseName, rKmsKeyName string) string {
+ return fmt.Sprintf(`
+resource "aws_kms_key" "test" {
+ description = %[1]q
+
+ policy = <<POLICY
+{
+ "Version": "2012-10-17",
+ "Statement": [
+ {
+ "Effect": "Allow",
+ "Principal": {
+ "AWS": "*"
+ },
+ "Action": "kms:*",
+ "Resource": "*"
+ }
+ ]
+}
+POLICY
+}
+
+resource "aws_timestreamwrite_database" "test" {
+ database_name = %[2]q
+ kms_key_id = aws_kms_key.test.arn
+}
+
+data "aws_timestreamwrite_database" "test" {
+ name = aws_timestreamwrite_database.test.database_name
+}
+`, rKmsKeyName, rDatabaseName)
+}

website/docs/d/timestreamwrite_database.html.markdown

@@ -0,0 +1,38 @@
+---
+subcategory: "Timestream Write"
+layout: "aws"
+page_title: "AWS: aws_timestreamwrite_database"
+description: |-
+ Terraform data source for managing an AWS Timestream Write Database.
+---
+
+# Data Source: aws_timestreamwrite_database
+
+Terraform data source for managing an AWS Timestream Write Database.
+
+## Example Usage
+
+### Basic Usage
+
+```terraform
+data "aws_timestreamwrite_database" "test" {
+ name = "database-example"
+}
+```
+
+## Argument Reference
+
+The following arguments are required:
+
+* `database_name` – (Required) The name of the Timestream database. Minimum length of 3. Maximum length of 256.
+
+## Attribute Reference
+
+This data source exports the following attributes in addition to the arguments above:
+
+* `arn` - The ARN that uniquely identifies this database.
+* `created_time` - Creation time of database.
+* `database_name` – (Required) The name of the Timestream database. Minimum length of 3. Maximum length of 256.
+* `kms_key_id` - The ARN of the KMS key used to encrypt the data stored in the database.
+* `last_updated_time` - Last time database was updated.
+* `table_count` - Total number of tables in the Timestream database.