Cancel event subscribe context in Wait.

Signed-off-by: Lantao Liu <lantaol@google.com>
2017-08-02 21:45:01 +00:00
parent 7e0063320c
commit 8fae8a5b77
2 changed files with 6 additions and 2 deletions
--- a/process.go
+++ b/process.go
@@ -69,7 +69,9 @@ func (p *process) Kill(ctx context.Context, s syscall.Signal) error {
 }

 func (p *process) Wait(ctx context.Context) (uint32, error) {
-	eventstream, err := p.task.client.EventService().Subscribe(ctx, &eventsapi.SubscribeRequest{
+	cancellable, cancel := context.WithCancel(ctx)
+	defer cancel()
+	eventstream, err := p.task.client.EventService().Subscribe(cancellable, &eventsapi.SubscribeRequest{
 		Filters: []string{"topic==" + runtime.TaskExitEventTopic},
 	})
 	if err != nil {