Chore: Refactor quota service (#58643)
Chore: Refactor quota service (#57586) * Chore: refactore quota service * Apply suggestions from code review
This commit is contained in:
@@ -145,3 +145,28 @@ func (api *API) RegisterAPIEndpoints(m *metrics.API) {
|
||||
alertRules: api.AlertRules,
|
||||
}), m)
|
||||
}
|
||||
|
||||
func (api *API) Usage(ctx context.Context, scopeParams *quota.ScopeParameters) (*quota.Map, error) {
|
||||
u := "a.Map{}
|
||||
if orgUsage, err := api.RuleStore.Count(ctx, scopeParams.OrgID); err != nil {
|
||||
return u, err
|
||||
} else {
|
||||
tag, err := quota.NewTag(models.QuotaTargetSrv, models.QuotaTarget, quota.OrgScope)
|
||||
if err != nil {
|
||||
return u, err
|
||||
}
|
||||
u.Set(tag, orgUsage)
|
||||
}
|
||||
|
||||
if globalUsage, err := api.RuleStore.Count(ctx, 0); err != nil {
|
||||
return u, err
|
||||
} else {
|
||||
tag, err := quota.NewTag(models.QuotaTargetSrv, models.QuotaTarget, quota.GlobalScope)
|
||||
if err != nil {
|
||||
return u, err
|
||||
}
|
||||
u.Set(tag, globalUsage)
|
||||
}
|
||||
|
||||
return u, nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user