Tags for volumes in an Instance

ec2/tags.go

@@ -2,6 +2,7 @@ package ec2
 
 import (
 	"context"
+	"strings"
 
 	"github.com/YaleSpinup/apierror"
 	"github.com/YaleSpinup/ec2-api/common"
@@ -19,6 +20,43 @@ func (e *Ec2) UpdateTags(ctx context.Context, rawTags map[string]string, ids ...
 		tags = append(tags, &ec2.Tag{Key: aws.String(key), Value: aws.String(val)})
 	}
 
+	instanceIDs := []*string{}
+	for _, id := range ids {
+		if strings.HasPrefix(id, "i-") {
+			instanceIDs = append(instanceIDs, aws.String(id))
+		}
+	}
+
+	describeVolumesInput := ec2.DescribeVolumesInput{
+		Filters: []*ec2.Filter{
+			{
+				Name:   aws.String("attachment.instance-id"),
+				Values: instanceIDs,
+			},
+		},
+		MaxResults: aws.Int64(1000),
+	}
+
+	for {
+		out, err := e.Service.DescribeVolumesWithContext(ctx, &describeVolumesInput)
+		if err != nil {
+			return common.ErrCode("describing volumes for instance", err)
+		}
+
+		log.Debugf("got describe volumes output %+v", out)
+
+		for _, v := range out.Volumes {
+			ids = append(ids, aws.StringValue(v.VolumeId))
+		}
+
+		if out.NextToken != nil {
+			describeVolumesInput.NextToken = out.NextToken
+			continue
+		}
+
+		break
+	}
+
 	log.Infof("updating resources: %v with tags %+v", ids, tags)
 
 	input := ec2.CreateTagsInput{

ec2/tags_test.go

@@ -29,6 +29,13 @@ func (m *mockEC2Client) CreateTagsWithContext(ctx context.Context, input *ec2.Cr
 	return &ec2.CreateTagsOutput{}, nil
 }
 
+func (m *mockEC2Client) DescribeVolumesWithContext(aws aws.Context, inp *ec2.DescribeVolumesInput, opt ...request.Option) (*ec2.DescribeVolumesOutput, error) {
+	if m.err != nil {
+		return nil, m.err
+	}
+	return &ec2.DescribeVolumesOutput{}, nil
+}
+
 func TestEc2_UpdateTags(t *testing.T) {
 	type fields struct {
 		Service ec2iface.EC2API
@@ -65,7 +72,7 @@ func TestEc2_UpdateTags(t *testing.T) {
 		{
 			name:    "no ids",
 			fields:  fields{Service: newmockEC2Client(t, nil)},
-			args:    args{ctx: context.TODO(), tags: inpTags, ids: nil},
+			args:    args{ctx: context.TODO(), tags: inpTags, ids: []string{}},
 			wantErr: true,
 		},
 	}