Update oviewer/document.go

Co-authored-by: ccoVeille <3875889+ccoVeille@users.noreply.github.com>
noborus · Dec 26, 2024 · c336015 · c336015
1 parent 05d0c46
commit c336015
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/oviewer/document.go b/oviewer/document.go
@@ -436,9 +436,10 @@ func (m *Document) storeEndNum() int {
 func (m *Document) WaitEOF() {
 	m.cond.L.Lock()
 	defer m.cond.L.Unlock()
-	if atomic.LoadInt32(&m.store.eof) == 0 {
-		m.cond.Wait()
+	if m.BufEOF() {
+		return
 	}
+	m.cond.Wait()
 }
 
 // BufEOF return true if EOF is reached.