Add Batch Service

resources/batch-computeenvironments.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/batch"
+)
+
+type BatchComputeEnvironment struct {
+	svc                    *batch.Batch
+	computeEnvironmentName *string
+}
+
+func init() {
+	register("BatchComputeEnvironment", ListBatchComputeEnvironments)
+}
+
+func ListBatchComputeEnvironments(sess *session.Session) ([]Resource, error) {
+	svc := batch.New(sess)
+	resources := []Resource{}
+
+	params := &batch.DescribeComputeEnvironmentsInput{
+		MaxResults: aws.Int64(100),
+	}
+
+	for {
+		output, err := svc.DescribeComputeEnvironments(params)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, computeEnvironment := range output.ComputeEnvironments {
+			resources = append(resources, &BatchComputeEnvironment{
+				svc: svc,
+				computeEnvironmentName: computeEnvironment.ComputeEnvironmentName,
+			})
+		}
+
+		if output.NextToken == nil {
+			break
+		}
+
+		params.NextToken = output.NextToken
+	}
+
+	return resources, nil
+}
+
+func (f *BatchComputeEnvironment) Remove() error {
+
+	_, err := f.svc.DeleteComputeEnvironment(&batch.DeleteComputeEnvironmentInput{
+		ComputeEnvironment: f.computeEnvironmentName,
+	})
+
+	return err
+}
+
+func (f *BatchComputeEnvironment) String() string {
+	return *f.computeEnvironmentName
+}

resources/batch-computeenvironmentstates.go

@@ -0,0 +1,73 @@
+package resources
+
+import (
+	"fmt"
+	"strings"
+
+	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/aws/session"
+	"github.com/aws/aws-sdk-go/service/batch"
+)
+
+type BatchComputeEnvironmentState struct {
+	svc                    *batch.Batch
+	computeEnvironmentName *string
+	state                  *string
+}
+
+func init() {
+	register("BatchComputeEnvironmentState", ListBatchComputeEnvironmentStates)
+}
+
+func ListBatchComputeEnvironmentStates(sess *session.Session) ([]Resource, error) {
+	svc := batch.New(sess)
+	resources := []Resource{}
+
+	params := &batch.DescribeComputeEnvironmentsInput{
+		MaxResults: aws.Int64(100),
+	}
+
+	for {
+		output, err := svc.DescribeComputeEnvironments(params)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, computeEnvironment := range output.ComputeEnvironments {
+			resources = append(resources, &BatchComputeEnvironmentState{
+				svc: svc,
+				computeEnvironmentName: computeEnvironment.ComputeEnvironmentName,
+				state: computeEnvironment.State,
+			})
+		}
+
+		if output.NextToken == nil {
+			break
+		}
+
+		params.NextToken = output.NextToken
+	}
+
+	return resources, nil
+}
+
+func (f *BatchComputeEnvironmentState) Remove() error {
+
+	_, err := f.svc.UpdateComputeEnvironment(&batch.UpdateComputeEnvironmentInput{
+		ComputeEnvironment: f.computeEnvironmentName,
+		State:              aws.String("DISABLED"),
+	})
+
+	return err
+}
+
+func (f *BatchComputeEnvironmentState) String() string {
+	return *f.computeEnvironmentName
+}
+
+func (f *BatchComputeEnvironmentState) Filter() error {
+	if strings.ToLower(*f.state) == "disabled" {
+		return fmt.Errorf("already disabled")
+	}
+	return nil
+}

resources/batch-jobqueues.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/batch"
+)
+
+type BatchJobQueue struct {
+	svc      *batch.Batch
+	jobQueue *string
+}
+
+func init() {
+	register("BatchJobQueue", ListBatchJobQueues)
+}
+
+func ListBatchJobQueues(sess *session.Session) ([]Resource, error) {
+	svc := batch.New(sess)
+	resources := []Resource{}
+
+	params := &batch.DescribeJobQueuesInput{
+		MaxResults: aws.Int64(100),
+	}
+
+	for {
+		output, err := svc.DescribeJobQueues(params)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, queue := range output.JobQueues {
+			resources = append(resources, &BatchJobQueue{
+				svc:      svc,
+				jobQueue: queue.JobQueueName,
+			})
+		}
+
+		if output.NextToken == nil {
+			break
+		}
+
+		params.NextToken = output.NextToken
+	}
+
+	return resources, nil
+}
+
+func (f *BatchJobQueue) Remove() error {
+
+	_, err := f.svc.DeleteJobQueue(&batch.DeleteJobQueueInput{
+		JobQueue: f.jobQueue,
+	})
+
+	return err
+}
+
+func (f *BatchJobQueue) String() string {
+	return *f.jobQueue
+}

resources/batch-jobqueuestates.go

@@ -0,0 +1,73 @@
+package resources
+
+import (
+	"fmt"
+	"strings"
+
+	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/aws/session"
+	"github.com/aws/aws-sdk-go/service/batch"
+)
+
+type BatchJobQueueState struct {
+	svc      *batch.Batch
+	jobQueue *string
+	state    *string
+}
+
+func init() {
+	register("BatchJobQueueState", ListBatchJobQueueStates)
+}
+
+func ListBatchJobQueueStates(sess *session.Session) ([]Resource, error) {
+	svc := batch.New(sess)
+	resources := []Resource{}
+
+	params := &batch.DescribeJobQueuesInput{
+		MaxResults: aws.Int64(100),
+	}
+
+	for {
+		output, err := svc.DescribeJobQueues(params)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, queue := range output.JobQueues {
+			resources = append(resources, &BatchJobQueueState{
+				svc:      svc,
+				jobQueue: queue.JobQueueName,
+				state:    queue.State,
+			})
+		}
+
+		if output.NextToken == nil {
+			break
+		}
+
+		params.NextToken = output.NextToken
+	}
+
+	return resources, nil
+}
+
+func (f *BatchJobQueueState) Remove() error {
+
+	_, err := f.svc.UpdateJobQueue(&batch.UpdateJobQueueInput{
+		JobQueue: f.jobQueue,
+		State:    aws.String("DISABLED"),
+	})
+
+	return err
+}
+
+func (f *BatchJobQueueState) String() string {
+	return *f.jobQueue
+}
+
+func (f *BatchJobQueueState) Filter() error {
+	if strings.ToLower(*f.state) == "disabled" {
+		return fmt.Errorf("already disabled")
+	}
+	return nil
+}