Merge pull request #332 from docker/feat-sleep-ctx

Sleep cancel sleep on log follow if the user cancels
docker-archive · Jul 3, 2020 · 1b9dd7a · 1b9dd7a
2 parents c903362 + 0057093
commit 1b9dd7a
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/azure/aci.go b/azure/aci.go
@@ -271,7 +271,11 @@ func streamLogs(ctx context.Context, aciContext store.AciContext, containerGroup
 				fmt.Fprintln(out, logLines[i])
 			}
 
-			time.Sleep(2 * time.Second)
+			select {
+			case <-ctx.Done():
+				return nil
+			case <-time.After(2 * time.Second):
+			}
 		}
 	}
 }