app: simplify sleepCtx

aws · Nov 14, 2017 · d83b49a · d83b49a
1 parent d8c4796
commit d83b49a
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/agent/app/agent_windows.go b/agent/app/agent_windows.go
@@ -175,14 +175,8 @@ func (h *handler) runAgent(ctx context.Context) {
 
 // sleepCtx provides a cancelable sleep
 func sleepCtx(ctx context.Context, duration time.Duration) {
-	done := make(chan struct{})
-	time.AfterFunc(duration, func() {
-		close(done)
-	})
-	select {
-	case <-ctx.Done():
-	case <-done:
-	}
+	derivedCtx, _ := context.WithDeadline(ctx, time.Now().Add(duration))
+	<-derivedCtx.Done()
 }
 
 type termHandlerIndicator struct {