Skip to content
Projects
Groups
Snippets
Help
This project
Loading...
Sign in / Register
Toggle navigation
K
k3s
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Registry
Registry
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Jacklull
k3s
Commits
16fbb471
Commit
16fbb471
authored
Sep 16, 2016
by
deads2k
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
fix up service lister
parent
185a7adf
Expand all
Hide whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
177 additions
and
150 deletions
+177
-150
endpoints_controller.go
contrib/mesos/pkg/service/endpoints_controller.go
+3
-2
cluster_helper.go
...ation/pkg/federation-controller/service/cluster_helper.go
+2
-1
service_helper.go
...ation/pkg/federation-controller/service/service_helper.go
+1
-1
servicecontroller.go
...on/pkg/federation-controller/service/servicecontroller.go
+3
-2
listers.go
pkg/client/cache/listers.go
+0
-41
listers_core.go
pkg/client/cache/listers_core.go
+62
-0
listers_test.go
pkg/client/cache/listers_test.go
+1
-1
endpoints_controller.go
pkg/controller/endpoint/endpoints_controller.go
+3
-2
endpoints_controller_test.go
pkg/controller/endpoint/endpoints_controller_test.go
+11
-11
servicecontroller.go
pkg/controller/service/servicecontroller.go
+3
-2
envvars.go
pkg/kubelet/envvars/envvars.go
+3
-3
envvars_test.go
pkg/kubelet/envvars/envvars_test.go
+43
-45
kubelet.go
pkg/kubelet/kubelet.go
+14
-11
kubelet_test.go
pkg/kubelet/kubelet_test.go
+7
-8
listers.go
plugin/pkg/scheduler/algorithm/listers.go
+8
-7
predicates.go
plugin/pkg/scheduler/algorithm/predicates/predicates.go
+1
-1
predicates_test.go
plugin/pkg/scheduler/algorithm/predicates/predicates_test.go
+9
-9
selector_spreading.go
.../pkg/scheduler/algorithm/priorities/selector_spreading.go
+1
-1
selector_spreading_test.go
...scheduler/algorithm/priorities/selector_spreading_test.go
+0
-0
factory.go
plugin/pkg/scheduler/factory/factory.go
+2
-2
No files found.
contrib/mesos/pkg/service/endpoints_controller.go
View file @
16fbb471
...
...
@@ -55,7 +55,7 @@ func NewEndpointController(client *clientset.Clientset) *endpointController {
client
:
client
,
queue
:
workqueue
.
NewNamedRateLimitingQueue
(
workqueue
.
DefaultControllerRateLimiter
(),
"endpoint"
),
}
e
.
serviceStore
.
Store
,
e
.
serviceController
=
cache
.
New
Informer
(
e
.
serviceStore
.
Indexer
,
e
.
serviceController
=
cache
.
NewIndexer
Informer
(
&
cache
.
ListWatch
{
ListFunc
:
func
(
options
api
.
ListOptions
)
(
runtime
.
Object
,
error
)
{
return
e
.
client
.
Core
()
.
Services
(
api
.
NamespaceAll
)
.
List
(
options
)
...
...
@@ -73,6 +73,7 @@ func NewEndpointController(client *clientset.Clientset) *endpointController {
},
DeleteFunc
:
e
.
enqueueService
,
},
cache
.
Indexers
{
cache
.
NamespaceIndex
:
cache
.
MetaNamespaceIndexFunc
},
)
e
.
podStore
.
Indexer
,
e
.
podController
=
cache
.
NewIndexerInformer
(
...
...
@@ -262,7 +263,7 @@ func (e *endpointController) syncService(key string) error {
defer
func
()
{
glog
.
V
(
4
)
.
Infof
(
"Finished syncing service %q endpoints. (%v)"
,
key
,
time
.
Now
()
.
Sub
(
startTime
))
}()
obj
,
exists
,
err
:=
e
.
serviceStore
.
Store
.
GetByKey
(
key
)
obj
,
exists
,
err
:=
e
.
serviceStore
.
Indexer
.
GetByKey
(
key
)
if
err
!=
nil
||
!
exists
{
// Delete the corresponding endpoint, as the service has been deleted.
// TODO: Please note that this will delete an endpoint when a
...
...
federation/pkg/federation-controller/service/cluster_helper.go
View file @
16fbb471
...
...
@@ -114,7 +114,7 @@ func (cc *clusterClientCache) startClusterLW(cluster *v1beta1.Cluster, clusterNa
},
)
cachedClusterClient
.
serviceStore
.
Store
,
cachedClusterClient
.
serviceController
=
cache
.
New
Informer
(
cachedClusterClient
.
serviceStore
.
Indexer
,
cachedClusterClient
.
serviceController
=
cache
.
NewIndexer
Informer
(
&
cache
.
ListWatch
{
ListFunc
:
func
(
options
api
.
ListOptions
)
(
pkg_runtime
.
Object
,
error
)
{
return
clientset
.
Core
()
.
Services
(
v1
.
NamespaceAll
)
.
List
(
options
)
...
...
@@ -149,6 +149,7 @@ func (cc *clusterClientCache) startClusterLW(cluster *v1beta1.Cluster, clusterNa
glog
.
V
(
2
)
.
Infof
(
"Service %s/%s deletion found and enque to service store %s"
,
service
.
Namespace
,
service
.
Name
,
clusterName
)
},
},
cache
.
Indexers
{
cache
.
NamespaceIndex
:
cache
.
MetaNamespaceIndexFunc
},
)
cc
.
clientMap
[
clusterName
]
=
cachedClusterClient
go
cachedClusterClient
.
serviceController
.
Run
(
wait
.
NeverStop
)
...
...
federation/pkg/federation-controller/service/service_helper.go
View file @
16fbb471
...
...
@@ -63,7 +63,7 @@ func (cc *clusterClientCache) syncService(key, clusterName string, clusterCache
// if serviceCache does not exists, that means the service is not created by federation, we should skip it
return
nil
}
serviceInterface
,
exists
,
err
:=
clusterCache
.
serviceStore
.
GetByKey
(
key
)
serviceInterface
,
exists
,
err
:=
clusterCache
.
serviceStore
.
Indexer
.
GetByKey
(
key
)
if
err
!=
nil
{
glog
.
Errorf
(
"Did not successfully get %v from store: %v, will retry later"
,
key
,
err
)
clusterCache
.
serviceQueue
.
Add
(
key
)
...
...
federation/pkg/federation-controller/service/servicecontroller.go
View file @
16fbb471
...
...
@@ -144,7 +144,7 @@ func New(federationClient federation_release_1_4.Interface, dns dnsprovider.Inte
queue
:
workqueue
.
New
(),
knownClusterSet
:
make
(
sets
.
String
),
}
s
.
serviceStore
.
Store
,
s
.
serviceController
=
cache
.
New
Informer
(
s
.
serviceStore
.
Indexer
,
s
.
serviceController
=
cache
.
NewIndexer
Informer
(
&
cache
.
ListWatch
{
ListFunc
:
func
(
options
api
.
ListOptions
)
(
pkg_runtime
.
Object
,
error
)
{
return
s
.
federationClient
.
Core
()
.
Services
(
v1
.
NamespaceAll
)
.
List
(
options
)
...
...
@@ -165,6 +165,7 @@ func New(federationClient federation_release_1_4.Interface, dns dnsprovider.Inte
},
DeleteFunc
:
s
.
enqueueService
,
},
cache
.
Indexers
{
cache
.
NamespaceIndex
:
cache
.
MetaNamespaceIndexFunc
},
)
s
.
clusterStore
.
Store
,
s
.
clusterController
=
cache
.
NewInformer
(
&
cache
.
ListWatch
{
...
...
@@ -816,7 +817,7 @@ func (s *ServiceController) syncService(key string) error {
glog
.
V
(
4
)
.
Infof
(
"Finished syncing service %q (%v)"
,
key
,
time
.
Now
()
.
Sub
(
startTime
))
}()
// obj holds the latest service info from apiserver
obj
,
exists
,
err
:=
s
.
serviceStore
.
Store
.
GetByKey
(
key
)
obj
,
exists
,
err
:=
s
.
serviceStore
.
Indexer
.
GetByKey
(
key
)
if
err
!=
nil
{
glog
.
Errorf
(
"Unable to retrieve service %v from store: %v"
,
key
,
err
)
s
.
queue
.
Add
(
key
)
...
...
pkg/client/cache/listers.go
View file @
16fbb471
...
...
@@ -409,47 +409,6 @@ func (s *StoreToDaemonSetLister) GetPodDaemonSets(pod *api.Pod) (daemonSets []ex
return
}
// StoreToServiceLister makes a Store that has the List method of the client.ServiceInterface
// The Store must contain (only) Services.
type
StoreToServiceLister
struct
{
Store
}
func
(
s
*
StoreToServiceLister
)
List
()
(
services
api
.
ServiceList
,
err
error
)
{
for
_
,
m
:=
range
s
.
Store
.
List
()
{
services
.
Items
=
append
(
services
.
Items
,
*
(
m
.
(
*
api
.
Service
)))
}
return
services
,
nil
}
// TODO: Move this back to scheduler as a helper function that takes a Store,
// rather than a method of StoreToServiceLister.
func
(
s
*
StoreToServiceLister
)
GetPodServices
(
pod
*
api
.
Pod
)
(
services
[]
api
.
Service
,
err
error
)
{
var
selector
labels
.
Selector
var
service
api
.
Service
for
_
,
m
:=
range
s
.
Store
.
List
()
{
service
=
*
m
.
(
*
api
.
Service
)
// consider only services that are in the same namespace as the pod
if
service
.
Namespace
!=
pod
.
Namespace
{
continue
}
if
service
.
Spec
.
Selector
==
nil
{
// services with nil selectors match nothing, not everything.
continue
}
selector
=
labels
.
Set
(
service
.
Spec
.
Selector
)
.
AsSelectorPreValidated
()
if
selector
.
Matches
(
labels
.
Set
(
pod
.
Labels
))
{
services
=
append
(
services
,
service
)
}
}
if
len
(
services
)
==
0
{
err
=
fmt
.
Errorf
(
"could not find service for pod %s in namespace %s with labels: %v"
,
pod
.
Name
,
pod
.
Namespace
,
pod
.
Labels
)
}
return
}
// StoreToEndpointsLister makes a Store that lists endpoints.
type
StoreToEndpointsLister
struct
{
Store
...
...
pkg/client/cache/listers_core.go
View file @
16fbb471
...
...
@@ -141,3 +141,65 @@ func (s *StoreToReplicationControllerLister) GetPodControllers(pod *api.Pod) (co
}
return
}
// StoreToServiceLister helps list services
type
StoreToServiceLister
struct
{
Indexer
Indexer
}
func
(
s
*
StoreToServiceLister
)
List
(
selector
labels
.
Selector
)
(
ret
[]
*
api
.
Service
,
err
error
)
{
err
=
ListAll
(
s
.
Indexer
,
selector
,
func
(
m
interface
{})
{
ret
=
append
(
ret
,
m
.
(
*
api
.
Service
))
})
return
ret
,
err
}
func
(
s
*
StoreToServiceLister
)
Services
(
namespace
string
)
storeServicesNamespacer
{
return
storeServicesNamespacer
{
s
.
Indexer
,
namespace
}
}
type
storeServicesNamespacer
struct
{
indexer
Indexer
namespace
string
}
func
(
s
storeServicesNamespacer
)
List
(
selector
labels
.
Selector
)
(
ret
[]
*
api
.
Service
,
err
error
)
{
err
=
ListAllByNamespace
(
s
.
indexer
,
s
.
namespace
,
selector
,
func
(
m
interface
{})
{
ret
=
append
(
ret
,
m
.
(
*
api
.
Service
))
})
return
ret
,
err
}
func
(
s
storeServicesNamespacer
)
Get
(
name
string
)
(
*
api
.
Service
,
error
)
{
obj
,
exists
,
err
:=
s
.
indexer
.
GetByKey
(
s
.
namespace
+
"/"
+
name
)
if
err
!=
nil
{
return
nil
,
err
}
if
!
exists
{
return
nil
,
errors
.
NewNotFound
(
api
.
Resource
(
"service"
),
name
)
}
return
obj
.
(
*
api
.
Service
),
nil
}
// TODO: Move this back to scheduler as a helper function that takes a Store,
// rather than a method of StoreToServiceLister.
func
(
s
*
StoreToServiceLister
)
GetPodServices
(
pod
*
api
.
Pod
)
(
services
[]
*
api
.
Service
,
err
error
)
{
allServices
,
err
:=
s
.
Services
(
pod
.
Namespace
)
.
List
(
labels
.
Everything
())
if
err
!=
nil
{
return
nil
,
err
}
for
i
:=
range
allServices
{
service
:=
allServices
[
i
]
if
service
.
Spec
.
Selector
==
nil
{
// services with nil selectors match nothing, not everything.
continue
}
selector
:=
labels
.
Set
(
service
.
Spec
.
Selector
)
.
AsSelectorPreValidated
()
if
selector
.
Matches
(
labels
.
Set
(
pod
.
Labels
))
{
services
=
append
(
services
,
service
)
}
}
return
services
,
nil
}
pkg/client/cache/listers_test.go
View file @
16fbb471
...
...
@@ -754,7 +754,7 @@ func TestStoreToPodLister(t *testing.T) {
}
func
TestStoreToServiceLister
(
t
*
testing
.
T
)
{
store
:=
New
Store
(
MetaNamespaceKeyFunc
)
store
:=
New
Indexer
(
MetaNamespaceKeyFunc
,
Indexers
{
NamespaceIndex
:
MetaNamespaceIndexFunc
}
)
store
.
Add
(
&
api
.
Service
{
ObjectMeta
:
api
.
ObjectMeta
{
Name
:
"foo"
},
Spec
:
api
.
ServiceSpec
{
...
...
pkg/controller/endpoint/endpoints_controller.go
View file @
16fbb471
...
...
@@ -79,7 +79,7 @@ func NewEndpointController(podInformer cache.SharedIndexInformer, client *client
queue
:
workqueue
.
NewNamedRateLimitingQueue
(
workqueue
.
DefaultControllerRateLimiter
(),
"endpoint"
),
}
e
.
serviceStore
.
Store
,
e
.
serviceController
=
cache
.
New
Informer
(
e
.
serviceStore
.
Indexer
,
e
.
serviceController
=
cache
.
NewIndexer
Informer
(
&
cache
.
ListWatch
{
ListFunc
:
func
(
options
api
.
ListOptions
)
(
runtime
.
Object
,
error
)
{
return
e
.
client
.
Core
()
.
Services
(
api
.
NamespaceAll
)
.
List
(
options
)
...
...
@@ -98,6 +98,7 @@ func NewEndpointController(podInformer cache.SharedIndexInformer, client *client
},
DeleteFunc
:
e
.
enqueueService
,
},
cache
.
Indexers
{
cache
.
NamespaceIndex
:
cache
.
MetaNamespaceIndexFunc
},
)
podInformer
.
AddEventHandler
(
cache
.
ResourceEventHandlerFuncs
{
...
...
@@ -334,7 +335,7 @@ func (e *EndpointController) syncService(key string) error {
glog
.
V
(
4
)
.
Infof
(
"Finished syncing service %q endpoints. (%v)"
,
key
,
time
.
Now
()
.
Sub
(
startTime
))
}()
obj
,
exists
,
err
:=
e
.
serviceStore
.
Store
.
GetByKey
(
key
)
obj
,
exists
,
err
:=
e
.
serviceStore
.
Indexer
.
GetByKey
(
key
)
if
err
!=
nil
||
!
exists
{
// Delete the corresponding endpoint, as the service has been deleted.
// TODO: Please note that this will delete an endpoint when a
...
...
pkg/controller/endpoint/endpoints_controller_test.go
View file @
16fbb471
...
...
@@ -110,7 +110,7 @@ func TestSyncEndpointsItemsPreserveNoSelector(t *testing.T) {
client
:=
clientset
.
NewForConfigOrDie
(
&
restclient
.
Config
{
Host
:
testServer
.
URL
,
ContentConfig
:
restclient
.
ContentConfig
{
GroupVersion
:
testapi
.
Default
.
GroupVersion
()}})
endpoints
:=
NewEndpointControllerFromClient
(
client
,
controller
.
NoResyncPeriodFunc
)
endpoints
.
podStoreSynced
=
alwaysReady
endpoints
.
serviceStore
.
Store
.
Add
(
&
api
.
Service
{
endpoints
.
serviceStore
.
Indexer
.
Add
(
&
api
.
Service
{
ObjectMeta
:
api
.
ObjectMeta
{
Name
:
"foo"
,
Namespace
:
ns
},
Spec
:
api
.
ServiceSpec
{
Ports
:
[]
api
.
ServicePort
{{
Port
:
80
}}},
})
...
...
@@ -174,7 +174,7 @@ func TestSyncEndpointsProtocolTCP(t *testing.T) {
endpoints
.
podStoreSynced
=
alwaysReady
addPods
(
endpoints
.
podStore
.
Indexer
,
ns
,
1
,
1
,
0
)
endpoints
.
serviceStore
.
Store
.
Add
(
&
api
.
Service
{
endpoints
.
serviceStore
.
Indexer
.
Add
(
&
api
.
Service
{
ObjectMeta
:
api
.
ObjectMeta
{
Name
:
"foo"
,
Namespace
:
ns
},
Spec
:
api
.
ServiceSpec
{
Selector
:
map
[
string
]
string
{},
...
...
@@ -216,7 +216,7 @@ func TestSyncEndpointsProtocolUDP(t *testing.T) {
endpoints
:=
NewEndpointControllerFromClient
(
client
,
controller
.
NoResyncPeriodFunc
)
endpoints
.
podStoreSynced
=
alwaysReady
addPods
(
endpoints
.
podStore
.
Indexer
,
ns
,
1
,
1
,
0
)
endpoints
.
serviceStore
.
Store
.
Add
(
&
api
.
Service
{
endpoints
.
serviceStore
.
Indexer
.
Add
(
&
api
.
Service
{
ObjectMeta
:
api
.
ObjectMeta
{
Name
:
"foo"
,
Namespace
:
ns
},
Spec
:
api
.
ServiceSpec
{
Selector
:
map
[
string
]
string
{},
...
...
@@ -255,7 +255,7 @@ func TestSyncEndpointsItemsEmptySelectorSelectsAll(t *testing.T) {
endpoints
:=
NewEndpointControllerFromClient
(
client
,
controller
.
NoResyncPeriodFunc
)
endpoints
.
podStoreSynced
=
alwaysReady
addPods
(
endpoints
.
podStore
.
Indexer
,
ns
,
1
,
1
,
0
)
endpoints
.
serviceStore
.
Store
.
Add
(
&
api
.
Service
{
endpoints
.
serviceStore
.
Indexer
.
Add
(
&
api
.
Service
{
ObjectMeta
:
api
.
ObjectMeta
{
Name
:
"foo"
,
Namespace
:
ns
},
Spec
:
api
.
ServiceSpec
{
Selector
:
map
[
string
]
string
{},
...
...
@@ -293,7 +293,7 @@ func TestSyncEndpointsItemsEmptySelectorSelectsAllNotReady(t *testing.T) {
endpoints
:=
NewEndpointControllerFromClient
(
client
,
controller
.
NoResyncPeriodFunc
)
endpoints
.
podStoreSynced
=
alwaysReady
addPods
(
endpoints
.
podStore
.
Indexer
,
ns
,
0
,
1
,
1
)
endpoints
.
serviceStore
.
Store
.
Add
(
&
api
.
Service
{
endpoints
.
serviceStore
.
Indexer
.
Add
(
&
api
.
Service
{
ObjectMeta
:
api
.
ObjectMeta
{
Name
:
"foo"
,
Namespace
:
ns
},
Spec
:
api
.
ServiceSpec
{
Selector
:
map
[
string
]
string
{},
...
...
@@ -331,7 +331,7 @@ func TestSyncEndpointsItemsEmptySelectorSelectsAllMixed(t *testing.T) {
endpoints
:=
NewEndpointControllerFromClient
(
client
,
controller
.
NoResyncPeriodFunc
)
endpoints
.
podStoreSynced
=
alwaysReady
addPods
(
endpoints
.
podStore
.
Indexer
,
ns
,
1
,
1
,
1
)
endpoints
.
serviceStore
.
Store
.
Add
(
&
api
.
Service
{
endpoints
.
serviceStore
.
Indexer
.
Add
(
&
api
.
Service
{
ObjectMeta
:
api
.
ObjectMeta
{
Name
:
"foo"
,
Namespace
:
ns
},
Spec
:
api
.
ServiceSpec
{
Selector
:
map
[
string
]
string
{},
...
...
@@ -373,7 +373,7 @@ func TestSyncEndpointsItemsPreexisting(t *testing.T) {
endpoints
:=
NewEndpointControllerFromClient
(
client
,
controller
.
NoResyncPeriodFunc
)
endpoints
.
podStoreSynced
=
alwaysReady
addPods
(
endpoints
.
podStore
.
Indexer
,
ns
,
1
,
1
,
0
)
endpoints
.
serviceStore
.
Store
.
Add
(
&
api
.
Service
{
endpoints
.
serviceStore
.
Indexer
.
Add
(
&
api
.
Service
{
ObjectMeta
:
api
.
ObjectMeta
{
Name
:
"foo"
,
Namespace
:
ns
},
Spec
:
api
.
ServiceSpec
{
Selector
:
map
[
string
]
string
{
"foo"
:
"bar"
},
...
...
@@ -414,7 +414,7 @@ func TestSyncEndpointsItemsPreexistingIdentical(t *testing.T) {
endpoints
:=
NewEndpointControllerFromClient
(
client
,
controller
.
NoResyncPeriodFunc
)
endpoints
.
podStoreSynced
=
alwaysReady
addPods
(
endpoints
.
podStore
.
Indexer
,
api
.
NamespaceDefault
,
1
,
1
,
0
)
endpoints
.
serviceStore
.
Store
.
Add
(
&
api
.
Service
{
endpoints
.
serviceStore
.
Indexer
.
Add
(
&
api
.
Service
{
ObjectMeta
:
api
.
ObjectMeta
{
Name
:
"foo"
,
Namespace
:
api
.
NamespaceDefault
},
Spec
:
api
.
ServiceSpec
{
Selector
:
map
[
string
]
string
{
"foo"
:
"bar"
},
...
...
@@ -435,7 +435,7 @@ func TestSyncEndpointsItems(t *testing.T) {
endpoints
.
podStoreSynced
=
alwaysReady
addPods
(
endpoints
.
podStore
.
Indexer
,
ns
,
3
,
2
,
0
)
addPods
(
endpoints
.
podStore
.
Indexer
,
"blah"
,
5
,
2
,
0
)
// make sure these aren't found!
endpoints
.
serviceStore
.
Store
.
Add
(
&
api
.
Service
{
endpoints
.
serviceStore
.
Indexer
.
Add
(
&
api
.
Service
{
ObjectMeta
:
api
.
ObjectMeta
{
Name
:
"foo"
,
Namespace
:
ns
},
Spec
:
api
.
ServiceSpec
{
Selector
:
map
[
string
]
string
{
"foo"
:
"bar"
},
...
...
@@ -478,7 +478,7 @@ func TestSyncEndpointsItemsWithLabels(t *testing.T) {
endpoints
.
podStoreSynced
=
alwaysReady
addPods
(
endpoints
.
podStore
.
Indexer
,
ns
,
3
,
2
,
0
)
serviceLabels
:=
map
[
string
]
string
{
"foo"
:
"bar"
}
endpoints
.
serviceStore
.
Store
.
Add
(
&
api
.
Service
{
endpoints
.
serviceStore
.
Indexer
.
Add
(
&
api
.
Service
{
ObjectMeta
:
api
.
ObjectMeta
{
Name
:
"foo"
,
Namespace
:
ns
,
...
...
@@ -539,7 +539,7 @@ func TestSyncEndpointsItemsPreexistingLabelsChange(t *testing.T) {
endpoints
.
podStoreSynced
=
alwaysReady
addPods
(
endpoints
.
podStore
.
Indexer
,
ns
,
1
,
1
,
0
)
serviceLabels
:=
map
[
string
]
string
{
"baz"
:
"blah"
}
endpoints
.
serviceStore
.
Store
.
Add
(
&
api
.
Service
{
endpoints
.
serviceStore
.
Indexer
.
Add
(
&
api
.
Service
{
ObjectMeta
:
api
.
ObjectMeta
{
Name
:
"foo"
,
Namespace
:
ns
,
...
...
pkg/controller/service/servicecontroller.go
View file @
16fbb471
...
...
@@ -119,7 +119,7 @@ func New(cloud cloudprovider.Interface, kubeClient clientset.Interface, clusterN
},
workingQueue
:
workqueue
.
NewDelayingQueue
(),
}
s
.
serviceStore
.
Store
,
s
.
serviceController
=
cache
.
New
Informer
(
s
.
serviceStore
.
Indexer
,
s
.
serviceController
=
cache
.
NewIndexer
Informer
(
&
cache
.
ListWatch
{
ListFunc
:
func
(
options
api
.
ListOptions
)
(
pkg_runtime
.
Object
,
error
)
{
return
s
.
kubeClient
.
Core
()
.
Services
(
api
.
NamespaceAll
)
.
List
(
options
)
...
...
@@ -141,6 +141,7 @@ func New(cloud cloudprovider.Interface, kubeClient clientset.Interface, clusterN
},
DeleteFunc
:
s
.
enqueueService
,
},
cache
.
Indexers
{
cache
.
NamespaceIndex
:
cache
.
MetaNamespaceIndexFunc
},
)
if
err
:=
s
.
init
();
err
!=
nil
{
return
nil
,
err
...
...
@@ -724,7 +725,7 @@ func (s *ServiceController) syncService(key string) error {
glog
.
V
(
4
)
.
Infof
(
"Finished syncing service %q (%v)"
,
key
,
time
.
Now
()
.
Sub
(
startTime
))
}()
// obj holds the latest service info from apiserver
obj
,
exists
,
err
:=
s
.
serviceStore
.
Store
.
GetByKey
(
key
)
obj
,
exists
,
err
:=
s
.
serviceStore
.
Indexer
.
GetByKey
(
key
)
if
err
!=
nil
{
glog
.
Infof
(
"Unable to retrieve service %v from store: %v"
,
key
,
err
)
s
.
workingQueue
.
Add
(
key
)
...
...
pkg/kubelet/envvars/envvars.go
View file @
16fbb471
...
...
@@ -27,10 +27,10 @@ import (
// FromServices builds environment variables that a container is started with,
// which tell the container where to find the services it may need, which are
// provided as an argument.
func
FromServices
(
services
*
api
.
ServiceList
)
[]
api
.
EnvVar
{
func
FromServices
(
services
[]
*
api
.
Service
)
[]
api
.
EnvVar
{
var
result
[]
api
.
EnvVar
for
i
:=
range
services
.
Items
{
service
:=
&
services
.
Item
s
[
i
]
for
i
:=
range
services
{
service
:=
service
s
[
i
]
// ignore services where ClusterIP is "None" or empty
// the services passed to this method should be pre-filtered
...
...
pkg/kubelet/envvars/envvars_test.go
View file @
16fbb471
...
...
@@ -25,63 +25,61 @@ import (
)
func
TestFromServices
(
t
*
testing
.
T
)
{
sl
:=
api
.
ServiceList
{
Items
:
[]
api
.
Service
{
{
ObjectMeta
:
api
.
ObjectMeta
{
Name
:
"foo-bar"
},
Spec
:
api
.
ServiceSpec
{
Selector
:
map
[
string
]
string
{
"bar"
:
"baz"
},
ClusterIP
:
"1.2.3.4"
,
Ports
:
[]
api
.
ServicePort
{
{
Port
:
8080
,
Protocol
:
"TCP"
},
},
sl
:=
[]
*
api
.
Service
{
{
ObjectMeta
:
api
.
ObjectMeta
{
Name
:
"foo-bar"
},
Spec
:
api
.
ServiceSpec
{
Selector
:
map
[
string
]
string
{
"bar"
:
"baz"
},
ClusterIP
:
"1.2.3.4"
,
Ports
:
[]
api
.
ServicePort
{
{
Port
:
8080
,
Protocol
:
"TCP"
},
},
},
{
ObjectMeta
:
api
.
ObjectMeta
{
Name
:
"abc-123"
},
Spec
:
api
.
ServiceSpec
{
Selector
:
map
[
string
]
string
{
"bar"
:
"baz"
},
ClusterIP
:
"5.6.7.8"
,
Ports
:
[]
api
.
ServicePort
{
{
Name
:
"u-d-p"
,
Port
:
8081
,
Protocol
:
"UDP"
},
{
Name
:
"t-c-p"
,
Port
:
8081
,
Protocol
:
"TC
P"
},
},
},
{
ObjectMeta
:
api
.
ObjectMeta
{
Name
:
"abc-123"
},
Spec
:
api
.
ServiceSpec
{
Selector
:
map
[
string
]
string
{
"bar"
:
"baz"
}
,
ClusterIP
:
"5.6.7.8"
,
Ports
:
[]
api
.
ServicePort
{
{
Name
:
"u-d-p"
,
Port
:
8081
,
Protocol
:
"UD
P"
},
{
Name
:
"t-c-p"
,
Port
:
8081
,
Protocol
:
"TCP"
},
},
},
{
ObjectMeta
:
api
.
ObjectMeta
{
Name
:
"q-u-u-x"
},
Spec
:
api
.
ServiceSpec
{
Selector
:
map
[
string
]
string
{
"bar"
:
"baz"
},
ClusterIP
:
"9.8.7.6"
,
Ports
:
[]
api
.
ServicePort
{
{
Port
:
8082
,
Protocol
:
"TCP"
},
{
Name
:
"8083"
,
Port
:
8083
,
Protocol
:
"TCP"
},
},
},
{
ObjectMeta
:
api
.
ObjectMeta
{
Name
:
"q-u-u-x"
},
Spec
:
api
.
ServiceSpec
{
Selector
:
map
[
string
]
string
{
"bar"
:
"baz"
}
,
ClusterIP
:
"9.8.7.6"
,
Ports
:
[]
api
.
ServicePort
{
{
Port
:
8082
,
Protocol
:
"TCP"
},
{
Name
:
"8083"
,
Port
:
8083
,
Protocol
:
"TCP"
},
},
},
{
ObjectMeta
:
api
.
ObjectMeta
{
Name
:
"svrc-clusterip-none"
},
Spec
:
api
.
ServiceSpec
{
Selector
:
map
[
string
]
string
{
"bar"
:
"baz"
},
ClusterIP
:
"None"
,
Ports
:
[]
api
.
ServicePort
{
{
Port
:
8082
,
Protocol
:
"TCP"
},
},
},
{
ObjectMeta
:
api
.
ObjectMeta
{
Name
:
"svrc-clusterip-none"
},
Spec
:
api
.
ServiceSpec
{
Selector
:
map
[
string
]
string
{
"bar"
:
"baz"
}
,
ClusterIP
:
"None"
,
Ports
:
[]
api
.
ServicePort
{
{
Port
:
8082
,
Protocol
:
"TCP"
},
},
},
{
ObjectMeta
:
api
.
ObjectMeta
{
Name
:
"svrc-clusterip-empty"
},
Spec
:
api
.
ServiceSpec
{
Selector
:
map
[
string
]
string
{
"bar"
:
"baz"
},
ClusterIP
:
""
,
Ports
:
[]
api
.
ServicePort
{
{
Port
:
8082
,
Protocol
:
"TCP"
},
},
},
{
ObjectMeta
:
api
.
ObjectMeta
{
Name
:
"svrc-clusterip-empty"
},
Spec
:
api
.
ServiceSpec
{
Selector
:
map
[
string
]
string
{
"bar"
:
"baz"
}
,
ClusterIP
:
""
,
Ports
:
[]
api
.
ServicePort
{
{
Port
:
8082
,
Protocol
:
"TCP"
},
},
},
},
}
vars
:=
envvars
.
FromServices
(
&
sl
)
vars
:=
envvars
.
FromServices
(
sl
)
expected
:=
[]
api
.
EnvVar
{
{
Name
:
"FOO_BAR_SERVICE_HOST"
,
Value
:
"1.2.3.4"
},
{
Name
:
"FOO_BAR_SERVICE_PORT"
,
Value
:
"8080"
},
...
...
pkg/kubelet/kubelet.go
View file @
16fbb471
...
...
@@ -76,6 +76,7 @@ import (
"k8s.io/kubernetes/pkg/kubelet/util/queue"
"k8s.io/kubernetes/pkg/kubelet/util/sliceutils"
"k8s.io/kubernetes/pkg/kubelet/volumemanager"
"k8s.io/kubernetes/pkg/labels"
"k8s.io/kubernetes/pkg/runtime"
"k8s.io/kubernetes/pkg/security/apparmor"
"k8s.io/kubernetes/pkg/types"
...
...
@@ -370,7 +371,7 @@ func NewMainKubelet(kubeCfg *componentconfig.KubeletConfiguration, kubeDeps *Kub
dockerExecHandler
=
&
dockertools
.
NativeExecHandler
{}
}
serviceStore
:=
cache
.
New
Store
(
cache
.
MetaNamespaceKeyFunc
)
serviceStore
:=
cache
.
New
Indexer
(
cache
.
MetaNamespaceKeyFunc
,
cache
.
Indexers
{
cache
.
NamespaceIndex
:
cache
.
MetaNamespaceIndexFunc
}
)
if
kubeClient
!=
nil
{
// TODO: cache.NewListWatchFromClient is limited as it takes a client implementation rather
// than an interface. There is no way to construct a list+watcher using resource name.
...
...
@@ -384,7 +385,7 @@ func NewMainKubelet(kubeCfg *componentconfig.KubeletConfiguration, kubeDeps *Kub
}
cache
.
NewReflector
(
listWatch
,
&
api
.
Service
{},
serviceStore
,
0
)
.
Run
()
}
serviceLister
:=
&
cache
.
StoreToServiceLister
{
Store
:
serviceStore
}
serviceLister
:=
&
cache
.
StoreToServiceLister
{
Indexer
:
serviceStore
}
nodeStore
:=
cache
.
NewStore
(
cache
.
MetaNamespaceKeyFunc
)
if
kubeClient
!=
nil
{
...
...
@@ -751,7 +752,7 @@ func NewMainKubelet(kubeCfg *componentconfig.KubeletConfiguration, kubeDeps *Kub
}
type
serviceLister
interface
{
List
(
)
(
api
.
ServiceList
,
error
)
List
(
labels
.
Selector
)
([]
*
api
.
Service
,
error
)
}
type
nodeLister
interface
{
...
...
@@ -1421,7 +1422,7 @@ var masterServices = sets.NewString("kubernetes")
// pod in namespace ns should see.
func
(
kl
*
Kubelet
)
getServiceEnvVarMap
(
ns
string
)
(
map
[
string
]
string
,
error
)
{
var
(
serviceMap
=
make
(
map
[
string
]
api
.
Service
)
serviceMap
=
make
(
map
[
string
]
*
api
.
Service
)
m
=
make
(
map
[
string
]
string
)
)
...
...
@@ -1431,15 +1432,16 @@ func (kl *Kubelet) getServiceEnvVarMap(ns string) (map[string]string, error) {
// Kubelets without masters (e.g. plain GCE ContainerVM) don't set env vars.
return
m
,
nil
}
services
,
err
:=
kl
.
serviceLister
.
List
()
services
,
err
:=
kl
.
serviceLister
.
List
(
labels
.
Everything
()
)
if
err
!=
nil
{
return
m
,
fmt
.
Errorf
(
"failed to list services when setting up env vars."
)
}
// project the services in namespace ns onto the master services
for
_
,
service
:=
range
services
.
Items
{
for
i
:=
range
services
{
service
:=
services
[
i
]
// ignore services where ClusterIP is "None" or empty
if
!
api
.
IsServiceIPSet
(
&
service
)
{
if
!
api
.
IsServiceIPSet
(
service
)
{
continue
}
serviceName
:=
service
.
Name
...
...
@@ -1459,12 +1461,13 @@ func (kl *Kubelet) getServiceEnvVarMap(ns string) (map[string]string, error) {
}
}
}
services
.
Items
=
[]
api
.
Service
{}
for
_
,
service
:=
range
serviceMap
{
services
.
Items
=
append
(
services
.
Items
,
service
)
mappedServices
:=
[]
*
api
.
Service
{}
for
key
:=
range
serviceMap
{
mappedServices
=
append
(
mappedServices
,
serviceMap
[
key
])
}
for
_
,
e
:=
range
envvars
.
FromServices
(
&
s
ervices
)
{
for
_
,
e
:=
range
envvars
.
FromServices
(
mappedS
ervices
)
{
m
[
e
.
Name
]
=
e
.
Value
}
return
m
,
nil
...
...
pkg/kubelet/kubelet_test.go
View file @
16fbb471
...
...
@@ -61,6 +61,7 @@ import (
kubetypes
"k8s.io/kubernetes/pkg/kubelet/types"
"k8s.io/kubernetes/pkg/kubelet/util/queue"
kubeletvolume
"k8s.io/kubernetes/pkg/kubelet/volumemanager"
"k8s.io/kubernetes/pkg/labels"
"k8s.io/kubernetes/pkg/runtime"
"k8s.io/kubernetes/pkg/types"
"k8s.io/kubernetes/pkg/util/clock"
...
...
@@ -896,13 +897,11 @@ func TestDNSConfigurationParams(t *testing.T) {
}
type
testServiceLister
struct
{
services
[]
api
.
Service
services
[]
*
api
.
Service
}
func
(
ls
testServiceLister
)
List
()
(
api
.
ServiceList
,
error
)
{
return
api
.
ServiceList
{
Items
:
ls
.
services
,
},
nil
func
(
ls
testServiceLister
)
List
(
labels
.
Selector
)
([]
*
api
.
Service
,
error
)
{
return
ls
.
services
,
nil
}
type
testNodeLister
struct
{
...
...
@@ -938,8 +937,8 @@ func (e envs) Swap(i, j int) { e[i], e[j] = e[j], e[i] }
func
(
e
envs
)
Less
(
i
,
j
int
)
bool
{
return
e
[
i
]
.
Name
<
e
[
j
]
.
Name
}
func
buildService
(
name
,
namespace
,
clusterIP
,
protocol
string
,
port
int
)
api
.
Service
{
return
api
.
Service
{
func
buildService
(
name
,
namespace
,
clusterIP
,
protocol
string
,
port
int
)
*
api
.
Service
{
return
&
api
.
Service
{
ObjectMeta
:
api
.
ObjectMeta
{
Name
:
name
,
Namespace
:
namespace
},
Spec
:
api
.
ServiceSpec
{
Ports
:
[]
api
.
ServicePort
{{
...
...
@@ -952,7 +951,7 @@ func buildService(name, namespace, clusterIP, protocol string, port int) api.Ser
}
func
TestMakeEnvironmentVariables
(
t
*
testing
.
T
)
{
services
:=
[]
api
.
Service
{
services
:=
[]
*
api
.
Service
{
buildService
(
"kubernetes"
,
api
.
NamespaceDefault
,
"1.2.3.1"
,
"TCP"
,
8081
),
buildService
(
"test"
,
"test1"
,
"1.2.3.3"
,
"TCP"
,
8083
),
buildService
(
"kubernetes"
,
"test2"
,
"1.2.3.4"
,
"TCP"
,
8084
),
...
...
plugin/pkg/scheduler/algorithm/listers.go
View file @
16fbb471
...
...
@@ -63,24 +63,25 @@ func (f FakePodLister) List(s labels.Selector) (selected []*api.Pod, err error)
// ServiceLister interface represents anything that can produce a list of services; the list is consumed by a scheduler.
type
ServiceLister
interface
{
// Lists all the services
List
(
)
(
api
.
ServiceList
,
error
)
List
(
labels
.
Selector
)
([]
*
api
.
Service
,
error
)
// Gets the services for the given pod
GetPodServices
(
*
api
.
Pod
)
([]
api
.
Service
,
error
)
GetPodServices
(
*
api
.
Pod
)
([]
*
api
.
Service
,
error
)
}
// FakeServiceLister implements ServiceLister on []api.Service for test purposes.
type
FakeServiceLister
[]
api
.
Service
type
FakeServiceLister
[]
*
api
.
Service
// List returns api.ServiceList, the list of all services.
func
(
f
FakeServiceLister
)
List
(
)
(
api
.
ServiceList
,
error
)
{
return
api
.
ServiceList
{
Items
:
f
}
,
nil
func
(
f
FakeServiceLister
)
List
(
labels
.
Selector
)
([]
*
api
.
Service
,
error
)
{
return
f
,
nil
}
// GetPodServices gets the services that have the selector that match the labels on the given pod
func
(
f
FakeServiceLister
)
GetPodServices
(
pod
*
api
.
Pod
)
(
services
[]
api
.
Service
,
err
error
)
{
func
(
f
FakeServiceLister
)
GetPodServices
(
pod
*
api
.
Pod
)
(
services
[]
*
api
.
Service
,
err
error
)
{
var
selector
labels
.
Selector
for
_
,
service
:=
range
f
{
for
i
:=
range
f
{
service
:=
f
[
i
]
// consider only services that are in the same namespace as the pod
if
service
.
Namespace
!=
pod
.
Namespace
{
continue
...
...
plugin/pkg/scheduler/algorithm/predicates/predicates.go
View file @
16fbb471
...
...
@@ -670,7 +670,7 @@ func (s *ServiceAffinity) CheckServiceAffinity(pod *api.Pod, meta interface{}, n
// skip looking at other pods in the service if the current pod defines all the required affinity labels
if
!
labelsExist
{
services
,
err
:=
s
.
serviceLister
.
GetPodServices
(
pod
)
if
err
==
nil
{
if
err
==
nil
&&
len
(
services
)
>
0
{
// just use the first service and get the other pods within the service
// TODO: a separate predicate can be created that tries to handle all services for the pod
selector
:=
labels
.
SelectorFromSet
(
services
[
0
]
.
Spec
.
Selector
)
...
...
plugin/pkg/scheduler/algorithm/predicates/predicates_test.go
View file @
16fbb471
...
...
@@ -1209,7 +1209,7 @@ func TestServiceAffinity(t *testing.T) {
tests
:=
[]
struct
{
pod
*
api
.
Pod
pods
[]
*
api
.
Pod
services
[]
api
.
Service
services
[]
*
api
.
Service
node
*
api
.
Node
labels
[]
string
fits
bool
...
...
@@ -1240,7 +1240,7 @@ func TestServiceAffinity(t *testing.T) {
pod
:
&
api
.
Pod
{
ObjectMeta
:
api
.
ObjectMeta
{
Labels
:
selector
}},
pods
:
[]
*
api
.
Pod
{{
Spec
:
api
.
PodSpec
{
NodeName
:
"machine1"
},
ObjectMeta
:
api
.
ObjectMeta
{
Labels
:
selector
}}},
node
:
&
node1
,
services
:
[]
api
.
Service
{{
Spec
:
api
.
ServiceSpec
{
Selector
:
selector
}}},
services
:
[]
*
api
.
Service
{{
Spec
:
api
.
ServiceSpec
{
Selector
:
selector
}}},
fits
:
true
,
labels
:
[]
string
{
"region"
},
test
:
"service pod on same node"
,
...
...
@@ -1249,7 +1249,7 @@ func TestServiceAffinity(t *testing.T) {
pod
:
&
api
.
Pod
{
ObjectMeta
:
api
.
ObjectMeta
{
Labels
:
selector
}},
pods
:
[]
*
api
.
Pod
{{
Spec
:
api
.
PodSpec
{
NodeName
:
"machine2"
},
ObjectMeta
:
api
.
ObjectMeta
{
Labels
:
selector
}}},
node
:
&
node1
,
services
:
[]
api
.
Service
{{
Spec
:
api
.
ServiceSpec
{
Selector
:
selector
}}},
services
:
[]
*
api
.
Service
{{
Spec
:
api
.
ServiceSpec
{
Selector
:
selector
}}},
fits
:
true
,
labels
:
[]
string
{
"region"
},
test
:
"service pod on different node, region match"
,
...
...
@@ -1258,7 +1258,7 @@ func TestServiceAffinity(t *testing.T) {
pod
:
&
api
.
Pod
{
ObjectMeta
:
api
.
ObjectMeta
{
Labels
:
selector
}},
pods
:
[]
*
api
.
Pod
{{
Spec
:
api
.
PodSpec
{
NodeName
:
"machine3"
},
ObjectMeta
:
api
.
ObjectMeta
{
Labels
:
selector
}}},
node
:
&
node1
,
services
:
[]
api
.
Service
{{
Spec
:
api
.
ServiceSpec
{
Selector
:
selector
}}},
services
:
[]
*
api
.
Service
{{
Spec
:
api
.
ServiceSpec
{
Selector
:
selector
}}},
fits
:
false
,
labels
:
[]
string
{
"region"
},
test
:
"service pod on different node, region mismatch"
,
...
...
@@ -1267,7 +1267,7 @@ func TestServiceAffinity(t *testing.T) {
pod
:
&
api
.
Pod
{
ObjectMeta
:
api
.
ObjectMeta
{
Labels
:
selector
,
Namespace
:
"ns1"
}},
pods
:
[]
*
api
.
Pod
{{
Spec
:
api
.
PodSpec
{
NodeName
:
"machine3"
},
ObjectMeta
:
api
.
ObjectMeta
{
Labels
:
selector
,
Namespace
:
"ns1"
}}},
node
:
&
node1
,
services
:
[]
api
.
Service
{{
Spec
:
api
.
ServiceSpec
{
Selector
:
selector
},
ObjectMeta
:
api
.
ObjectMeta
{
Namespace
:
"ns2"
}}},
services
:
[]
*
api
.
Service
{{
Spec
:
api
.
ServiceSpec
{
Selector
:
selector
},
ObjectMeta
:
api
.
ObjectMeta
{
Namespace
:
"ns2"
}}},
fits
:
true
,
labels
:
[]
string
{
"region"
},
test
:
"service in different namespace, region mismatch"
,
...
...
@@ -1276,7 +1276,7 @@ func TestServiceAffinity(t *testing.T) {
pod
:
&
api
.
Pod
{
ObjectMeta
:
api
.
ObjectMeta
{
Labels
:
selector
,
Namespace
:
"ns1"
}},
pods
:
[]
*
api
.
Pod
{{
Spec
:
api
.
PodSpec
{
NodeName
:
"machine3"
},
ObjectMeta
:
api
.
ObjectMeta
{
Labels
:
selector
,
Namespace
:
"ns2"
}}},
node
:
&
node1
,
services
:
[]
api
.
Service
{{
Spec
:
api
.
ServiceSpec
{
Selector
:
selector
},
ObjectMeta
:
api
.
ObjectMeta
{
Namespace
:
"ns1"
}}},
services
:
[]
*
api
.
Service
{{
Spec
:
api
.
ServiceSpec
{
Selector
:
selector
},
ObjectMeta
:
api
.
ObjectMeta
{
Namespace
:
"ns1"
}}},
fits
:
true
,
labels
:
[]
string
{
"region"
},
test
:
"pod in different namespace, region mismatch"
,
...
...
@@ -1285,7 +1285,7 @@ func TestServiceAffinity(t *testing.T) {
pod
:
&
api
.
Pod
{
ObjectMeta
:
api
.
ObjectMeta
{
Labels
:
selector
,
Namespace
:
"ns1"
}},
pods
:
[]
*
api
.
Pod
{{
Spec
:
api
.
PodSpec
{
NodeName
:
"machine3"
},
ObjectMeta
:
api
.
ObjectMeta
{
Labels
:
selector
,
Namespace
:
"ns1"
}}},
node
:
&
node1
,
services
:
[]
api
.
Service
{{
Spec
:
api
.
ServiceSpec
{
Selector
:
selector
},
ObjectMeta
:
api
.
ObjectMeta
{
Namespace
:
"ns1"
}}},
services
:
[]
*
api
.
Service
{{
Spec
:
api
.
ServiceSpec
{
Selector
:
selector
},
ObjectMeta
:
api
.
ObjectMeta
{
Namespace
:
"ns1"
}}},
fits
:
false
,
labels
:
[]
string
{
"region"
},
test
:
"service and pod in same namespace, region mismatch"
,
...
...
@@ -1294,7 +1294,7 @@ func TestServiceAffinity(t *testing.T) {
pod
:
&
api
.
Pod
{
ObjectMeta
:
api
.
ObjectMeta
{
Labels
:
selector
}},
pods
:
[]
*
api
.
Pod
{{
Spec
:
api
.
PodSpec
{
NodeName
:
"machine2"
},
ObjectMeta
:
api
.
ObjectMeta
{
Labels
:
selector
}}},
node
:
&
node1
,
services
:
[]
api
.
Service
{{
Spec
:
api
.
ServiceSpec
{
Selector
:
selector
}}},
services
:
[]
*
api
.
Service
{{
Spec
:
api
.
ServiceSpec
{
Selector
:
selector
}}},
fits
:
false
,
labels
:
[]
string
{
"region"
,
"zone"
},
test
:
"service pod on different node, multiple labels, not all match"
,
...
...
@@ -1303,7 +1303,7 @@ func TestServiceAffinity(t *testing.T) {
pod
:
&
api
.
Pod
{
ObjectMeta
:
api
.
ObjectMeta
{
Labels
:
selector
}},
pods
:
[]
*
api
.
Pod
{{
Spec
:
api
.
PodSpec
{
NodeName
:
"machine5"
},
ObjectMeta
:
api
.
ObjectMeta
{
Labels
:
selector
}}},
node
:
&
node4
,
services
:
[]
api
.
Service
{{
Spec
:
api
.
ServiceSpec
{
Selector
:
selector
}}},
services
:
[]
*
api
.
Service
{{
Spec
:
api
.
ServiceSpec
{
Selector
:
selector
}}},
fits
:
true
,
labels
:
[]
string
{
"region"
,
"zone"
},
test
:
"service pod on different node, multiple labels, all match"
,
...
...
plugin/pkg/scheduler/algorithm/priorities/selector_spreading.go
View file @
16fbb471
...
...
@@ -194,7 +194,7 @@ func NewServiceAntiAffinityPriority(podLister algorithm.PodLister, serviceLister
// The label to be considered is provided to the struct (ServiceAntiAffinity).
func
(
s
*
ServiceAntiAffinity
)
CalculateAntiAffinityPriority
(
pod
*
api
.
Pod
,
nodeNameToInfo
map
[
string
]
*
schedulercache
.
NodeInfo
,
nodes
[]
*
api
.
Node
)
(
schedulerapi
.
HostPriorityList
,
error
)
{
var
nsServicePods
[]
*
api
.
Pod
if
services
,
err
:=
s
.
serviceLister
.
GetPodServices
(
pod
);
err
==
nil
{
if
services
,
err
:=
s
.
serviceLister
.
GetPodServices
(
pod
);
err
==
nil
&&
len
(
services
)
>
0
{
// just use the first service and get the other pods within the service
// TODO: a separate predicate can be created that tries to handle all services for the pod
selector
:=
labels
.
SelectorFromSet
(
services
[
0
]
.
Spec
.
Selector
)
...
...
plugin/pkg/scheduler/algorithm/priorities/selector_spreading_test.go
View file @
16fbb471
This diff is collapsed.
Click to expand it.
plugin/pkg/scheduler/factory/factory.go
View file @
16fbb471
...
...
@@ -108,7 +108,7 @@ func NewConfigFactory(client *client.Client, schedulerName string, hardPodAffini
NodeLister
:
&
cache
.
StoreToNodeLister
{},
PVLister
:
&
cache
.
StoreToPVFetcher
{
Store
:
cache
.
NewStore
(
cache
.
MetaNamespaceKeyFunc
)},
PVCLister
:
&
cache
.
StoreToPVCFetcher
{
Store
:
cache
.
NewStore
(
cache
.
MetaNamespaceKeyFunc
)},
ServiceLister
:
&
cache
.
StoreToServiceLister
{
Store
:
cache
.
NewStore
(
cache
.
MetaNamespaceKeyFunc
)},
ServiceLister
:
&
cache
.
StoreToServiceLister
{
Indexer
:
cache
.
NewIndexer
(
cache
.
MetaNamespaceKeyFunc
,
cache
.
Indexers
{
cache
.
NamespaceIndex
:
cache
.
MetaNamespaceIndexFunc
}
)},
ControllerLister
:
&
cache
.
StoreToReplicationControllerLister
{
Indexer
:
cache
.
NewIndexer
(
cache
.
MetaNamespaceKeyFunc
,
cache
.
Indexers
{
cache
.
NamespaceIndex
:
cache
.
MetaNamespaceIndexFunc
})},
ReplicaSetLister
:
&
cache
.
StoreToReplicaSetLister
{
Store
:
cache
.
NewStore
(
cache
.
MetaNamespaceKeyFunc
)},
schedulerCache
:
schedulerCache
,
...
...
@@ -401,7 +401,7 @@ func (f *ConfigFactory) Run() {
// Watch and cache all service objects. Scheduler needs to find all pods
// created by the same services or ReplicationControllers/ReplicaSets, so that it can spread them correctly.
// Cache this locally.
cache
.
NewReflector
(
f
.
createServiceLW
(),
&
api
.
Service
{},
f
.
ServiceLister
.
Store
,
0
)
.
RunUntil
(
f
.
StopEverything
)
cache
.
NewReflector
(
f
.
createServiceLW
(),
&
api
.
Service
{},
f
.
ServiceLister
.
Indexer
,
0
)
.
RunUntil
(
f
.
StopEverything
)
// Watch and cache all ReplicationController objects. Scheduler needs to find all pods
// created by the same services or ReplicationControllers/ReplicaSets, so that it can spread them correctly.
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment