use the same error for node not found in basic & delta snapshots

This commit is contained in:
Aleksandra Malinowska 2020-02-06 12:42:21 +01:00
parent ad888ce87d
commit 8ad81ef4ae
3 changed files with 8 additions and 7 deletions

View File

@ -59,7 +59,7 @@ func (data *internalBasicSnapshotData) getNodeInfo(nodeName string) (*schedulern
if v, ok := data.nodeInfoMap[nodeName]; ok {
return v, nil
}
return nil, fmt.Errorf("node %s not in snapshot", nodeName)
return nil, errNodeNotFound
}
func (data *internalBasicSnapshotData) listPods(selector labels.Selector) ([]*apiv1.Pod, error) {
@ -119,7 +119,7 @@ func (data *internalBasicSnapshotData) addNodes(nodes []*apiv1.Node) error {
func (data *internalBasicSnapshotData) removeNode(nodeName string) error {
if _, found := data.nodeInfoMap[nodeName]; !found {
return fmt.Errorf("node %s not in snapshot", nodeName)
return errNodeNotFound
}
delete(data.nodeInfoMap, nodeName)
return nil
@ -127,7 +127,7 @@ func (data *internalBasicSnapshotData) removeNode(nodeName string) error {
func (data *internalBasicSnapshotData) addPod(pod *apiv1.Pod, nodeName string) error {
if _, found := data.nodeInfoMap[nodeName]; !found {
return fmt.Errorf("node %s not in snapshot", nodeName)
return errNodeNotFound
}
data.nodeInfoMap[nodeName].AddPod(pod)
return nil
@ -136,7 +136,7 @@ func (data *internalBasicSnapshotData) addPod(pod *apiv1.Pod, nodeName string) e
func (data *internalBasicSnapshotData) removePod(namespace, podName, nodeName string) error {
nodeInfo, found := data.nodeInfoMap[nodeName]
if !found {
return fmt.Errorf("node not found")
return errNodeNotFound
}
for _, pod := range nodeInfo.Pods() {
if pod.Namespace == namespace && pod.Name == podName {

View File

@ -17,6 +17,8 @@ limitations under the License.
package simulator
import (
"errors"
apiv1 "k8s.io/api/core/v1"
schedulerlisters "k8s.io/kubernetes/pkg/scheduler/listers"
)
@ -48,3 +50,5 @@ type ClusterSnapshot interface {
// Clear reset cluster snapshot to empty, unforked state
Clear()
}
var errNodeNotFound = errors.New("node not found")

View File

@ -17,7 +17,6 @@ limitations under the License.
package simulator
import (
"errors"
"fmt"
apiv1 "k8s.io/api/core/v1"
@ -61,8 +60,6 @@ type internalDeltaSnapshotData struct {
havePodsWithAffinity []*schedulernodeinfo.NodeInfo
}
var errNodeNotFound = errors.New("node not found")
func newInternalDeltaSnapshotData() *internalDeltaSnapshotData {
return &internalDeltaSnapshotData{
addedNodeInfoMap: make(map[string]*schedulernodeinfo.NodeInfo),