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

Check grpc server ready properly

Browse Source
This commit is contained in:
Penghao Cen
2018-01-13 05:47:49 +08:00
parent 90bc1265cf
commit b96c383ef7
2 changed files with 3 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pkg/kubelet/cm/deviceplugin/device_plugin_stub.go
View File

@@ -70,7 +70,7 @@ func (m *Stub) Start() error {
// Wait till grpc server is ready. // Wait till grpc server is ready.
for i := 0; i < 10; i++ { for i := 0; i < 10; i++ {
services := m.server.GetServiceInfo() services := m.server.GetServiceInfo()
if len(services) > 1 { if len(services) > 0 {
break break
} }
time.Sleep(1 * time.Second) time.Sleep(1 * time.Second)

12
pkg/kubelet/cm/deviceplugin/endpoint_test.go
View File

@@ -90,20 +90,12 @@ func TestRun(t *testing.T) {
go e.run() go e.run()
// Wait for the first callback to be issued. // Wait for the first callback to be issued.
select { <-callbackChan
case <-callbackChan:
break
}
p.Update(updated) p.Update(updated)
// Wait for the second callback to be issued. // Wait for the second callback to be issued.
select { <-callbackChan
case <-callbackChan:
break
}
time.Sleep(time.Second)
e.mutex.Lock() e.mutex.Lock()
defer e.mutex.Unlock() defer e.mutex.Unlock()