add SpotInstance resource

Author: pww217

resources/ec2-spot-instance-request.go

@@ -0,0 +1,66 @@
+package resources
+
+import (
+	"fmt"
+
+	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/aws/session"
+	"github.com/aws/aws-sdk-go/service/ec2"
+)
+
+type EC2SpotInstanceRequest struct {
+	svc   *ec2.EC2
+	id    string
+	state string
+}
+
+func init() {
+	register("EC2SpotInstanceRequest", ListEC2SpotInstanceRequests)
+}
+
+func ListEC2SpotInstanceRequests(sess *session.Session) ([]Resource, error) {
+	svc := ec2.New(sess)
+
+	resp, err := svc.DescribeSpotInstanceRequests(nil)
+	if err != nil {
+		return nil, err
+	}
+
+	resources := make([]Resource, 0)
+	for _, config := range resp.SpotInstanceRequestConfigs {
+		resources = append(resources, &EC2SpotInstanceRequest{
+			svc:   svc,
+			id:    *config.SpotInstanceRequestId,
+			state: *config.SpotInstanceRequestState,
+		})
+	}
+
+	return resources, nil
+}
+
+func (i *EC2SpotInstanceRequest) Filter() error {
+	if i.state == "cancelled" {
+		return fmt.Errorf("already cancelled")
+	}
+	return nil
+}
+
+func (i *EC2SpotInstanceRequest) Remove() error {
+	params := &ec2.CancelSpotInstanceRequestsInput{
+		TerminateInstances: aws.Bool(true),
+		SpotInstanceRequestIds: []*string{
+			&i.id,
+		},
+	}
+
+	_, err := i.svc.CancelSpotInstanceRequests(params)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (i *EC2SpotInstanceRequest) String() string {
+	return i.id
+}