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

Simplify FitPredicate for NodeSelector

Browse Source 
We can assume that n != nil holds, because otherwise offers are skipped.
This commit is contained in:
Dr. Stefan Schimanski 2015-09-25 17:15:58 +02:00
parent 6f2a1742ab
commit 112f80fa4a
2 changed files with 34 additions and 48 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

64
contrib/mesos/pkg/scheduler/podtask/pod_task_test.go
View File

@ -19,14 +19,14 @@ package podtask
import ( import (
"testing" "testing"
"github.com/gogo/protobuf/proto"
mesos "github.com/mesos/mesos-go/mesosproto" mesos "github.com/mesos/mesos-go/mesosproto"
mutil "github.com/mesos/mesos-go/mesosutil" mutil "github.com/mesos/mesos-go/mesosutil"
"github.com/stretchr/testify/assert"
"k8s.io/kubernetes/contrib/mesos/pkg/node"
mresource "k8s.io/kubernetes/contrib/mesos/pkg/scheduler/resource" mresource "k8s.io/kubernetes/contrib/mesos/pkg/scheduler/resource"
"k8s.io/kubernetes/pkg/api" "k8s.io/kubernetes/pkg/api"
"k8s.io/kubernetes/pkg/api/resource" "k8s.io/kubernetes/pkg/api/resource"
"github.com/gogo/protobuf/proto"
"github.com/stretchr/testify/assert"
) )
const ( const (
@ -270,13 +270,6 @@ func TestGeneratePodName(t *testing.T) {
func TestNodeSelector(t *testing.T) { func TestNodeSelector(t *testing.T) {
t.Parallel() t.Parallel()
sel1 := map[string]string{"k8s.mesosphere.io/attribute-rack": "a"}
sel2 := map[string]string{"k8s.mesosphere.io/attribute-rack": "a", "k8s.mesosphere.io/attribute-gen": "2014"}
sel3 := map[string]string{"kubernetes.io/hostname": "node1"}
sel4 := map[string]string{"kubernetes.io/hostname": "node2"}
sel5 := map[string]string{"k8s.mesosphere.io/attribute-old": "42"}
sel6 := map[string]string{"some.other/label": "43"}
newNode := func(hostName string, l map[string]string) *api.Node { newNode := func(hostName string, l map[string]string) *api.Node {
nodeLabels := map[string]string{"kubernetes.io/hostname": hostName} nodeLabels := map[string]string{"kubernetes.io/hostname": hostName}
if l != nil { if l != nil {
@ -294,37 +287,43 @@ func TestNodeSelector(t *testing.T) {
}, },
} }
} }
node1 := newNode("node1", nil) node1 := newNode("node1", node.SlaveAttributesToLabels([]*mesos.Attribute{
node2 := newNode("node2", nil) newTextAttribute("rack", "a"),
node3 := newNode("node3", map[string]string{ newTextAttribute("gen", "2014"),
"k8s.mesosphere.io/attribute-old": "42", newScalarAttribute("num", 42.0),
"k8s.mesosphere.io/attribute-gen": "2015", }))
"some.other/label": "43", node2 := newNode("node2", node.SlaveAttributesToLabels([]*mesos.Attribute{
newTextAttribute("rack", "b"),
newTextAttribute("gen", "2015"),
newScalarAttribute("num", 0.0),
}))
labels3 := node.SlaveAttributesToLabels([]*mesos.Attribute{
newTextAttribute("rack", "c"),
newTextAttribute("gen", "2015"),
newScalarAttribute("old", 42),
}) })
labels3["some.other/label"] = "43"
node3 := newNode("node3", labels3)
tests := []struct { tests := []struct {
selector map[string]string selector map[string]string
attrs []*mesos.Attribute
node *api.Node node *api.Node
ok bool ok bool
desc string desc string
}{ }{
{sel1, []*mesos.Attribute{newTextAttribute("rack", "a")}, node1, true, "label value matches"}, {map[string]string{"k8s.mesosphere.io/attribute-rack": "a"}, node1, true, "label value matches"},
{sel1, []*mesos.Attribute{newTextAttribute("rack", "b")}, node1, false, "label value does not match"}, {map[string]string{"k8s.mesosphere.io/attribute-rack": "b"}, node1, false, "label value does not match"},
{sel1, []*mesos.Attribute{newTextAttribute("rack", "a"), newTextAttribute("gen", "2014")}, node1, true, "required labels match"}, {map[string]string{"k8s.mesosphere.io/attribute-rack": "a", "k8s.mesosphere.io/attribute-gen": "2014"}, node1, true, "multiple required labels match"},
{sel1, []*mesos.Attribute{newTextAttribute("rack", "a"), newScalarAttribute("num", 42.0)}, node1, true, "scalar label matches"}, {map[string]string{"k8s.mesosphere.io/attribute-rack": "a", "k8s.mesosphere.io/attribute-gen": "2015"}, node1, false, "one label does not match"},
{sel1, []*mesos.Attribute{newScalarAttribute("rack", 42.0)}, node1, false, "scalar label does not match"}, {map[string]string{"k8s.mesosphere.io/attribute-rack": "a", "k8s.mesosphere.io/attribute-num": "42"}, node1, true, "scalar label matches"},
{sel2, []*mesos.Attribute{newTextAttribute("rack", "a"), newTextAttribute("gen", "2014")}, node1, true, "all labels match"}, {map[string]string{"k8s.mesosphere.io/attribute-rack": "a", "k8s.mesosphere.io/attribute-num": "43"}, node1, false, "scalar label does not match"},
{sel2, []*mesos.Attribute{newTextAttribute("rack", "a"), newTextAttribute("gen", "2015")}, node1, false, "one label does not match"},
{sel3, []*mesos.Attribute{}, node1, true, "hostname label matches"}, {map[string]string{"kubernetes.io/hostname": "node1"}, node1, true, "hostname label matches"},
{sel4, []*mesos.Attribute{}, node1, false, "hostname label does not match"}, {map[string]string{"kubernetes.io/hostname": "node2"}, node1, false, "hostname label does not match"},
{sel4, []*mesos.Attribute{}, node2, true, "hostname label does not match"}, {map[string]string{"kubernetes.io/hostname": "node2"}, node2, true, "hostname label matches"},
{sel5, []*mesos.Attribute{}, node3, false, "old slave attribute is removed"}, {map[string]string{"some.other/label": "43"}, node1, false, "non-slave attribute does not match"},
{sel6, []*mesos.Attribute{}, node1, false, "non-slave attribute does not match"}, {map[string]string{"some.other/label": "43"}, node3, true, "non-slave attribute matches"},
{sel6, []*mesos.Attribute{}, node3, true, "non-slave attribute matches"},
{sel2, []*mesos.Attribute{newTextAttribute("rack", "a"), newTextAttribute("gen", "2014")}, node3, true, "old slave attributes are overwritten"},
} }
for _, ts := range tests { for _, ts := range tests {
@ -335,8 +334,7 @@ func TestNodeSelector(t *testing.T) {
mutil.NewScalarResource("cpus", t_min_cpu), mutil.NewScalarResource("cpus", t_min_cpu),
mutil.NewScalarResource("mem", t_min_mem), mutil.NewScalarResource("mem", t_min_mem),
}, },
Attributes: ts.attrs, Hostname: &ts.node.Name,
Hostname: &ts.node.Name,
} }
if got, want := DefaultPredicate(task, offer, ts.node), ts.ok; got != want { if got, want := DefaultPredicate(task, offer, ts.node), ts.ok; got != want {
t.Fatalf("expected acceptance of offer for selector %v to be %v, got %v: %q", ts.selector, want, got, ts.desc) t.Fatalf("expected acceptance of offer for selector %v to be %v, got %v: %q", ts.selector, want, got, ts.desc)

18
contrib/mesos/pkg/scheduler/podtask/predicate.go
View File

@ -19,7 +19,6 @@ package podtask
import ( import (
log "github.com/golang/glog" log "github.com/golang/glog"
mesos "github.com/mesos/mesos-go/mesosproto" mesos "github.com/mesos/mesos-go/mesosproto"
"k8s.io/kubernetes/contrib/mesos/pkg/node"
mresource "k8s.io/kubernetes/contrib/mesos/pkg/scheduler/resource" mresource "k8s.io/kubernetes/contrib/mesos/pkg/scheduler/resource"
"k8s.io/kubernetes/pkg/api" "k8s.io/kubernetes/pkg/api"
"k8s.io/kubernetes/pkg/labels" "k8s.io/kubernetes/pkg/labels"
@ -59,22 +58,11 @@ func NodeSelectorPredicate(t *T, offer *mesos.Offer, n *api.Node) bool {
// check the NodeSelector // check the NodeSelector
if len(t.Pod.Spec.NodeSelector) > 0 { if len(t.Pod.Spec.NodeSelector) > 0 {
l := map[string]string{ if n.Labels == nil {
"kubernetes.io/hostname": offer.GetHostname(), return false
} }
if n != nil && n.Labels != nil {
for k, v := range n.Labels {
if !node.IsSlaveAttributeLabel(k) {
l[k] = v
}
}
}
for k, v := range node.SlaveAttributesToLabels(offer.Attributes) {
l[k] = v
}
selector := labels.SelectorFromSet(t.Pod.Spec.NodeSelector) selector := labels.SelectorFromSet(t.Pod.Spec.NodeSelector)
if !selector.Matches(labels.Set(l)) { if !selector.Matches(labels.Set(n.Labels)) {
return false return false
} }
} }