-
Notifications
You must be signed in to change notification settings - Fork 0
/
metadata.go
74 lines (68 loc) · 1.74 KB
/
metadata.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
package metadata
import (
"context"
"github.com/aws/aws-sdk-go-v2/aws"
"github.com/aws/aws-sdk-go-v2/service/ec2"
ec2types "github.com/aws/aws-sdk-go-v2/service/ec2/types"
"github.com/cantara/nerthus2/cloud/aws/util"
)
type VPC struct {
Id string `json:"id"`
}
func GetAllServerIds(e2 *ec2.Client) (ids []string, err error) {
first := true
var nextToken *string
for nextToken != nil || first {
first = false
result, err := e2.DescribeInstances(context.Background(), &ec2.DescribeInstancesInput{
NextToken: nextToken,
})
if err != nil {
return nil, err
}
nextToken = result.NextToken
for _, reservation := range result.Reservations {
for _, instance := range reservation.Instances {
ids = append(ids, aws.ToString(instance.InstanceId))
}
}
}
return
}
func GetAllServersWithMetadataV1IDs(e2 *ec2.Client) (ids []string, err error) {
first := true
var nextToken *string
for nextToken != nil || first {
first = false
result, err := e2.DescribeInstances(context.Background(), &ec2.DescribeInstancesInput{
NextToken: nextToken,
})
if err != nil {
return nil, err
}
nextToken = result.NextToken
for _, reservation := range result.Reservations {
for _, instance := range reservation.Instances {
if instance.MetadataOptions.HttpTokens == ec2types.HttpTokensStateRequired {
continue
}
ids = append(ids, aws.ToString(instance.InstanceId))
}
}
}
return
}
func SetMetadataV2(id string, e2 *ec2.Client) (err error) {
err = util.CheckEC2Session(e2)
if err != nil {
return
}
_, err = e2.ModifyInstanceMetadataOptions(context.Background(), &ec2.ModifyInstanceMetadataOptionsInput{
InstanceId: &id,
HttpTokens: ec2types.HttpTokensStateRequired,
})
if err != nil {
return
}
return
}