Ensure all bytes are written to result stream

ansible#597
shanemcd · Apr 25, 2022 · 4c9f5ae · 4c9f5ae
1 parent 8b88e0c
commit 4c9f5ae
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pkg/workceptor/workceptor.go b/pkg/workceptor/workceptor.go
@@ -564,8 +564,8 @@ func (w *Workceptor) GetResults(ctx context.Context, unitID string, startPos int
 				}
 			}
 			if err == io.EOF {
-				stdoutSize := stdoutSize(unitdir)
-				if IsComplete(unit.Status().State) && stdoutSize >= unit.Status().StdoutSize {
+				unitStatus := unit.Status()
+				if IsComplete(unitStatus.State) && filePos >= unitStatus.StdoutSize {
 					logger.Debug("Stdout complete - closing channel for: %s \n", unitID)
 
 					return