[Rules] Feat: Added Termination Policy for unused resources

Former-commit-id: 8170c60
MeNsaaH · Jan 4, 2021 · fe61f6a · fe61f6a
1 parent dd4a7c0
commit fe61f6a
Show file tree

Hide file tree

Showing 4 changed files with 33 additions and 9 deletions.
diff --git a/cmd/root.go b/cmd/root.go
@@ -150,8 150,8 @@ func refreshResources(providers []*provider.Provider) {
  // errs = provider.ResumeResources(resumableResources)
  // fmt.Println("Errors Resuming Resources: ", errs)
 
- // destroyableResources := provider.GetDestroyableResources(allResources)
- // fmt.Println("Destroyable Resources: ", destroyableResources)
  destroyableResources := provider.GetDestroyableResources(allResources)
  fmt.Println("Destroyable Resources: ", destroyableResources)
  // errs = provider.DestroyResources(destroyableResources)
  // fmt.Println("Errors Destroying Resources: ", errs)
  }

diff --git a/config/config.example.yaml b/config/config.example.yaml
@@ -62,22 62,18 @@ rules:
  env: test
  ci: true
  condition:
- terminationPolicy:
- - older than 24hrs
  terminationPolicy: older than 24hrs
  - name: Nuke all project A instances after Demo october 9th (staging)
  tags:
  env: staging
  project: A
  condition:
  terminationDate: october 9
- action: terminate
  - name: Delete all unused instances older than 48hrs (staging)
  tags:
  env: staging
  condition:
- terminationPolicy:
- - unused
- action: terminate
  terminationPolicy: unused
 
 
 timezone: Africa/Lagos

diff --git a/rules/rule.go b/rules/rule.go
@@ -75,6 75,8 @@ func ParseRule(rule Rule) Ruler {
  activeRule = &ActiveDurationRule{Rule: &rule}
  } else if rule.Condition.TerminationDate != "" {
  activeRule = &TerminationDateRule{Rule: &rule}
  } else if rule.Condition.TerminationPolicy != "" {
  activeRule = &TerminationPolicyRule{Rule: &rule}
  } else {
  log.Fatalf("No Conditions specified for rule: `%s`", rule.Name)
  }

diff --git a/rules/rules.go b/rules/rules.go
@@ -8,7 8,8 @@ import (
  "github.com/mensaah/reka/resource"
 )
 
-// TerminationDateRule defines rule that sets when a resource should be terminated
 // TerminationDateRule defines rule that sets when a resource should be terminated based on termination-date rule
 // set in config file. This rule checks if the set termination-date is past and activates destroy on the instance
 type TerminationDateRule struct {
  *Rule
  Date time.Time
@@ -72,3 73,28 @@ func (r ActiveDurationRule) CheckResource(res *resource.Resource) Action {
  }
  return DoNothing
 }
 
 // TerminationPolicyRule defines rule that sets when a resource should be terminated.
 type TerminationPolicyRule struct {
  *Rule
  Policy string
 }
 
 func (r *TerminationPolicyRule) validate() error {
  validPolicies := []string{"unused"}
  for _, p := range validPolicies {
  if p == r.Condition.TerminationPolicy {
  return nil
  }
  }
  r.Policy = r.Condition.TerminationPolicy
  return fmt.Errorf("Error parsing condition.terminationPolicy: Invalid Policy %s", r.Policy)
 }
 
 // CheckResource Returns a list of resources whose termination Date is exceeed
 func (r TerminationPolicyRule) CheckResource(res *resource.Resource) Action {
  if res.IsUnused() {
  return Destroy
  }
  return DoNothing
 }