scheduler schedule according to quota limit

2026-09-24 16:03:43 +08:00 · 2020-01-03 13:58:04 +08:00
parent 61f04ff338
commit bc7f62a2ad
22 changed files with 392 additions and 43 deletions
@@ -0,0 +1,131 @@
+// Copyright 2019 Yunion
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package predicates
+
+import (
+	"context"
+	"yunion.io/x/onecloud/pkg/cloudcommon/db/quotas"
+	computemodels "yunion.io/x/onecloud/pkg/compute/models"
+	"yunion.io/x/onecloud/pkg/scheduler/api"
+	"yunion.io/x/onecloud/pkg/scheduler/core"
+)
+
+type SQuotaPredicate struct {
+	BasePredicate
+}
+
+func (p *SQuotaPredicate) Name() string {
+	return "quota"
+}
+
+func (p *SQuotaPredicate) Clone() core.FitPredicate {
+	return &SQuotaPredicate{}
+}
+
+func (p *SQuotaPredicate) PreExecute(u *core.Unit, cs []core.Candidater) (bool, error) {
+	return true, nil
+}
+
+func fetchGuestUsageFromSchedInfo(s *api.SchedInfo) (computemodels.SQuota, computemodels.SRegionQuota) {
+	vcpuCount := s.Ncpu
+	if vcpuCount == 0 {
+		vcpuCount = 1
+	}
+
+	vmemSize := s.Memory
+
+	diskSize := 0
+
+	for _, diskConfig := range s.Disks {
+		diskSize += diskConfig.SizeMb
+	}
+
+	devCount := len(s.IsolatedDevices)
+
+	eNicCnt := 0
+	iNicCnt := 0
+	eBw := 0
+	iBw := 0
+
+	for _, netConfig := range s.Networks {
+		if computemodels.IsExitNetworkInfo(netConfig) {
+			eNicCnt += 1
+			eBw += netConfig.BwLimit
+		} else {
+			iNicCnt += 1
+			iBw += netConfig.BwLimit
+		}
+	}
+
+	req := computemodels.SQuota{
+		Count:          1,
+		Cpu:            int(vcpuCount),
+		Memory:         int(vmemSize),
+		Storage:        diskSize,
+		IsolatedDevice: devCount,
+	}
+	regionReq := computemodels.SRegionQuota{
+		Port:  iNicCnt,
+		Eport: eNicCnt,
+		Bw:    iBw,
+		Ebw:   eBw,
+	}
+	return req, regionReq
+}
+
+func (p *SQuotaPredicate) Execute(u *core.Unit, c core.Candidater) (bool, []core.PredicateFailureReason, error) {
+	h := NewPredicateHelper(p, u, c)
+
+	d := u.SchedData()
+
+	computePending := computemodels.SQuota{}
+	regionPending := computemodels.SRegionQuota{}
+	if len(d.PendingUsages) > 0 {
+		d.PendingUsages[0].Unmarshal(&computePending)
+	}
+	if len(d.PendingUsages) > 1 {
+		d.PendingUsages[1].Unmarshal(&regionPending)
+	}
+
+	computeKeys := c.Getter().GetQuotaKeys(d)
+
+	computeQuota, regionQuota := fetchGuestUsageFromSchedInfo(d)
+
+	computeQuota.SetKeys(computeKeys)
+	regionQuota.SetKeys(computeKeys.SRegionalCloudResourceKeys)
+
+	ctx := context.Background()
+	minCnt := -1
+	if !computePending.IsEmpty() {
+		computeCnt, _ := quotas.GetQuotaCount(ctx, &computeQuota, computePending.GetKeys())
+		if minCnt < 0 || minCnt > computeCnt {
+			minCnt = computeCnt
+		}
+	}
+	if !regionPending.IsEmpty() {
+		regionCnt, _ := quotas.GetQuotaCount(ctx, &regionQuota, regionPending.GetKeys())
+		if minCnt < 0 || minCnt > regionCnt {
+			minCnt = regionCnt
+		}
+	}
+
+	if minCnt == 0 {
+		h.Exclude("quota limit")
+	} else if minCnt > 0 {
+		h.SetCapacity(int64(minCnt))
+	}
+
+	return h.GetResult()
+}