Skip to content

Commit

Permalink
update methods
Browse files Browse the repository at this point in the history
  • Loading branch information
dwertent committed Oct 13, 2021
1 parent 7a5f1e1 commit 5ec3d57
Showing 1 changed file with 22 additions and 19 deletions.
41 changes: 22 additions & 19 deletions opapolicy/datastructuresmethods.go
Original file line number Diff line number Diff line change
Expand Up @@ -55,22 +55,22 @@ func (ruleReport *RuleReport) GetRuleStatus() (string, []RuleResponse, []RuleRes
return status, failed, exceptions
}


func (controlReport *ControlReport) GetNumberOfResources() int {
sum := 0
for i := range controlReport.RuleReports {
sum += controlReport.RuleReports[i].GetNumberOfResources()
}
return sum
}

func (controlReport *ControlReport) GetNumberOfFailedResources() int {
sum := 0
for i := range controlReport.RuleReports {
sum += controlReport.RuleReports[i].GetNumberOfFailedResources()
}
return sum
}

// func (controlReport *ControlReport) GetNumberOfResources() int {
// sum := 0
// for i := range controlReport.RuleReports {
// sum += controlReport.RuleReports[i].GetNumberOfResources()
// }
// return sum
// }

func ParseRegoResult(regoResult *rego.ResultSet) ([]RuleResponse, error) {
var errs error
ruleResponses := []RuleResponse{}
Expand Down Expand Up @@ -110,6 +110,16 @@ func (controlReport *ControlReport) GetNumberOfResources() int {
return sum
}

func (ruleReport *RuleReport) GetNumberOfWarningResources() int {
sum := 0
for i := range ruleReport.RuleResponses {
if ruleReport.RuleResponses[i].GetSingleResultStatus() == "warning" {
sum += 1
}
}
return sum
}

func (controlReport *ControlReport) GetNumberOfWarningResources() int {
sum := 0
for i := range controlReport.RuleReports {
Expand Down Expand Up @@ -181,13 +191,6 @@ func (ctrl *ControlReport) GetID() string {
return "C-" + s
}


func (ruleReport *RuleReport) DeleteIfRedundantResponse(RuleResponse *RuleResponse, index int) bool {
if b, rr := ruleReport.IsDuplicateResponseOfResource(RuleResponse, index); b {
rr.AddMessageToResponse(RuleResponse.AlertMessage)
ruleReport.RuleResponses = removeResponse(ruleReport.RuleResponses, index)
return true
}
return false
}

// func RemoveResponse(slice []RuleResponse, index int) []RuleResponse {
// return append(slice[:index], slice[index+1:]...)
// }

0 comments on commit 5ec3d57

Please sign in to comment.