Add DMS support

Author: tomvachon

resources/databasemigrationservice-certificates.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/databasemigrationservice"
+)
+
+type DatabaseMigrationServiceCertificate struct {
+	svc *databasemigrationservice.DatabaseMigrationService
+	ARN *string
+}
+
+func init() {
+	register("DatabaseMigrationServiceCertificate", ListDatabaseMigrationServiceCertificates)
+}
+
+func ListDatabaseMigrationServiceCertificates(sess *session.Session) ([]Resource, error) {
+	svc := databasemigrationservice.New(sess)
+	resources := []Resource{}
+
+	params := &databasemigrationservice.DescribeCertificatesInput{
+		MaxRecords: aws.Int64(100),
+	}
+
+	for {
+		output, err := svc.DescribeCertificates(params)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, certificate := range output.Certificates {
+			resources = append(resources, &DatabaseMigrationServiceCertificate{
+				svc: svc,
+				ARN: certificate.CertificateArn,
+			})
+		}
+
+		if output.Marker == nil {
+			break
+		}
+
+		params.Marker = output.Marker
+	}
+
+	return resources, nil
+}
+
+func (f *DatabaseMigrationServiceCertificate) Remove() error {
+
+	_, err := f.svc.DeleteEndpoint(&databasemigrationservice.DeleteEndpointInput{
+		EndpointArn: f.ARN,
+	})
+
+	return err
+}
+
+func (f *DatabaseMigrationServiceCertificate) String() string {
+	return *f.ARN
+}

resources/databasemigrationservice-endpoints.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/databasemigrationservice"
+)
+
+type DatabaseMigrationServiceEndpoint struct {
+	svc *databasemigrationservice.DatabaseMigrationService
+	ARN *string
+}
+
+func init() {
+	register("DatabaseMigrationServiceEndpoint", ListDatabaseMigrationServiceEndpoints)
+}
+
+func ListDatabaseMigrationServiceEndpoints(sess *session.Session) ([]Resource, error) {
+	svc := databasemigrationservice.New(sess)
+	resources := []Resource{}
+
+	params := &databasemigrationservice.DescribeEndpointsInput{
+		MaxRecords: aws.Int64(100),
+	}
+
+	for {
+		output, err := svc.DescribeEndpoints(params)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, endpoint := range output.Endpoints {
+			resources = append(resources, &DatabaseMigrationServiceEndpoint{
+				svc: svc,
+				ARN: endpoint.EndpointArn,
+			})
+		}
+
+		if output.Marker == nil {
+			break
+		}
+
+		params.Marker = output.Marker
+	}
+
+	return resources, nil
+}
+
+func (f *DatabaseMigrationServiceEndpoint) Remove() error {
+
+	_, err := f.svc.DeleteEndpoint(&databasemigrationservice.DeleteEndpointInput{
+		EndpointArn: f.ARN,
+	})
+
+	return err
+}
+
+func (f *DatabaseMigrationServiceEndpoint) String() string {
+	return *f.ARN
+}

resources/databasemigrationservice-eventsubscriptions.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/databasemigrationservice"
+)
+
+type DatabaseMigrationServiceEventSubscription struct {
+	svc              *databasemigrationservice.DatabaseMigrationService
+	subscriptionName *string
+}
+
+func init() {
+	register("DatabaseMigrationServiceEventSubscription", ListDatabaseMigrationServiceEventSubscriptions)
+}
+
+func ListDatabaseMigrationServiceEventSubscriptions(sess *session.Session) ([]Resource, error) {
+	svc := databasemigrationservice.New(sess)
+	resources := []Resource{}
+
+	params := &databasemigrationservice.DescribeEventSubscriptionsInput{
+		MaxRecords: aws.Int64(100),
+	}
+
+	for {
+		output, err := svc.DescribeEventSubscriptions(params)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, eventSubscription := range output.EventSubscriptionsList {
+			resources = append(resources, &DatabaseMigrationServiceEventSubscription{
+				svc:              svc,
+				subscriptionName: eventSubscription.CustSubscriptionId,
+			})
+		}
+
+		if output.Marker == nil {
+			break
+		}
+
+		params.Marker = output.Marker
+	}
+
+	return resources, nil
+}
+
+func (f *DatabaseMigrationServiceEventSubscription) Remove() error {
+
+	_, err := f.svc.DeleteEventSubscription(&databasemigrationservice.DeleteEventSubscriptionInput{
+		SubscriptionName: f.subscriptionName,
+	})
+
+	return err
+}
+
+func (f *DatabaseMigrationServiceEventSubscription) String() string {
+	return *f.subscriptionName
+}

resources/databasemigrationservice-replicationinstances.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/databasemigrationservice"
+)
+
+type DatabaseMigrationServiceReplicationInstance struct {
+	svc *databasemigrationservice.DatabaseMigrationService
+	ARN *string
+}
+
+func init() {
+	register("DatabaseMigrationServiceReplicationInstance", ListDatabaseMigrationServiceReplicationInstances)
+}
+
+func ListDatabaseMigrationServiceReplicationInstances(sess *session.Session) ([]Resource, error) {
+	svc := databasemigrationservice.New(sess)
+	resources := []Resource{}
+
+	params := &databasemigrationservice.DescribeReplicationInstancesInput{
+		MaxRecords: aws.Int64(100),
+	}
+
+	for {
+		output, err := svc.DescribeReplicationInstances(params)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, replicationInstance := range output.ReplicationInstances {
+			resources = append(resources, &DatabaseMigrationServiceReplicationInstance{
+				svc: svc,
+				ARN: replicationInstance.ReplicationInstanceArn,
+			})
+		}
+
+		if output.Marker == nil {
+			break
+		}
+
+		params.Marker = output.Marker
+	}
+
+	return resources, nil
+}
+
+func (f *DatabaseMigrationServiceReplicationInstance) Remove() error {
+
+	_, err := f.svc.DeleteReplicationInstance(&databasemigrationservice.DeleteReplicationInstanceInput{
+		ReplicationInstanceArn: f.ARN,
+	})
+
+	return err
+}
+
+func (f *DatabaseMigrationServiceReplicationInstance) String() string {
+	return *f.ARN
+}

resources/databasemigrationservice-replicationtasks.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/databasemigrationservice"
+)
+
+type DatabaseMigrationServiceReplicationTask struct {
+	svc *databasemigrationservice.DatabaseMigrationService
+	ARN *string
+}
+
+func init() {
+	register("DatabaseMigrationServiceReplicationTask", ListDatabaseMigrationServiceReplicationTasks)
+}
+
+func ListDatabaseMigrationServiceReplicationTasks(sess *session.Session) ([]Resource, error) {
+	svc := databasemigrationservice.New(sess)
+	resources := []Resource{}
+
+	params := &databasemigrationservice.DescribeReplicationTasksInput{
+		MaxRecords: aws.Int64(100),
+	}
+
+	for {
+		output, err := svc.DescribeReplicationTasks(params)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, replicationTask := range output.ReplicationTasks {
+			resources = append(resources, &DatabaseMigrationServiceReplicationTask{
+				svc: svc,
+				ARN: replicationTask.ReplicationTaskArn,
+			})
+		}
+
+		if output.Marker == nil {
+			break
+		}
+
+		params.Marker = output.Marker
+	}
+
+	return resources, nil
+}
+
+func (f *DatabaseMigrationServiceReplicationTask) Remove() error {
+
+	_, err := f.svc.DeleteReplicationTask(&databasemigrationservice.DeleteReplicationTaskInput{
+		ReplicationTaskArn: f.ARN,
+	})
+
+	return err
+}
+
+func (f *DatabaseMigrationServiceReplicationTask) String() string {
+	return *f.ARN
+}

resources/databasemigrationservice-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/databasemigrationservice"
+)
+
+type DatabaseMigrationServiceSubnetGroup struct {
+	svc *databasemigrationservice.DatabaseMigrationService
+	ID  *string
+}
+
+func init() {
+	register("DatabaseMigrationServiceSubnetGroup", ListDatabaseMigrationServiceSubnetGroups)
+}
+
+func ListDatabaseMigrationServiceSubnetGroups(sess *session.Session) ([]Resource, error) {
+	svc := databasemigrationservice.New(sess)
+	resources := []Resource{}
+
+	params := &databasemigrationservice.DescribeReplicationSubnetGroupsInput{
+		MaxRecords: aws.Int64(100),
+	}
+
+	for {
+		output, err := svc.DescribeReplicationSubnetGroups(params)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, replicationSubnetGroup := range output.ReplicationSubnetGroups {
+			resources = append(resources, &DatabaseMigrationServiceSubnetGroup{
+				svc: svc,
+				ID:  replicationSubnetGroup.ReplicationSubnetGroupIdentifier,
+			})
+		}
+
+		if output.Marker == nil {
+			break
+		}
+
+		params.Marker = output.Marker
+	}
+
+	return resources, nil
+}
+
+func (f *DatabaseMigrationServiceSubnetGroup) Remove() error {
+
+	_, err := f.svc.DeleteReplicationSubnetGroup(&databasemigrationservice.DeleteReplicationSubnetGroupInput{
+		ReplicationSubnetGroupIdentifier: f.ID,
+	})
+
+	return err
+}
+
+func (f *DatabaseMigrationServiceSubnetGroup) String() string {
+	return *f.ID
+}