/
main.go
129 lines (114 loc) · 4.17 KB
/
main.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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
/*
* Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License").
* You may not use this file except in compliance with the License.
* A copy of the License is located at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* or in the "license" file accompanying this file. This file is distributed
* on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
* express or implied. See the License for the specific language governing
* permissions and limitations under the License.
*/
package main
import (
"flag"
"fmt"
"log"
"strings"
"time"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws-observability/aws-otel-collector/tools/workflow/cleaner/apigw"
"github.com/aws-observability/aws-otel-collector/tools/workflow/cleaner/aps"
"github.com/aws-observability/aws-otel-collector/tools/workflow/cleaner/autoscaling"
"github.com/aws-observability/aws-otel-collector/tools/workflow/cleaner/ebs"
"github.com/aws-observability/aws-otel-collector/tools/workflow/cleaner/ec2"
"github.com/aws-observability/aws-otel-collector/tools/workflow/cleaner/ecs"
"github.com/aws-observability/aws-otel-collector/tools/workflow/cleaner/efs"
"github.com/aws-observability/aws-otel-collector/tools/workflow/cleaner/iam"
"github.com/aws-observability/aws-otel-collector/tools/workflow/cleaner/lambda"
"github.com/aws-observability/aws-otel-collector/tools/workflow/cleaner/launchconfig"
"github.com/aws-observability/aws-otel-collector/tools/workflow/cleaner/loadbalancer"
)
const (
defaultDaysToKeep = 5
cleanerTypeAll = "all"
delimiter = ","
)
var (
// flags
daysToKeep int
cleanersToRun string
cleanerTypes = []string{aps.Type, autoscaling.Type, ec2.Type, ecs.Type, efs.Type, iam.Type, launchconfig.Type, loadbalancer.Type, ebs.Type, apigw.Type, lambda.Type}
cleanerOptions = strings.Join(cleanerTypes, delimiter)
)
func init() {
flag.IntVar(&daysToKeep, "keep", defaultDaysToKeep, "Days to keep a resource before cleaning it.")
flag.StringVar(&cleanersToRun, "clean", cleanerTypeAll,
fmt.Sprintf("Determines which cleaners to run. (e.g. -clean ec2%siam)\nOptions: %s,%s", delimiter, cleanerTypeAll, cleanerOptions))
}
func main() {
flag.Parse()
sess := session.Must(session.NewSessionWithOptions(session.Options{SharedConfigState: session.SharedConfigEnable}))
keepDuration := -1 * time.Hour * 24 * time.Duration(daysToKeep)
if strings.Contains(cleanersToRun, cleanerTypeAll) {
cleanersToRun = cleanerOptions
}
// The date used to determine if a resource can be cleaned.
// Anything older than the date will be removed.
expirationDate := time.Now().UTC().Add(keepDuration)
log.Printf("Expiration date set to %v", expirationDate)
for _, cleaner := range strings.Split(cleanersToRun, delimiter) {
switch cleaner {
case aps.Type:
if err := aps.Clean(sess, expirationDate); err != nil {
log.Fatalf("%v", err)
}
case autoscaling.Type:
if err := autoscaling.Clean(sess, expirationDate); err != nil {
log.Fatalf("%v", err)
}
case ec2.Type:
if err := ec2.Clean(sess, expirationDate); err != nil {
log.Fatalf("%v", err)
}
case ecs.Type:
if err := ecs.Clean(sess, expirationDate); err != nil {
log.Fatalf("%v", err)
}
case efs.Type:
if err := efs.Clean(sess, expirationDate); err != nil {
log.Fatalf("%v", err)
}
case iam.Type:
if err := iam.Clean(sess, expirationDate); err != nil {
log.Fatalf("%v", err)
}
case lambda.Type:
if err := lambda.Clean(sess, expirationDate); err != nil {
log.Fatalf("%v", err)
}
case launchconfig.Type:
if err := launchconfig.Clean(sess, expirationDate); err != nil {
log.Fatalf("%v", err)
}
case loadbalancer.Type:
if err := loadbalancer.Clean(sess, expirationDate); err != nil {
log.Fatalf("%v", err)
}
case ebs.Type:
if err := ebs.Clean(sess, expirationDate); err != nil {
log.Fatalf("%v", err)
}
case apigw.Type:
if err := apigw.Clean(sess, expirationDate); err != nil {
log.Fatalf("%v", err)
}
default:
log.Printf("Skipping invalid cleaner '%s'. Please see -h for options.", cleaner)
}
}
log.Printf("Finished cleaning AWS resources")
}