fix: send stop command after normal stop of reading

0xPolygonHermez · Oct 11, 2024 · 561a4b7 · 561a4b7
1 parent 1a0812e
commit 561a4b7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/zk/datastream/client/stream_client.go b/zk/datastream/client/stream_client.go
@@ -551,6 +551,9 @@ LOOP:
 
 		if c.header.TotalEntries == entryNum+1 {
 			log.Trace("reached the end of the stream", "header_totalEntries", c.header.TotalEntries, "entryNum", entryNum)
+			if err = c.sendStopCmd(); err != nil {
+				return fmt.Errorf("failed to send the stop command: %v", err)
+			}
 			break LOOP
 		}
 	}