Additional file creation loop logic

mcphailtom · Oct 9, 2023 · d385cd1 · d385cd1
1 parent 056de6f
commit d385cd1
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/logfile.go b/logfile.go
@@ -105,13 +105,18 @@ func (lf *logFile) dataProcessor(ctx context.Context, ewCancelChan <-chan error,
 				case <-ctx.Done():
 					return
 				case so := <-lf.stateEvents:
-					if so == stateEventCreated {
+					switch so {
+					case stateEventCreated:
 						lf.created = true
 						break
+					case stateEventRemoved:
+						lf.created = false
+						continue creationLoop
+					default:
+						stopChan <- lf.filePath
+						lf.errorChan <- NewLogWhaleError(ErrorStateInternal, fmt.Sprintf("unexpected state event while waiting for file creation: %s", so), nil)
+						return
 					}
-					stopChan <- lf.filePath
-					lf.errorChan <- NewLogWhaleError(ErrorStateInternal, fmt.Sprintf("unexpected state event while waiting for file creation: %s", so), nil)
-					return
 				}
 			}