Terminal wrapper: exit on terminal expiration (#195)

cirruslabs · Dec 7, 2021 · a656cd0 · a656cd0
1 parent 73166ba
commit a656cd0
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/internal/executor/terminalwrapper/terminalwrapper.go b/internal/executor/terminalwrapper/terminalwrapper.go
@@ -150,7 +150,9 @@ func (wrapper *Wrapper) Wait() chan Operation {
 				})
 
 				if numActiveSessions == 0 {
-					break
+					wrapper.operationChan <- &ExitOperation{Success: true}
+
+					return
 				}
 
 				message := fmt.Sprintf("Waited %.1f seconds, but there are still %d terminal sessions open "+
@@ -160,10 +162,10 @@ func (wrapper *Wrapper) Wait() chan Operation {
 
 				continue
 			case <-wrapper.ctx.Done():
-				break
-			}
+				wrapper.operationChan <- &ExitOperation{Success: false}
 
-			wrapper.operationChan <- &ExitOperation{Success: true}
+				return
+			}
 		}
 	}()