Adding support for AWS DynamoDB Table for StreamSpecifications
This commit is contained in:
parent
d69abbaeda
commit
d46348c233
|
@ -12,6 +12,7 @@ import (
|
|||
"github.com/aws/aws-sdk-go/aws/awserr"
|
||||
"github.com/aws/aws-sdk-go/service/dynamodb"
|
||||
"github.com/hashicorp/terraform/helper/hashcode"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// Number of times to retry if a throttling-related exception occurs
|
||||
|
@ -158,6 +159,20 @@ func resourceAwsDynamoDbTable() *schema.Resource {
|
|||
return hashcode.String(buf.String())
|
||||
},
|
||||
},
|
||||
"stream_enabled": &schema.Schema{
|
||||
Type: schema.TypeBool,
|
||||
Optional: true,
|
||||
Computed: true,
|
||||
},
|
||||
"stream_view_type": &schema.Schema{
|
||||
Type: schema.TypeString,
|
||||
Optional: true,
|
||||
Computed: true,
|
||||
StateFunc: func(v interface{}) string {
|
||||
value := v.(string)
|
||||
return strings.ToUpper(value)
|
||||
},
|
||||
},
|
||||
},
|
||||
}
|
||||
}
|
||||
|
@ -263,6 +278,16 @@ func resourceAwsDynamoDbTableCreate(d *schema.ResourceData, meta interface{}) er
|
|||
req.GlobalSecondaryIndexes = globalSecondaryIndexes
|
||||
}
|
||||
|
||||
if _, ok := d.GetOk("stream_enabled"); ok {
|
||||
|
||||
req.StreamSpecification = &dynamodb.StreamSpecification{
|
||||
StreamEnabled: aws.Bool(d.Get("stream_enabled").(bool)),
|
||||
StreamViewType: aws.String(d.Get("stream_view_type").(string)),
|
||||
}
|
||||
|
||||
fmt.Printf("[DEBUG] Adding StreamSpecifications to the table")
|
||||
}
|
||||
|
||||
attemptCount := 1
|
||||
for attemptCount <= DYNAMODB_MAX_THROTTLE_RETRIES {
|
||||
output, err := dynamodbconn.CreateTable(req)
|
||||
|
@ -340,6 +365,25 @@ func resourceAwsDynamoDbTableUpdate(d *schema.ResourceData, meta interface{}) er
|
|||
waitForTableToBeActive(d.Id(), meta)
|
||||
}
|
||||
|
||||
if d.HasChange("stream_enabled") || d.HasChange("stream_view_type") {
|
||||
req := &dynamodb.UpdateTableInput{
|
||||
TableName: aws.String(d.Id()),
|
||||
}
|
||||
|
||||
req.StreamSpecification = &dynamodb.StreamSpecification{
|
||||
StreamEnabled: aws.Bool(d.Get("stream_enabled").(bool)),
|
||||
StreamViewType: aws.String(d.Get("stream_view_type").(string)),
|
||||
}
|
||||
|
||||
_, err := dynamodbconn.UpdateTable(req)
|
||||
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
waitForTableToBeActive(d.Id(), meta)
|
||||
}
|
||||
|
||||
if d.HasChange("global_secondary_index") {
|
||||
log.Printf("[DEBUG] Changed GSI data")
|
||||
req := &dynamodb.UpdateTableInput{
|
||||
|
@ -587,6 +631,11 @@ func resourceAwsDynamoDbTableRead(d *schema.ResourceData, meta interface{}) erro
|
|||
log.Printf("[DEBUG] Added GSI: %s - Read: %d / Write: %d", gsi["name"], gsi["read_capacity"], gsi["write_capacity"])
|
||||
}
|
||||
|
||||
if table.StreamSpecification != nil {
|
||||
d.Set("stream_view_type", table.StreamSpecification.StreamViewType)
|
||||
d.Set("stream_enabled", table.StreamSpecification.StreamEnabled)
|
||||
}
|
||||
|
||||
err = d.Set("global_secondary_index", gsiList)
|
||||
if err != nil {
|
||||
return err
|
||||
|
|
|
@ -33,6 +33,26 @@ func TestAccAWSDynamoDbTable(t *testing.T) {
|
|||
})
|
||||
}
|
||||
|
||||
func TestAccAWSDynamoDbTable_streamSpecification(t *testing.T) {
|
||||
resource.Test(t, resource.TestCase{
|
||||
PreCheck: func() { testAccPreCheck(t) },
|
||||
Providers: testAccProviders,
|
||||
CheckDestroy: testAccCheckAWSDynamoDbTableDestroy,
|
||||
Steps: []resource.TestStep{
|
||||
resource.TestStep{
|
||||
Config: testAccAWSDynamoDbConfigStreamSpecification,
|
||||
Check: resource.ComposeTestCheckFunc(
|
||||
testAccCheckInitialAWSDynamoDbTableExists("aws_dynamodb_table.basic-dynamodb-table"),
|
||||
resource.TestCheckResourceAttr(
|
||||
"aws_dynamodb_table.basic-dynamodb-table", "stream_enabled", "true"),
|
||||
resource.TestCheckResourceAttr(
|
||||
"aws_dynamodb_table.basic-dynamodb-table", "stream_view_type", "KEYS_ONLY"),
|
||||
),
|
||||
},
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
func testAccCheckAWSDynamoDbTableDestroy(s *terraform.State) error {
|
||||
conn := testAccProvider.Meta().(*AWSClient).dynamodbconn
|
||||
|
||||
|
@ -295,3 +315,44 @@ resource "aws_dynamodb_table" "basic-dynamodb-table" {
|
|||
}
|
||||
}
|
||||
`
|
||||
|
||||
const testAccAWSDynamoDbConfigStreamSpecification = `
|
||||
resource "aws_dynamodb_table" "basic-dynamodb-table" {
|
||||
name = "TerraformTestStreamTable"
|
||||
read_capacity = 10
|
||||
write_capacity = 20
|
||||
hash_key = "TestTableHashKey"
|
||||
range_key = "TestTableRangeKey"
|
||||
attribute {
|
||||
name = "TestTableHashKey"
|
||||
type = "S"
|
||||
}
|
||||
attribute {
|
||||
name = "TestTableRangeKey"
|
||||
type = "S"
|
||||
}
|
||||
attribute {
|
||||
name = "TestLSIRangeKey"
|
||||
type = "N"
|
||||
}
|
||||
attribute {
|
||||
name = "TestGSIRangeKey"
|
||||
type = "S"
|
||||
}
|
||||
local_secondary_index {
|
||||
name = "TestTableLSI"
|
||||
range_key = "TestLSIRangeKey"
|
||||
projection_type = "ALL"
|
||||
}
|
||||
global_secondary_index {
|
||||
name = "InitialTestTableGSI"
|
||||
hash_key = "TestTableHashKey"
|
||||
range_key = "TestGSIRangeKey"
|
||||
write_capacity = 10
|
||||
read_capacity = 10
|
||||
projection_type = "KEYS_ONLY"
|
||||
}
|
||||
stream_enabled = true
|
||||
stream_view_type = "KEYS_ONLY"
|
||||
}
|
||||
`
|
||||
|
|
|
@ -84,6 +84,8 @@ parameter.
|
|||
* `non_key_attributes` - (Optional) Only required with *INCLUDE* as a
|
||||
projection type; a list of attributes to project into the index. These
|
||||
do not need to be defined as attributes on the table.
|
||||
* `stream_enabled` - (Optional) Indicates whether Streams is to be enabled (true) or disabled (false).
|
||||
* `stream_view_type` - (Optional) When an item in the table is modified, StreamViewType determines what information is written to the table's stream. Valid values are KEYS_ONLY, NEW_IMAGE, OLD_IMAGE, NEW_AND_OLD_IMAGES.
|
||||
|
||||
For `global_secondary_index` objects only, you need to specify
|
||||
`write_capacity` and `read_capacity` in the same way you would for the
|
||||
|
|
Loading…
Reference in New Issue