github/kubernetes
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update CanAdmitPodResult() in TopologyManager to take a TopologyHint

Browse Source 
Previously it only took a bool, which limited the logic it could perform
to determine if a pod should be admitted or not based on the merged hint
from the policy.
This commit is contained in:
Kevin Klues 2019-08-28 08:15:19 -05:00 committed by Louise Daly
parent c4f39177e0
commit 5ed80dadcf
8 changed files with 24 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pkg/kubelet/cm/topologymanager/policy.go
View File

@ -25,5 +25,5 @@ type Policy interface {
//Returns Policy Name
Name() string
//Returns Pod Admit Handler Response based on hints and policy type
CanAdmitPodResult(admit bool) lifecycle.PodAdmitResult
CanAdmitPodResult(hint *TopologyHint) lifecycle.PodAdmitResult
}

2
pkg/kubelet/cm/topologymanager/policy_best_effort.go
View File

@ -36,7 +36,7 @@ func (p *bestEffortPolicy) Name() string {
return PolicyBestEffort
}
func (p *bestEffortPolicy) CanAdmitPodResult(admit bool) lifecycle.PodAdmitResult {
func (p *bestEffortPolicy) CanAdmitPodResult(hint *TopologyHint) lifecycle.PodAdmitResult {
return lifecycle.PodAdmitResult{
Admit: true,
}

13
pkg/kubelet/cm/topologymanager/policy_best_effort_test.go
View File

@ -23,25 +23,24 @@ import (
func TestPolicyBestEffortCanAdmitPodResult(t *testing.T) {
tcases := []struct {
name string
admit bool
hint TopologyHint
expected bool
}{
{
name: "Affinity is set to false in topology hints",
admit: false,
name: "Preferred is set to false in topology hints",
hint: TopologyHint{nil, false},
expected: true,
},
{
name: "Affinity is set to true in topology hints",
admit: true,
name: "Preferred is set to true in topology hints",
hint: TopologyHint{nil, true},
expected: true,
},
}
for _, tc := range tcases {
policy := NewBestEffortPolicy()
admit := tc.admit
result := policy.CanAdmitPodResult(admit)
result := policy.CanAdmitPodResult(&tc.hint)
if result.Admit != tc.expected {
t.Errorf("Expected Admit field in result to be %t, got %t", tc.expected, result.Admit)

2
pkg/kubelet/cm/topologymanager/policy_none.go
View File

@ -36,7 +36,7 @@ func (p *nonePolicy) Name() string {
return PolicyNone
}
func (p *nonePolicy) CanAdmitPodResult(admit bool) lifecycle.PodAdmitResult {
func (p *nonePolicy) CanAdmitPodResult(hint *TopologyHint) lifecycle.PodAdmitResult {
return lifecycle.PodAdmitResult{
Admit: true,
}

13
pkg/kubelet/cm/topologymanager/policy_none_test.go
View File

@ -41,25 +41,24 @@ func TestName(t *testing.T) {
func TestPolicyNoneCanAdmitPodResult(t *testing.T) {
tcases := []struct {
name string
admit bool
hint TopologyHint
expected bool
}{
{
name: "Affinity is set to false in topology hints",
admit: false,
name: "Preferred is set to false in topology hints",
hint: TopologyHint{nil, false},
expected: true,
},
{
name: "Affinity is set to true in topology hints",
admit: true,
name: "Preferred is set to true in topology hints",
hint: TopologyHint{nil, true},
expected: true,
},
}
for _, tc := range tcases {
policy := NewNonePolicy()
admit := tc.admit
result := policy.CanAdmitPodResult(admit)
result := policy.CanAdmitPodResult(&tc.hint)
if result.Admit != tc.expected {
t.Errorf("Expected Admit field in result to be %t, got %t", tc.expected, result.Admit)

4
pkg/kubelet/cm/topologymanager/policy_restricted.go
View File

@ -36,8 +36,8 @@ func (p *restrictedPolicy) Name() string {
return PolicyRestricted
}
func (p *restrictedPolicy) CanAdmitPodResult(admit bool) lifecycle.PodAdmitResult {
if !admit {
func (p *restrictedPolicy) CanAdmitPodResult(hint *TopologyHint) lifecycle.PodAdmitResult {
if !hint.Preferred {
return lifecycle.PodAdmitResult{
Admit: false,
Reason: "Topology Affinity Error",

13
pkg/kubelet/cm/topologymanager/policy_restricted_test.go
View File

@ -23,25 +23,24 @@ import (
func TestPolicyRestrictedCanAdmitPodResult(t *testing.T) {
tcases := []struct {
name string
admit bool
hint TopologyHint
expected bool
}{
{
name: "Affinity is set to false in topology hints",
admit: false,
name: "Preferred is set to false in topology hints",
hint: TopologyHint{nil, false},
expected: false,
},
{
name: "Affinity is set to true in topology hints",
admit: true,
name: "Preferred is set to true in topology hints",
hint: TopologyHint{nil, true},
expected: true,
},
}
for _, tc := range tcases {
policy := NewRestrictedPolicy()
admit := tc.admit
result := policy.CanAdmitPodResult(admit)
result := policy.CanAdmitPodResult(&tc.hint)
if result.Admit != tc.expected {
t.Errorf("Expected Admit field in result to be %t, got %t", tc.expected, result.Admit)

2
pkg/kubelet/cm/topologymanager/topology_manager.go
View File

@ -308,7 +308,7 @@ func (m *manager) Admit(attrs *lifecycle.PodAdmitAttributes) lifecycle.PodAdmitR
if pod.Status.QOSClass == v1.PodQOSGuaranteed {
for _, container := range append(pod.Spec.InitContainers, pod.Spec.Containers...) {
result := m.calculateAffinity(*pod, container)
admitPod := m.policy.CanAdmitPodResult(result.Preferred)
admitPod := m.policy.CanAdmitPodResult(&result)
if !admitPod.Admit {
return admitPod
}