feature: allow manager to preapre query context (#19451)

Co-authored-by: Qiu Jian <qiujian@yunionyun.com>
This commit is contained in:
Jian Qiu
2024-02-05 20:06:40 +08:00
committed by GitHub
co-authored by Qiu Jian
parent 7ad776ed5d
commit 81a12d1f16
270 changed files with 1460 additions and 1323 deletions
@@ -46,7 +46,7 @@ func (p *SQuotaPredicate) PreExecute(ctx context.Context, u *core.Unit, cs []cor
return true, nil
}
func fetchGuestUsageFromSchedInfo(s *api.SchedInfo) (computemodels.SQuota, computemodels.SRegionQuota) {
func fetchGuestUsageFromSchedInfo(ctx context.Context, s *api.SchedInfo) (computemodels.SQuota, computemodels.SRegionQuota) {
vcpuCount := s.Ncpu
if vcpuCount == 0 {
vcpuCount = 1
@@ -66,7 +66,7 @@ func fetchGuestUsageFromSchedInfo(s *api.SchedInfo) (computemodels.SQuota, compu
iNicCnt := 0
for _, netConfig := range s.Networks {
if computemodels.IsExitNetworkInfo(s.UserCred, netConfig) {
if computemodels.IsExitNetworkInfo(ctx, s.UserCred, netConfig) {
eNicCnt += 1
} else {
iNicCnt += 1
@@ -107,7 +107,7 @@ func (p *SQuotaPredicate) Execute(ctx context.Context, u *core.Unit, c core.Cand
computeKeys := c.Getter().GetQuotaKeys(d)
computeQuota, regionQuota := fetchGuestUsageFromSchedInfo(d)
computeQuota, regionQuota := fetchGuestUsageFromSchedInfo(ctx, d)
computeQuota.SetKeys(computeKeys)
regionQuota.SetKeys(computeKeys.SRegionalCloudResourceKeys)