kemotaha/toolkit
1
0
Fork 0
forked from sig_core/toolkit
Code Pull Requests Activity

fix: quotas code was broken

Browse Source
This commit is contained in:
Neil Hanlon 2023-12-20 19:51:27 -05:00
parent d42c919b09
commit dfa4caec54
Signed by untrusted user: neil
GPG Key ID: 705BC21EC3C70F34
1 changed files with 16 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
mangle/quotas.go
View File

@ -96,9 +96,19 @@ func getQuotaInfo(sqSvc *servicequotas.ServiceQuotas, quotaCode string, region s
var status string var status string
var caseId string var caseId string
if len(requestOutput.RequestedQuotas) > 0 { if len(requestOutput.RequestedQuotas) > 0 {
desiredValue = *requestOutput.RequestedQuotas[len(requestOutput.RequestedQuotas)-1].DesiredValue lastQuota := requestOutput.RequestedQuotas[len(requestOutput.RequestedQuotas)-1]
status = *requestOutput.RequestedQuotas[len(requestOutput.RequestedQuotas)-1].Status desiredValue = *lastQuota.DesiredValue
caseId = *requestOutput.RequestedQuotas[len(requestOutput.RequestedQuotas)-1].CaseId status = *lastQuota.Status
switch {
case status == "PENDING":
caseId = "N/A"
case status == "APPROVED":
caseId = "APPROVED"
case lastQuota.CaseId == nil:
panic("Unhandled case status. Please report this")
default:
caseId = *lastQuota.CaseId
}
} }
return &QuotaInfo{currentValue, desiredValue, status, caseId} return &QuotaInfo{currentValue, desiredValue, status, caseId}
} }
@ -122,7 +132,7 @@ func listQuotas(sess *session.Session, quotaCode string, regions []*string) {
wg.Wait() wg.Wait()
} }
func requestQuotaIncrease(sess *session.Session, quotaCode string, regions []string, quota float64) { func requestQuotaIncrease(sess *session.Session, quotaCode string, regions []*string, quota float64) {
var wg sync.WaitGroup var wg sync.WaitGroup
wg.Add(len(regions)) wg.Add(len(regions))
@ -146,7 +156,7 @@ func requestQuotaIncrease(sess *session.Session, quotaCode string, regions []str
} }
fmt.Printf("Successfully submitted request with ID: %s\n", aws.StringValue(output.RequestedQuota.Id)) fmt.Printf("Successfully submitted request with ID: %s\n", aws.StringValue(output.RequestedQuota.Id))
} }
}(region) }(*region)
} }
wg.Wait() wg.Wait()
} }
@ -174,5 +184,5 @@ func main() {
listQuotas(sess, quotaCode, regions) listQuotas(sess, quotaCode, regions)
// Request quota increase for all regions // Request quota increase for all regions
// requestQuotaIncrease(sess, quotaCode, regions) //requestQuotaIncrease(sess, quotaCode, regions, 50)
} }