mirror of
https://gitee.com/rainbond/Rainbond.git
synced 2024-12-10 19:59:01 +08:00
74 lines
2.5 KiB
Go
74 lines
2.5 KiB
Go
// RAINBOND, Application Management Platform
|
|
// Copyright (C) 2014-2019 Goodrain Co., Ltd.
|
|
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version. For any non-GPL usage of Rainbond,
|
|
// one or multiple Commercial Licenses authorized by Goodrain Co., Ltd.
|
|
// must be obtained first.
|
|
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package model
|
|
|
|
import (
|
|
v1 "k8s.io/api/core/v1"
|
|
"k8s.io/apimachinery/pkg/api/resource"
|
|
)
|
|
|
|
// NodeResource is a collection of compute resource.
|
|
type NodeResource struct {
|
|
MilliCPU int64 `json:"milli_cpu"`
|
|
Memory int64 `json:"memory"`
|
|
NvidiaGPU int64 `json:"nvidia_gpu"`
|
|
EphemeralStorage int64 `json:"ephemeral_storage"`
|
|
// We store allowedPodNumber (which is Node.Status.Allocatable.Pods().Value())
|
|
// explicitly as int, to avoid conversions and improve performance.
|
|
AllowedPodNumber int `json:"allowed_pod_number"`
|
|
}
|
|
|
|
// NewResource creates a Resource from ResourceList
|
|
func NewResource(rl v1.ResourceList) *NodeResource {
|
|
r := &NodeResource{}
|
|
r.Add(rl)
|
|
return r
|
|
}
|
|
|
|
// Add adds ResourceList into Resource.
|
|
func (r *NodeResource) Add(rl v1.ResourceList) {
|
|
if r == nil {
|
|
return
|
|
}
|
|
|
|
for rName, rQuant := range rl {
|
|
switch rName {
|
|
case v1.ResourceCPU:
|
|
r.MilliCPU += rQuant.MilliValue()
|
|
case v1.ResourceMemory:
|
|
r.Memory += rQuant.Value()
|
|
case v1.ResourcePods:
|
|
r.AllowedPodNumber += int(rQuant.Value())
|
|
case v1.ResourceEphemeralStorage:
|
|
r.EphemeralStorage += rQuant.Value()
|
|
}
|
|
}
|
|
}
|
|
|
|
// ResourceList returns a resource list of this resource.
|
|
func (r *NodeResource) ResourceList() v1.ResourceList {
|
|
result := v1.ResourceList{
|
|
v1.ResourceCPU: *resource.NewMilliQuantity(r.MilliCPU, resource.DecimalSI),
|
|
v1.ResourceMemory: *resource.NewQuantity(r.Memory, resource.BinarySI),
|
|
v1.ResourcePods: *resource.NewQuantity(int64(r.AllowedPodNumber), resource.BinarySI),
|
|
v1.ResourceEphemeralStorage: *resource.NewQuantity(r.EphemeralStorage, resource.BinarySI),
|
|
}
|
|
return result
|
|
}
|