Sleep cancel sleep on log follow if the user cancels

docker-archive · Jul 3, 2020 · 0057093 · 0057093
1 parent ae76e0c
commit 0057093
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):
+			}
 		}
 	}
 }