Add DAX Resources for deletion

Author: tomvachon

resources/dax-clusters.go

@@ -0,0 +1,60 @@
+package resources
+
+import (
+	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/aws/session"
+	"github.com/aws/aws-sdk-go/service/dax"
+)
+
+type DAXCluster struct {
+	svc         *dax.DAX
+	clusterName *string
+}
+
+func init() {
+	register("DAXCluster", ListDAXClusters)
+}
+
+func ListDAXClusters(sess *session.Session) ([]Resource, error) {
+	svc := dax.New(sess)
+	resources := []Resource{}
+
+	params := &dax.DescribeClustersInput{
+		MaxResults: aws.Int64(100),
+	}
+
+	for {
+		output, err := svc.DescribeClusters(params)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, cluster := range output.Clusters {
+			resources = append(resources, &DAXCluster{
+				svc:         svc,
+				clusterName: cluster.ClusterName,
+			})
+		}
+
+		if output.NextToken == nil {
+			break
+		}
+
+		params.NextToken = output.NextToken
+	}
+
+	return resources, nil
+}
+
+func (f *DAXCluster) Remove() error {
+
+	_, err := f.svc.DeleteCluster(&dax.DeleteClusterInput{
+		ClusterName: f.clusterName,
+	})
+
+	return err
+}
+
+func (f *DAXCluster) String() string {
+	return *f.clusterName
+}

resources/dax-parametergroups.go

@@ -0,0 +1,65 @@
+package resources
+
+import (
+	"strings"
+
+	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/aws/session"
+	"github.com/aws/aws-sdk-go/service/dax"
+)
+
+type DAXParameterGroup struct {
+	svc                *dax.DAX
+	parameterGroupName *string
+}
+
+func init() {
+	register("DAXParameterGroup", ListDAXParameterGroups)
+}
+
+func ListDAXParameterGroups(sess *session.Session) ([]Resource, error) {
+	svc := dax.New(sess)
+	resources := []Resource{}
+
+	params := &dax.DescribeParameterGroupsInput{
+		MaxResults: aws.Int64(100),
+	}
+
+	for {
+		output, err := svc.DescribeParameterGroups(params)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, parameterGroup := range output.ParameterGroups {
+			//Ensure default is not deleted
+			if !strings.Contains(*parameterGroup.ParameterGroupName, "default") {
+				resources = append(resources, &DAXParameterGroup{
+					svc:                svc,
+					parameterGroupName: parameterGroup.ParameterGroupName,
+				})
+			}
+		}
+
+		if output.NextToken == nil {
+			break
+		}
+
+		params.NextToken = output.NextToken
+	}
+
+	return resources, nil
+}
+
+func (f *DAXParameterGroup) Remove() error {
+
+	_, err := f.svc.DeleteParameterGroup(&dax.DeleteParameterGroupInput{
+		ParameterGroupName: f.parameterGroupName,
+	})
+
+	return err
+}
+
+func (f *DAXParameterGroup) String() string {
+	return *f.parameterGroupName
+}

resources/dax-subnetgroups.go

@@ -0,0 +1,60 @@
+package resources
+
+import (
+	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/aws/session"
+	"github.com/aws/aws-sdk-go/service/dax"
+)
+
+type DAXSubnetGroup struct {
+	svc             *dax.DAX
+	subnetGroupName *string
+}
+
+func init() {
+	register("DAXSubnetGroup", ListDAXSubnetGroups)
+}
+
+func ListDAXSubnetGroups(sess *session.Session) ([]Resource, error) {
+	svc := dax.New(sess)
+	resources := []Resource{}
+
+	params := &dax.DescribeSubnetGroupsInput{
+		MaxResults: aws.Int64(100),
+	}
+
+	for {
+		output, err := svc.DescribeSubnetGroups(params)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, subnet := range output.SubnetGroups {
+			resources = append(resources, &DAXSubnetGroup{
+				svc:             svc,
+				subnetGroupName: subnet.SubnetGroupName,
+			})
+		}
+
+		if output.NextToken == nil {
+			break
+		}
+
+		params.NextToken = output.NextToken
+	}
+
+	return resources, nil
+}
+
+func (f *DAXSubnetGroup) Remove() error {
+
+	_, err := f.svc.DeleteSubnetGroup(&dax.DeleteSubnetGroupInput{
+		SubnetGroupName: f.subnetGroupName,
+	})
+
+	return err
+}
+
+func (f *DAXSubnetGroup) String() string {
+	return *f.subnetGroupName
+}