initial commit

.gitignore

@@ -0,0 +1,2 @@
+*~
+bin

LICENSE

@@ -0,0 +1,27 @@
+Copyright (c) 2019, Aaron Straup Cope
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+* Redistributions of source code must retain the above copyright notice, this
+  list of conditions and the following disclaimer.
+
+* Redistributions in binary form must reproduce the above copyright notice,
+  this list of conditions and the following disclaimer in the documentation
+  and/or other materials provided with the distribution.
+
+* Neither the name of the {organization} nor the names of its
+  contributors may be used to endorse or promote products derived from
+  this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

README.md

@@ -0,0 +1,9 @@
+# go-aws-dynamodb
+
+Go package providing convenience methods for working with aws-sdk-go/service/dynamodb instances.
+
+## Documentation
+
+[![Go Reference](https://pkg.go.dev/badge/github.com/aaronland/go-aws-dynamodb.svg)](https://pkg.go.dev/github.com/aaronland/go-aws-dynamodb)
+
+Documentation is incomplete.

doc.go

@@ -0,0 +1,2 @@
+// Package dynamodb provides convenience methods for working with aws-sdk-go/service/dynamodb instances.
+package dynamodb

go.mod

@@ -0,0 +1,5 @@
+module github.com/aaronland/go-aws-dynamodb
+
+go 1.12
+
+require github.com/aws/aws-sdk-go v1.38.53

go.sum

@@ -0,0 +1,21 @@
+github.com/aws/aws-sdk-go v1.38.53 h1:Qj5OvKPrDGTiCnWj+kwQXAlBO6OaFBH/WaRzJPZPg3w=
+github.com/aws/aws-sdk-go v1.38.53/go.mod h1:hcU610XS61/+aQV88ixoOzUoG7v3b31pl2zKMmprdro=
+github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/jmespath/go-jmespath v0.4.0 h1:BEgLn5cpjn8UN1mAw4NjwDrS35OdebyEtFe+9YPoQUg=
+github.com/jmespath/go-jmespath v0.4.0/go.mod h1:T8mJZnbsbmF+m6zOOFylbeCJqk5+pHWvzYPziyZiYoo=
+github.com/jmespath/go-jmespath/internal/testify v1.5.1/go.mod h1:L3OGu8Wl2/fWfCI6z80xFu9LTZmf1ZRjMHUOPmWr69U=
+github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
+golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
+golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
+golang.org/x/net v0.0.0-20201110031124-69a78807bb2b/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU=
+golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
+golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20200930185726-fdedc70b468f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
+golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
+golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v2 v2.2.8/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=

tables.go

@@ -0,0 +1,103 @@
+package dynamodb
+
+import (
+	"fmt"
+	"github.com/aws/aws-sdk-go/aws"
+	aws_dynamodb "github.com/aws/aws-sdk-go/service/dynamodb"
+)
+
+// CreateTablesOptions defines options for the CreateTables method
+type CreateTablesOptions struct {
+	// A hash map containing table names and their dynamodb.CreateTableInput defintions
+	Tables map[string]*aws_dynamodb.CreateTableInput
+	// If true and the table already exists, delete and recreate the table
+	Refresh bool
+}
+
+// Create one or more tables associated with the dynamodb.DynamoDB instance.
+func CreateTables(client *aws_dynamodb.DynamoDB, opts *CreateTablesOptions) error {
+
+	for table_name, def := range opts.Tables {
+
+		has_table, err := HasTable(client, table_name)
+
+		if err != nil {
+			return fmt.Errorf("Failed to determined whether table exists, %w", err)
+		}
+
+		if has_table {
+
+			if !opts.Refresh {
+				continue
+			}
+
+			req := &aws_dynamodb.DeleteTableInput{
+				TableName: aws.String(table_name),
+			}
+
+			client.DeleteTable(req)
+		}
+
+		def.TableName = aws.String(table_name)
+
+		_, err = client.CreateTable(def)
+
+		if err != nil {
+			return fmt.Errorf("Failed to create table '%s', %w", table_name, err)
+		}
+	}
+
+	return nil
+}
+
+// Return a boolean value indication whether or not the dynamodb.DynamoDB instances contains a table matching table_name.
+func HasTable(client *aws_dynamodb.DynamoDB, table_name string) (bool, error) {
+
+	tables, err := ListTables(client)
+
+	if err != nil {
+		return false, err
+	}
+
+	has_table := false
+
+	for _, name := range tables {
+
+		if name == table_name {
+			has_table = true
+			break
+		}
+	}
+
+	return has_table, nil
+}
+
+// Return the list of table names associated with the dynamodb.DynamoDB instance.
+func ListTables(client *aws_dynamodb.DynamoDB) ([]string, error) {
+
+	tables := make([]string, 0)
+
+	input := &aws_dynamodb.ListTablesInput{}
+
+	for {
+
+		rsp, err := client.ListTables(input)
+
+		if err != nil {
+			return nil, err
+		}
+
+		for _, n := range rsp.TableNames {
+			tables = append(tables, *n)
+		}
+
+		input.ExclusiveStartTableName = rsp.LastEvaluatedTableName
+
+		if rsp.LastEvaluatedTableName == nil {
+			break
+		}
+
+	}
+
+	return tables, nil
+}