Skip to content
This repository was archived by the owner on May 25, 2023. It is now read-only.

Commit b9bf82a

Browse files
authored
Merge pull request #875 from asifdxtreme/bump
Cherry-Pick #306 from Volcano
2 parents a041fda + 10fd703 commit b9bf82a

File tree

2 files changed

+37
-2
lines changed

2 files changed

+37
-2
lines changed

pkg/scheduler/api/node_info.go

Lines changed: 18 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -158,6 +158,18 @@ func (ni *NodeInfo) SetNode(node *v1.Node) {
158158
}
159159
}
160160

161+
func (ni *NodeInfo) allocateIdleResource(ti *TaskInfo) error {
162+
if ti.Resreq.LessEqual(ni.Idle) {
163+
ni.Idle.Sub(ti.Resreq)
164+
return nil
165+
}
166+
ni.State = NodeState{
167+
Phase: NotReady,
168+
Reason: "OutOfSync",
169+
}
170+
return fmt.Errorf("Selected node NotReady")
171+
}
172+
161173
// AddTask is used to add a task in nodeInfo object
162174
func (ni *NodeInfo) AddTask(task *TaskInfo) error {
163175
key := PodKey(task.Pod)
@@ -173,12 +185,16 @@ func (ni *NodeInfo) AddTask(task *TaskInfo) error {
173185
if ni.Node != nil {
174186
switch ti.Status {
175187
case Releasing:
188+
if err := ni.allocateIdleResource(ti); err != nil {
189+
return err
190+
}
176191
ni.Releasing.Add(ti.Resreq)
177-
ni.Idle.Sub(ti.Resreq)
178192
case Pipelined:
179193
ni.Releasing.Sub(ti.Resreq)
180194
default:
181-
ni.Idle.Sub(ti.Resreq)
195+
if err := ni.allocateIdleResource(ti); err != nil {
196+
return err
197+
}
182198
}
183199

184200
ni.Used.Add(ti.Resreq)

pkg/scheduler/api/node_info_test.go

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,9 @@ func TestNodeInfo_AddPod(t *testing.T) {
3737
case01Node := buildNode("n1", buildResourceList("8000m", "10G"))
3838
case01Pod1 := buildPod("c1", "p1", "n1", v1.PodRunning, buildResourceList("1000m", "1G"), []metav1.OwnerReference{}, make(map[string]string))
3939
case01Pod2 := buildPod("c1", "p2", "n1", v1.PodRunning, buildResourceList("2000m", "2G"), []metav1.OwnerReference{}, make(map[string]string))
40+
// case2
41+
case02Node := buildNode("n2", buildResourceList("2000m", "1G"))
42+
case02Pod1 := buildPod("c2", "p1", "n2", v1.PodUnknown, buildResourceList("1000m", "2G"), []metav1.OwnerReference{}, make(map[string]string))
4043

4144
tests := []struct {
4245
name string
@@ -63,6 +66,22 @@ func TestNodeInfo_AddPod(t *testing.T) {
6366
},
6467
},
6568
},
69+
{
70+
name: "add 1 unknown pod",
71+
node: case02Node,
72+
pods: []*v1.Pod{case02Pod1},
73+
expected: &NodeInfo{
74+
Name: "n2",
75+
Node: case02Node,
76+
Idle: buildResource("2000m", "1G"),
77+
Used: EmptyResource(),
78+
Releasing: EmptyResource(),
79+
Allocatable: buildResource("2000m", "1G"),
80+
Capability: buildResource("2000m", "1G"),
81+
State: NodeState{Phase: NotReady, Reason: "OutOfSync"},
82+
Tasks: map[TaskID]*TaskInfo{},
83+
},
84+
},
6685
}
6786

6887
for i, test := range tests {

0 commit comments

Comments
 (0)