github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-08-01 15:58:37 +00:00
Code Issues Projects Releases Wiki Activity

fixup: fix comments and use a channel to pass err

Browse Source
This commit is contained in:
Wei Huang 2019-07-19 00:20:37 -07:00
parent dce6686c9a
commit 249752cc1f
No known key found for this signature in database
GPG Key ID: BE5E9752F8B6E005
2 changed files with 13 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
pkg/scheduler/algorithm/predicates/metadata.go
View File

@ -202,8 +202,8 @@ func getTPMapMatchingSpreadConstraints(pod *v1.Pod, nodeInfoMap map[string]*sche
allNodeNames = append(allNodeNames, name)
}
errCh := schedutil.NewErrorChannel()
var lock sync.Mutex
var firstError error
topologyPairsPodSpreadMap := &topologyPairsPodSpreadMap{
// topologyKeyToMinPodsMap will be initilized with proper size later.
@ -215,13 +215,6 @@ func getTPMapMatchingSpreadConstraints(pod *v1.Pod, nodeInfoMap map[string]*sche
topologyPairsPodSpreadMap.appendMaps(toAppend)
lock.Unlock()
}
catchError := func(err error) {
lock.Lock()
if firstError == nil {
firstError = err
}
lock.Unlock()
}
ctx, cancel := context.WithCancel(context.Background())
@ -229,12 +222,11 @@ func getTPMapMatchingSpreadConstraints(pod *v1.Pod, nodeInfoMap map[string]*sche
nodeInfo := nodeInfoMap[allNodeNames[i]]
node := nodeInfo.Node()
if node == nil {
catchError(fmt.Errorf("node %q not found", allNodeNames[i]))
cancel()
klog.Errorf("node %q not found", allNodeNames[i])
return
}
// Be design if NodeAffinity or NodeSelector is defined, spreading is
// applied to nodes that pass those filters.
// In accordance to design, if NodeAffinity or NodeSelector is defined,
// spreading is applied to nodes that pass those filters.
if !podMatchesNodeSelectorAndAffinityTerms(pod, node) {
return
}
@ -250,8 +242,7 @@ func getTPMapMatchingSpreadConstraints(pod *v1.Pod, nodeInfoMap map[string]*sche
for _, existingPod := range nodeInfo.Pods() {
ok, err := podMatchesAllSpreadConstraints(existingPod, pod.Namespace, constraints)
if err != nil {
catchError(err)
cancel()
errCh.SendErrorWithCancel(err, cancel)
return
}
if ok {
@ -280,8 +271,8 @@ func getTPMapMatchingSpreadConstraints(pod *v1.Pod, nodeInfoMap map[string]*sche
}
workqueue.ParallelizeUntil(ctx, 16, len(allNodeNames), processNode)
if firstError != nil {
return nil, firstError
if err := errCh.ReceiveError(); err != nil {
return nil, err
}
// calculate min match for each topology pair
@ -395,19 +386,18 @@ func (m *topologyPairsMaps) clone() *topologyPairsMaps {
return copy
}
func (podSpreadMap *topologyPairsPodSpreadMap) clone() *topologyPairsPodSpreadMap {
// podSpreadMap could be nil when EvenPodsSpread feature is disabled
if podSpreadMap == nil {
func (m *topologyPairsPodSpreadMap) clone() *topologyPairsPodSpreadMap {
// m could be nil when EvenPodsSpread feature is disabled
if m == nil {
return nil
}
copy := &topologyPairsPodSpreadMap{
topologyKeyToMinPodsMap: make(map[string]int32),
topologyPairsMaps: newTopologyPairsMaps(),
topologyPairsMaps: m.topologyPairsMaps.clone(),
}
for key, minMatched := range podSpreadMap.topologyKeyToMinPodsMap {
for key, minMatched := range m.topologyKeyToMinPodsMap {
copy.topologyKeyToMinPodsMap[key] = minMatched
}
copy.topologyPairsMaps.appendMaps(podSpreadMap.topologyPairsMaps)
return copy
}

2
pkg/scheduler/testing/utils.go → pkg/scheduler/testing/wrappers.go
View File

@ -214,7 +214,7 @@ func (p *PodWrapper) Label(k, v string) *PodWrapper {
return p
}
// NodeWrapper wraps a LabelSelector inside.
// NodeWrapper wraps a Node inside.
type NodeWrapper struct{ v1.Node }
// MakeNode creates a Node wrapper.